bitwarden-server/test/Core.Test/OrganizationFeatures/OrganizationApiKeys/GetOrganizationApiKeyCommandTests.cs

using Bit.Core.Entities;
using Bit.Core.Enums;
using Bit.Core.OrganizationFeatures.OrganizationApiKeys;
using Bit.Core.Repositories;
using Bit.Test.Common.AutoFixture;
using Bit.Test.Common.AutoFixture.Attributes;
using NSubstitute;
using Xunit;

namespace Bit.Core.Test.OrganizationFeatures.OrganizationApiKeys
{
    [SutProviderCustomize]
    public class GetOrganizationApiKeyCommandTests
    {
        [Theory]
        [BitAutoData]
        public async Task GetOrganizationApiKey_HasOne_Returns(SutProvider<GetOrganizationApiKeyCommand> sutProvider,
            Guid id, Guid organizationId, OrganizationApiKeyType keyType)
        {
            sutProvider.GetDependency<IOrganizationApiKeyRepository>()
                .GetManyByOrganizationIdTypeAsync(organizationId, keyType)
                .Returns(new List<OrganizationApiKey>
                {
                    new OrganizationApiKey
                    {
                        Id = id,
                        OrganizationId = organizationId,
                        ApiKey = "test",
                        Type = keyType,
                        RevisionDate = DateTime.Now.AddDays(-1),
                    },
                });

            var apiKey = await sutProvider.Sut.GetOrganizationApiKeyAsync(organizationId, keyType);
            Assert.NotNull(apiKey);
            Assert.Equal(id, apiKey.Id);
        }

        [Theory]
        [BitAutoData]
        public async Task GetOrganizationApiKey_HasTwo_Throws(SutProvider<GetOrganizationApiKeyCommand> sutProvider,
            Guid organizationId, OrganizationApiKeyType keyType)
        {
            sutProvider.GetDependency<IOrganizationApiKeyRepository>()
                .GetManyByOrganizationIdTypeAsync(organizationId, keyType)
                .Returns(new List<OrganizationApiKey>
                {
                    new OrganizationApiKey
                    {
                        Id = Guid.NewGuid(),
                        OrganizationId = organizationId,
                        ApiKey = "test",
                        Type = keyType,
                        RevisionDate = DateTime.Now.AddDays(-1),
                    },
                    new OrganizationApiKey
                    {
                        Id = Guid.NewGuid(),
                        OrganizationId = organizationId,
                        ApiKey = "test_other",
                        Type = keyType,
                        RevisionDate = DateTime.Now.AddDays(-1),
                    },
                });

            await Assert.ThrowsAsync<InvalidOperationException>(
                async () => await sutProvider.Sut.GetOrganizationApiKeyAsync(organizationId, keyType));
        }

        [Theory]
        [BitAutoData]
        public async Task GetOrganizationApiKey_HasNone_CreatesAndReturns(SutProvider<GetOrganizationApiKeyCommand> sutProvider,
            Guid organizationId, OrganizationApiKeyType keyType)
        {
            sutProvider.GetDependency<IOrganizationApiKeyRepository>()
                .GetManyByOrganizationIdTypeAsync(organizationId, keyType)
                .Returns(Enumerable.Empty<OrganizationApiKey>());

            var apiKey = await sutProvider.Sut.GetOrganizationApiKeyAsync(organizationId, keyType);

            Assert.NotNull(apiKey);
            Assert.Equal(organizationId, apiKey.OrganizationId);
            Assert.Equal(keyType, apiKey.Type);
            await sutProvider.GetDependency<IOrganizationApiKeyRepository>()
                .Received(1)
                .CreateAsync(Arg.Any<OrganizationApiKey>());
        }

        [Theory]
        [BitAutoData]
        public async Task GetOrganizationApiKey_BadType_Throws(SutProvider<GetOrganizationApiKeyCommand> sutProvider,
            Guid organizationId, OrganizationApiKeyType keyType)
        {
            keyType = (OrganizationApiKeyType)byte.MaxValue;

            await Assert.ThrowsAsync<ArgumentOutOfRangeException>(
                async () => await sutProvider.Sut.GetOrganizationApiKeyAsync(organizationId, keyType));
        }
    }
}