1
0
mirror of https://github.com/bitwarden/server.git synced 2024-12-12 15:26:48 +01:00
bitwarden-server/test/Infrastructure.EFIntegration.Test/AutoFixture/CipherFixtures.cs

108 lines
4.0 KiB
C#
Raw Normal View History

using System.Text.Json;
using AutoFixture;
using AutoFixture.Kernel;
using Bit.Core.Entities;
using Bit.Core.Models.Data;
using Bit.Core.Test.AutoFixture.UserFixtures;
using Bit.Infrastructure.EFIntegration.Test.AutoFixture.Relays;
using Bit.Infrastructure.EntityFramework.Repositories;
using Bit.Test.Common.AutoFixture;
using Bit.Test.Common.AutoFixture.Attributes;
2022-08-29 22:06:55 +02:00
namespace Bit.Infrastructure.EFIntegration.Test.AutoFixture;
internal class CipherBuilder : ISpecimenBuilder
{
2022-08-29 22:06:55 +02:00
public bool OrganizationOwned { get; set; }
public object Create(object request, ISpecimenContext context)
{
2022-08-29 22:06:55 +02:00
if (context == null)
{
2022-08-29 22:06:55 +02:00
throw new ArgumentNullException(nameof(context));
}
2022-08-29 22:06:55 +02:00
var type = request as Type;
if (type == null || (type != typeof(Cipher) && type != typeof(List<Cipher>)))
{
return new NoSpecimen();
}
2022-08-29 22:06:55 +02:00
var fixture = new Fixture();
fixture.Customizations.Insert(0, new MaxLengthStringRelay());
2022-08-29 22:06:55 +02:00
if (!OrganizationOwned)
{
fixture.Customize<Cipher>(composer => composer
.Without(c => c.OrganizationId));
}
2022-08-29 20:53:16 +02:00
2022-08-29 22:06:55 +02:00
// Can't test valid Favorites and Folders without creating those values inide each test,
// since we won't have any UserIds until the test is running & creating data
fixture.Customize<Cipher>(c => c
.Without(e => e.Favorites)
.Without(e => e.Folders));
//
var serializerOptions = new JsonSerializerOptions()
{
PropertyNamingPolicy = JsonNamingPolicy.CamelCase
};
if (type == typeof(Cipher))
{
var obj = fixture.WithAutoNSubstitutions().Create<Cipher>();
var cipherData = fixture.WithAutoNSubstitutions().Create<CipherLoginData>();
var cipherAttachements = fixture.WithAutoNSubstitutions().Create<List<CipherAttachment>>();
obj.Data = JsonSerializer.Serialize(cipherData, serializerOptions);
obj.Attachments = JsonSerializer.Serialize(cipherAttachements, serializerOptions);
2022-08-29 22:06:55 +02:00
return obj;
}
if (type == typeof(List<Cipher>))
{
var ciphers = fixture.WithAutoNSubstitutions().CreateMany<Cipher>().ToArray();
for (var i = 0; i < ciphers.Count(); i++)
{
var cipherData = fixture.WithAutoNSubstitutions().Create<CipherLoginData>();
var cipherAttachements = fixture.WithAutoNSubstitutions().Create<List<CipherAttachment>>();
2022-08-29 22:06:55 +02:00
ciphers[i].Data = JsonSerializer.Serialize(cipherData, serializerOptions);
ciphers[i].Attachments = JsonSerializer.Serialize(cipherAttachements, serializerOptions);
}
2022-08-29 22:06:55 +02:00
return ciphers;
}
2022-08-29 22:06:55 +02:00
return new NoSpecimen();
}
2022-08-29 22:06:55 +02:00
}
2022-08-29 22:06:55 +02:00
internal class EfCipher : ICustomization
{
public bool OrganizationOwned { get; set; }
public void Customize(IFixture fixture)
{
2022-08-29 22:06:55 +02:00
fixture.Customizations.Add(new GlobalSettingsBuilder());
fixture.Customizations.Add(new CipherBuilder()
{
2022-08-29 22:06:55 +02:00
OrganizationOwned = OrganizationOwned
});
fixture.Customizations.Add(new UserBuilder());
fixture.Customizations.Add(new OrganizationBuilder());
fixture.Customizations.Add(new OrganizationUserBuilder());
fixture.Customizations.Add(new EfRepositoryListBuilder<CipherRepository>());
fixture.Customizations.Add(new EfRepositoryListBuilder<OrganizationRepository>());
fixture.Customizations.Add(new EfRepositoryListBuilder<OrganizationUserRepository>());
fixture.Customizations.Add(new EfRepositoryListBuilder<UserRepository>());
fixture.Customizations.Add(new EfRepositoryListBuilder<CollectionRepository>());
}
2022-08-29 22:06:55 +02:00
}
internal class EfUserCipherCustomizeAttribute : BitCustomizeAttribute
2022-08-29 22:06:55 +02:00
{
public override ICustomization GetCustomization() => new EfCipher();
2022-08-29 22:06:55 +02:00
}
internal class EfOrganizationCipherCustomizeAttribute : BitCustomizeAttribute
2022-08-29 22:06:55 +02:00
{
public override ICustomization GetCustomization() => new EfCipher();
}