mirror of
https://github.com/bitwarden/server.git
synced 2024-11-25 12:45:18 +01:00
6f04298e17
* ServerProtectedData for user entity * remove using statements * formatting * use data protection libs * no async * add data protection to ef user repo * switch to `SetApplicationName` per ASPNET docs * null checks * cleanup * value converter for EF * new line at eof * fix using * remove folder ref * restore ctor * fix lint * use global constant * UseApplicationServiceProvider for integration tests * implement constant for DatabaseFieldProtectedPrefix * Fix EF IntegrationTest * restore original values after protect and save * lint fixes * Use Constants Co-authored-by: Justin Baur <19896123+justindbaur@users.noreply.github.com>
57 lines
1.6 KiB
C#
57 lines
1.6 KiB
C#
using System.Reflection;
|
|
using System.Text;
|
|
using AutoFixture;
|
|
using AutoFixture.Kernel;
|
|
using AutoFixture.Xunit2;
|
|
using Bit.Core;
|
|
using Bit.Core.Test.Helpers.Factories;
|
|
using Microsoft.AspNetCore.DataProtection;
|
|
using Moq;
|
|
|
|
namespace Bit.Test.Common.AutoFixture;
|
|
|
|
public class GlobalSettingsBuilder : ISpecimenBuilder
|
|
{
|
|
public object Create(object request, ISpecimenContext context)
|
|
{
|
|
if (context == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(context));
|
|
}
|
|
|
|
var fixture = new Fixture();
|
|
|
|
if (request is not ParameterInfo pi)
|
|
{
|
|
return new NoSpecimen();
|
|
}
|
|
|
|
if (pi.ParameterType == typeof(Bit.Core.Settings.GlobalSettings))
|
|
{
|
|
return GlobalSettingsFactory.GlobalSettings;
|
|
}
|
|
|
|
if (pi.ParameterType == typeof(IDataProtectionProvider))
|
|
{
|
|
var dataProtector = new Mock<IDataProtector>();
|
|
dataProtector
|
|
.Setup(d => d.Unprotect(It.IsAny<byte[]>()))
|
|
.Returns<byte[]>(data => Encoding.UTF8.GetBytes(Constants.DatabaseFieldProtectedPrefix + Encoding.UTF8.GetString(data)));
|
|
|
|
var dataProtectionProvider = new Mock<IDataProtectionProvider>();
|
|
dataProtectionProvider
|
|
.Setup(x => x.CreateProtector(Constants.DatabaseFieldProtectorPurpose))
|
|
.Returns(dataProtector.Object);
|
|
|
|
return dataProtectionProvider.Object;
|
|
}
|
|
|
|
return new NoSpecimen();
|
|
}
|
|
}
|
|
|
|
public class GlobalSettingsCustomizeAttribute : CustomizeAttribute
|
|
{
|
|
public override ICustomization GetCustomization(ParameterInfo parameter) => new GlobalSettings();
|
|
}
|