mirror of
https://github.com/bitwarden/server.git
synced 2025-07-04 01:22:50 -05:00
[SM-151] Move EF Dapper tests to Infrastructure.EFIntegration.Test (#2204)
This commit is contained in:
@ -1,15 +1,5 @@
|
||||
using System.Text.Json;
|
||||
using AutoFixture;
|
||||
using AutoFixture.Kernel;
|
||||
using AutoFixture;
|
||||
using Bit.Core.Entities;
|
||||
using Bit.Core.Models.Data;
|
||||
using Bit.Core.Test.AutoFixture.EntityFrameworkRepositoryFixtures;
|
||||
using Bit.Core.Test.AutoFixture.OrganizationFixtures;
|
||||
using Bit.Core.Test.AutoFixture.OrganizationUserFixtures;
|
||||
using Bit.Core.Test.AutoFixture.Relays;
|
||||
using Bit.Core.Test.AutoFixture.UserFixtures;
|
||||
using Bit.Infrastructure.EntityFramework.Repositories;
|
||||
using Bit.Test.Common.AutoFixture;
|
||||
using Bit.Test.Common.AutoFixture.Attributes;
|
||||
using Core.Models.Data;
|
||||
|
||||
@ -43,92 +33,6 @@ namespace Bit.Core.Test.AutoFixture.CipherFixtures
|
||||
}
|
||||
}
|
||||
|
||||
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());
|
||||
fixture.Customizations.Add(new IgnoreVirtualMembersCustomization());
|
||||
|
||||
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 UserCipherAutoDataAttribute : CustomAutoDataAttribute
|
||||
{
|
||||
public UserCipherAutoDataAttribute(string userId = null) : base(new SutProviderCustomization(),
|
||||
@ -162,26 +66,4 @@ namespace Bit.Core.Test.AutoFixture.CipherFixtures
|
||||
typeof(OrganizationCipher) }, values)
|
||||
{ }
|
||||
}
|
||||
|
||||
internal class EfUserCipherAutoDataAttribute : CustomAutoDataAttribute
|
||||
{
|
||||
public EfUserCipherAutoDataAttribute() : base(new SutProviderCustomization(), new EfCipher())
|
||||
{ }
|
||||
}
|
||||
|
||||
internal class EfOrganizationCipherAutoDataAttribute : CustomAutoDataAttribute
|
||||
{
|
||||
public EfOrganizationCipherAutoDataAttribute() : base(new SutProviderCustomization(), new EfCipher()
|
||||
{
|
||||
OrganizationOwned = true,
|
||||
})
|
||||
{ }
|
||||
}
|
||||
|
||||
internal class InlineEfCipherAutoDataAttribute : InlineCustomAutoDataAttribute
|
||||
{
|
||||
public InlineEfCipherAutoDataAttribute(params object[] values) : base(new[] { typeof(SutProviderCustomization),
|
||||
typeof(EfCipher) }, values)
|
||||
{ }
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user