abp/modules/cms-kit/test/Volo.CmsKit.Domain.Tests/Menus/MenuManager_Test.cs

using Shouldly;
using System;
using System.Linq;
using System.Threading.Tasks;
using Volo.CmsKit.Pages;
using Xunit;

namespace Volo.CmsKit.Menus
{
    public class MenuManager_Test : CmsKitDomainTestBase
    {
        private readonly MenuItemManager menuManager;
        private readonly CmsKitTestData testData;
        private readonly IMenuItemRepository menuItemRepository;
        private readonly IPageRepository pageRepository;

        public MenuManager_Test()
        {
            menuManager = GetRequiredService<MenuItemManager>();
            testData = GetRequiredService<CmsKitTestData>();
            menuItemRepository = GetRequiredService<IMenuItemRepository>();
            pageRepository = GetRequiredService<IPageRepository>();
        }

        [Fact]
        public async Task SetPageUrl_ShouldSetUrlSameWithPage_WithExistingPage()
        {
            var page = await pageRepository.GetAsync(testData.Page_1_Id);

            var menuItem = await menuItemRepository.GetAsync(testData.MenuItem_1_Id);

            menuManager.SetPageUrl(menuItem, page);

            menuItem.Url.ShouldNotBeNullOrEmpty();
            menuItem.Url.ShouldBe(PageConsts.UrlPrefix + page.Slug);
        }

        [Fact]
        public async Task MoveAsync_ShouldMoveCorrectly_UnderAnotherMenu()
        {
            await menuManager.MoveAsync(testData.MenuItem_3_Id, testData.MenuItem_1_Id);

            var menu = await menuItemRepository.GetAsync(testData.MenuItem_3_Id);

            menu.ParentId.ShouldBe(testData.MenuItem_1_Id);
        }

        [Fact]
        public async Task MoveAsync_ShouldChangePositionCorrectly_UnderRoot()
        {
            await menuManager.MoveAsync(testData.MenuItem_2_Id, null, 0);

            var menuItems = await menuItemRepository.GetListAsync();

            var menuItem1 = menuItems.First(x => x.Id == testData.MenuItem_1_Id);
            var menuItem2 = menuItems.First(x => x.Id == testData.MenuItem_2_Id);

            menuItem1.Order.ShouldBeGreaterThan(menuItem2.Order);
        }

        [Fact]
        public async Task OrganizeTreeOrderForMenuItem_ShouldWorkProperly_WithNewMenuItem()
        {
            var menu3Id = Guid.NewGuid();

            var menuItems = await menuItemRepository.GetListAsync();

            var menuItem1 = menuItems.First(x => x.Id == testData.MenuItem_1_Id);
            var menuItem2 = menuItems.First(x => x.Id == testData.MenuItem_2_Id);
            var menuItem3 = new MenuItem(menu3Id, "Menu 3", "#");

            menuItems.Add(menuItem3);

            menuItem3.Order = 0;

            menuManager.OrganizeTreeOrderForMenuItem(menuItems, menuItem3);

            menuItem3.Order.ShouldBeLessThan(menuItem1.Order);
            menuItem3.Order.ShouldBeLessThan(menuItem2.Order);
        }
    }
}