squidex/backend/tests/Squidex.Web.Tests/CommandMiddlewares/EnrichWithTeamIdCommandMiddlewareTests.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using FakeItEasy;
using Microsoft.AspNetCore.Http;
using Squidex.Domain.Apps.Entities;
using Squidex.Domain.Apps.Entities.Teams.Commands;
using Squidex.Domain.Apps.Entities.TestHelpers;
using Squidex.Infrastructure;
using Squidex.Infrastructure.Commands;
using Squidex.Web.Pipeline;
using Xunit;

namespace Squidex.Web.CommandMiddlewares
{
    public class EnrichWithTeamIdCommandMiddlewareTests
    {
        private readonly IHttpContextAccessor httpContextAccessor = A.Fake<IHttpContextAccessor>();
        private readonly DomainId teamId = DomainId.NewGuid();
        private readonly HttpContext httpContext = new DefaultHttpContext();
        private readonly EnrichWithTeamIdCommandMiddleware sut;

        public EnrichWithTeamIdCommandMiddlewareTests()
        {
            httpContext.Features.Set<ITeamFeature>(new TeamFeature(Mocks.Team(teamId)));

            A.CallTo(() => httpContextAccessor.HttpContext)
                .Returns(httpContext);

            sut = new EnrichWithTeamIdCommandMiddleware(httpContextAccessor);
        }

        [Fact]
        public async Task Should_throw_exception_if_team_not_found()
        {
            httpContext.Features.Set<ITeamFeature>(null);

            await Assert.ThrowsAsync<InvalidOperationException>(() => HandleAsync(new UpdateTeam()));
        }

        [Fact]
        public async Task Should_assign_id_to_command()
        {
            var context = await HandleAsync(new UpdateTeam());

            Assert.Equal(teamId, ((ITeamCommand)context.Command).TeamId);
        }

        [Fact]
        public async Task Should_not_override_existing_id()
        {
            var customId = DomainId.NewGuid();

            var context = await HandleAsync(new UpdateTeam { TeamId = customId });

            Assert.Equal(customId, ((ITeamCommand)context.Command).TeamId);
        }

        private async Task<CommandContext> HandleAsync(ITeamCommand command)
        {
            var commandContext = new CommandContext(command, A.Fake<ICommandBus>());

            await sut.HandleAsync(commandContext, default);

            return commandContext;
        }
    }
}