Umbraco-CMS/tests/Umbraco.Tests.Integration/Umbraco.Infrastructure/Persistence/Repositories/NotificationsRepositoryTest.cs

// Copyright (c) Umbraco.
// See LICENSE for more details.

using System.Globalization;
using System.Linq;
using Moq;
using NUnit.Framework;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Models;
using Umbraco.Cms.Core.Models.Entities;
using Umbraco.Cms.Core.Models.Membership;
using Umbraco.Cms.Core.Services;
using Umbraco.Cms.Infrastructure.Persistence.Dtos;
using Umbraco.Cms.Infrastructure.Persistence.Repositories.Implement;
using Umbraco.Cms.Infrastructure.Scoping;
using Umbraco.Cms.Tests.Common.Testing;
using Umbraco.Cms.Tests.Integration.Testing;

namespace Umbraco.Cms.Tests.Integration.Umbraco.Infrastructure.Persistence.Repositories;

[TestFixture]
[UmbracoTest(Database = UmbracoTestOptions.Database.NewSchemaPerTest)]
internal sealed class NotificationsRepositoryTest : UmbracoIntegrationTest
{
    [Test]
    public void CreateNotification()
    {
        var provider = ScopeProvider;
        using (var scope = provider.CreateScope())
        {
            var repo = new NotificationsRepository((IScopeAccessor)provider);

            var node = new NodeDto // create bogus item so we can add a notification
            {
                CreateDate = DateTime.UtcNow,
                Level = 1,
                NodeObjectType = Constants.ObjectTypes.ContentItem,
                ParentId = -1,
                Path = "-1,123",
                SortOrder = 1,
                Text = "hello",
                Trashed = false,
                UniqueId = Guid.NewGuid(),
                UserId = Constants.Security.SuperUserId
            };
            var result = ScopeAccessor.AmbientScope.Database.Insert(node);
            var entity = Mock.Of<IEntity>(e => e.Id == node.NodeId);
            var user = Mock.Of<IUser>(e => e.Id == node.UserId);

            repo.TryCreateNotification(user, entity, "A", out Notification? notification);
            Assert.IsNotNull(notification);

            Assert.AreEqual("A", notification.Action);
            Assert.AreEqual(node.NodeId, notification.EntityId);
            Assert.AreEqual(node.NodeObjectType, notification.EntityType);
            Assert.AreEqual(node.UserId, notification.UserId);
        }
    }

    [Test]
    public void GetUserNotifications()
    {
        var provider = ScopeProvider;
        using (var scope = provider.CreateScope())
        {
            var repo = new NotificationsRepository((IScopeAccessor)provider);

            var userDto = new UserDto
            {
                Email = "test",
                Login = "test",
                Password = "test",
                UserName = "test",
                UserLanguage = "en",
                CreateDate = DateTime.UtcNow,
                UpdateDate = DateTime.UtcNow,
            };
            ScopeAccessor.AmbientScope.Database.Insert(userDto);

            var userNew = Mock.Of<IUser>(e => e.Id == userDto.Id);
            var userAdmin = Mock.Of<IUser>(e => e.Id == Constants.Security.SuperUserId);

            for (var i = 0; i < 10; i++)
            {
                var node = new NodeDto
                {
                    CreateDate = DateTime.UtcNow,
                    Level = 1,
                    NodeObjectType = Constants.ObjectTypes.ContentItem,
                    ParentId = -1,
                    Path = "-1," + i,
                    SortOrder = 1,
                    Text = "hello" + i,
                    Trashed = false,
                    UniqueId = Guid.NewGuid(),
                    UserId = -1
                };
                var result = ScopeAccessor.AmbientScope.Database.Insert(node);
                var entity = Mock.Of<IEntity>(e => e.Id == node.NodeId);
                repo.TryCreateNotification(i % 2 == 0 ? userAdmin : userNew, entity, i.ToString(CultureInfo.InvariantCulture), out Notification? notification);
            }

            var notifications = repo.GetUserNotifications(userAdmin);

            Assert.AreEqual(5, notifications.Count());
        }
    }

    [Test]
    public void GetEntityNotifications()
    {
        var provider = ScopeProvider;
        using (var scope = provider.CreateScope())
        {
            var repo = new NotificationsRepository((IScopeAccessor)provider);

            var node1 = new NodeDto
            {
                CreateDate = DateTime.UtcNow,
                Level = 1,
                NodeObjectType = Constants.ObjectTypes.ContentItem,
                ParentId = -1,
                Path = "-1,1",
                SortOrder = 1,
                Text = "hello1",
                Trashed = false,
                UniqueId = Guid.NewGuid(),
                UserId = -1
            };
            ScopeAccessor.AmbientScope.Database.Insert(node1);
            var entity1 = Mock.Of<IEntity>(e => e.Id == node1.NodeId);
            var node2 = new NodeDto
            {
                CreateDate = DateTime.UtcNow,
                Level = 1,
                NodeObjectType = Constants.ObjectTypes.ContentItem,
                ParentId = -1,
                Path = "-1,2",
                SortOrder = 1,
                Text = "hello2",
                Trashed = false,
                UniqueId = Guid.NewGuid(),
                UserId = -1
            };
            ScopeAccessor.AmbientScope.Database.Insert(node2);
            var entity2 = Mock.Of<IEntity>(e => e.Id == node2.NodeId);

            for (var i = 0; i < 10; i++)
            {
                var userDto = new UserDto
                {
                    Key = Guid.NewGuid(),
                    Email = "test" + i,
                    Login = "test" + i,
                    Password = "test",
                    UserName = "test" + i,
                    UserLanguage = "en",
                    CreateDate = DateTime.UtcNow,
                    UpdateDate = DateTime.UtcNow
                };
                ScopeAccessor.AmbientScope.Database.Insert(userDto);
                var userNew = Mock.Of<IUser>(e => e.Id == userDto.Id);
                repo.TryCreateNotification(userNew, i % 2 == 0 ? entity1 : entity2, i.ToString(CultureInfo.InvariantCulture), out Notification? notification);
            }

            var notifications = repo.GetEntityNotifications(entity1);

            Assert.AreEqual(5, notifications.Count());
        }
    }

    [Test]
    public void Delete_By_Entity()
    {
        var provider = ScopeProvider;
        using (var scope = provider.CreateScope())
        {
            var repo = new NotificationsRepository((IScopeAccessor)provider);

            var node1 = new NodeDto
            {
                CreateDate = DateTime.UtcNow,
                Level = 1,
                NodeObjectType = Constants.ObjectTypes.ContentItem,
                ParentId = -1,
                Path = "-1,1",
                SortOrder = 1,
                Text = "hello1",
                Trashed = false,
                UniqueId = Guid.NewGuid(),
                UserId = -1
            };
            ScopeAccessor.AmbientScope.Database.Insert(node1);
            var entity1 = Mock.Of<IEntity>(e => e.Id == node1.NodeId);
            var node2 = new NodeDto
            {
                CreateDate = DateTime.UtcNow,
                Level = 1,
                NodeObjectType = Constants.ObjectTypes.ContentItem,
                ParentId = -1,
                Path = "-1,2",
                SortOrder = 1,
                Text = "hello2",
                Trashed = false,
                UniqueId = Guid.NewGuid(),
                UserId = -1
            };
            ScopeAccessor.AmbientScope.Database.Insert(node2);
            var entity2 = Mock.Of<IEntity>(e => e.Id == node2.NodeId);

            for (var i = 0; i < 10; i++)
            {
                var userDto = new UserDto
                {
                    Key = Guid.NewGuid(),
                    Email = "test" + i,
                    Login = "test" + i,
                    Password = "test",
                    UserName = "test" + i,
                    UserLanguage = "en",
                    CreateDate = DateTime.UtcNow,
                    UpdateDate = DateTime.UtcNow
                };
                ScopeAccessor.AmbientScope.Database.Insert(userDto);
                var userNew = Mock.Of<IUser>(e => e.Id == userDto.Id);
                repo.TryCreateNotification(userNew, i % 2 == 0 ? entity1 : entity2, i.ToString(CultureInfo.InvariantCulture), out Notification? notification);
            }

            var delCount = repo.DeleteNotifications(entity1);

            Assert.AreEqual(5, delCount);
        }
    }

    [Test]
    public void Delete_By_User()
    {
        var provider = ScopeProvider;
        using (var scope = provider.CreateScope())
        {
            var repo = new NotificationsRepository((IScopeAccessor)provider);

            var userDto = new UserDto
            {
                Email = "test",
                Login = "test",
                Password = "test",
                UserName = "test",
                UserLanguage = "en",
                CreateDate = DateTime.UtcNow,
                UpdateDate = DateTime.UtcNow
            };
            ScopeAccessor.AmbientScope.Database.Insert(userDto);

            var userNew = Mock.Of<IUser>(e => e.Id == userDto.Id);
            var userAdmin = Mock.Of<IUser>(e => e.Id == Constants.Security.SuperUserId);

            for (var i = 0; i < 10; i++)
            {
                var node = new NodeDto
                {
                    CreateDate = DateTime.UtcNow,
                    Level = 1,
                    NodeObjectType = Constants.ObjectTypes.ContentItem,
                    ParentId = -1,
                    Path = "-1," + i,
                    SortOrder = 1,
                    Text = "hello" + i,
                    Trashed = false,
                    UniqueId = Guid.NewGuid(),
                    UserId = -1
                };
                var result = ScopeAccessor.AmbientScope.Database.Insert(node);
                var entity = Mock.Of<IEntity>(e => e.Id == node.NodeId);
                repo.TryCreateNotification(i % 2 == 0 ? userAdmin : userNew, entity, i.ToString(CultureInfo.InvariantCulture), out Notification? notification);
            }

            var delCount = repo.DeleteNotifications(userAdmin);

            Assert.AreEqual(5, delCount);
        }
    }
}