bitwarden-server/test/Api.Test/Controllers/CollectionsControllerTests.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Bit.Api.Controllers;
using Bit.Api.Models.Request;
using Bit.Api.Test.AutoFixture.Attributes;
using Bit.Core.Context;
using Bit.Core.Entities;
using Bit.Core.Exceptions;
using Bit.Core.Models.Data;
using Bit.Core.Repositories;
using Bit.Core.Services;
using Bit.Test.Common.AutoFixture;
using Bit.Test.Common.AutoFixture.Attributes;
using NSubstitute;
using Xunit;

namespace Bit.Api.Test.Controllers
{
    [ControllerCustomize(typeof(CollectionsController))]
    [SutProviderCustomize]
    public class CollectionsControllerTests
    {
        [Theory, BitAutoData]
        public async Task Post_Success(Guid orgId, SutProvider<CollectionsController> sutProvider)
        {
            sutProvider.GetDependency<ICurrentContext>()
                .CreateNewCollections(orgId)
                .Returns(true);

            sutProvider.GetDependency<ICurrentContext>()
                .EditAnyCollection(orgId)
                .Returns(false);

            var collectionRequest = new CollectionRequestModel
            {
                Name = "encrypted_string",
                ExternalId = "my_external_id"
            };

            _ = await sutProvider.Sut.Post(orgId, collectionRequest);

            await sutProvider.GetDependency<ICollectionService>()
                .Received(1)
                .SaveAsync(Arg.Any<Collection>(), Arg.Any<IEnumerable<SelectionReadOnly>>(), null);
        }

        [Theory, BitAutoData]
        public async Task Put_Success(Guid orgId, Guid collectionId, Guid userId, CollectionRequestModel collectionRequest,
            SutProvider<CollectionsController> sutProvider)
        {
            sutProvider.GetDependency<ICurrentContext>()
                .ViewAssignedCollections(orgId)
                .Returns(true);

            sutProvider.GetDependency<ICurrentContext>()
                .EditAssignedCollections(orgId)
                .Returns(true);

            sutProvider.GetDependency<ICurrentContext>()
                .UserId
                .Returns(userId);

            sutProvider.GetDependency<ICollectionRepository>()
                .GetByIdAsync(collectionId, userId)
                .Returns(new CollectionDetails
                {
                    OrganizationId = orgId,
                });

            _ = await sutProvider.Sut.Put(orgId, collectionId, collectionRequest);
        }

        [Theory, BitAutoData]
        public async Task Put_CanNotEditAssignedCollection_ThrowsNotFound(Guid orgId, Guid collectionId, Guid userId, CollectionRequestModel collectionRequest,
            SutProvider<CollectionsController> sutProvider)
        {
            sutProvider.GetDependency<ICurrentContext>()
                .EditAssignedCollections(orgId)
                .Returns(true);

            sutProvider.GetDependency<ICurrentContext>()
                .UserId
                .Returns(userId);

            sutProvider.GetDependency<ICollectionRepository>()
                .GetByIdAsync(collectionId, userId)
                .Returns(Task.FromResult<CollectionDetails>(null));

            _ = await Assert.ThrowsAsync<NotFoundException>(async () => await sutProvider.Sut.Put(orgId, collectionId, collectionRequest));
        }
    }
}