mirror of
https://github.com/bitwarden/server.git
synced 2024-12-11 15:17:44 +01:00
a6d97118fa
* Update ProviderService tests * Use BitAutoData in CipherService tests * Simplify UserCipher fixture Because we use a single customizer for all ciphers, they all have the same userId. * Clean up more cipher fixtures * Swap Cipher Fixtures to BitCustomizeAttribute * Clean up collection fixtures * Clean up GroupFixtures * Move SendService Tests to BitAutoData * Clean up Organization Fixtures TODO: The customize attributes should not be customizing more than one class * Name files after the class they contain * Clear up usage of CustomAutoDataAttribute in tests * Clean up usages of InlineCustomAutoData * format * Manually merge with file-scoped-namespace changes
108 lines
4.0 KiB
C#
108 lines
4.0 KiB
C#
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;
|
|
|
|
namespace Bit.Infrastructure.EFIntegration.Test.AutoFixture;
|
|
|
|
internal class CipherBuilder : ISpecimenBuilder
|
|
{
|
|
public bool OrganizationOwned { get; set; }
|
|
public object Create(object request, ISpecimenContext context)
|
|
{
|
|
if (context == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(context));
|
|
}
|
|
|
|
var type = request as Type;
|
|
if (type == null || (type != typeof(Cipher) && type != typeof(List<Cipher>)))
|
|
{
|
|
return new NoSpecimen();
|
|
}
|
|
|
|
var fixture = new Fixture();
|
|
fixture.Customizations.Insert(0, new MaxLengthStringRelay());
|
|
|
|
if (!OrganizationOwned)
|
|
{
|
|
fixture.Customize<Cipher>(composer => composer
|
|
.Without(c => c.OrganizationId));
|
|
}
|
|
|
|
// 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);
|
|
|
|
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>>();
|
|
ciphers[i].Data = JsonSerializer.Serialize(cipherData, serializerOptions);
|
|
ciphers[i].Attachments = JsonSerializer.Serialize(cipherAttachements, serializerOptions);
|
|
}
|
|
|
|
return ciphers;
|
|
}
|
|
|
|
return new NoSpecimen();
|
|
}
|
|
}
|
|
|
|
internal class EfCipher : ICustomization
|
|
{
|
|
public bool OrganizationOwned { get; set; }
|
|
public void Customize(IFixture fixture)
|
|
{
|
|
fixture.Customizations.Add(new GlobalSettingsBuilder());
|
|
fixture.Customizations.Add(new CipherBuilder()
|
|
{
|
|
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>());
|
|
}
|
|
}
|
|
|
|
internal class EfUserCipherCustomizeAttribute : BitCustomizeAttribute
|
|
{
|
|
public override ICustomization GetCustomization() => new EfCipher();
|
|
}
|
|
|
|
internal class EfOrganizationCipherCustomizeAttribute : BitCustomizeAttribute
|
|
{
|
|
public override ICustomization GetCustomization() => new EfCipher();
|
|
}
|