mirror of
https://github.com/bitwarden/server.git
synced 2024-12-05 14:23:30 +01:00
88dd745070
* [PM-1188] add sso project to auth * [PM-1188] move sso api models to auth * [PM-1188] fix sso api model namespace & imports * [PM-1188] move core files to auth * [PM-1188] fix core sso namespace & models * [PM-1188] move sso repository files to auth * [PM-1188] fix sso repo files namespace & imports * [PM-1188] move sso sql files to auth folder * [PM-1188] move sso test files to auth folders * [PM-1188] fix sso tests namespace & imports * [PM-1188] move auth api files to auth folder * [PM-1188] fix auth api files namespace & imports * [PM-1188] move auth core files to auth folder * [PM-1188] fix auth core files namespace & imports * [PM-1188] move auth email templates to auth folder * [PM-1188] move auth email folder back into shared directory * [PM-1188] fix auth email names * [PM-1188] move auth core models to auth folder * [PM-1188] fix auth model namespace & imports * [PM-1188] add entire Identity project to auth codeowners * [PM-1188] fix auth orm files namespace & imports * [PM-1188] move auth orm files to auth folder * [PM-1188] move auth sql files to auth folder * [PM-1188] move auth tests to auth folder * [PM-1188] fix auth test files namespace & imports * [PM-1188] move emergency access api files to auth folder * [PM-1188] fix emergencyaccess api files namespace & imports * [PM-1188] move emergency access core files to auth folder * [PM-1188] fix emergency access core files namespace & imports * [PM-1188] move emergency access orm files to auth folder * [PM-1188] fix emergency access orm files namespace & imports * [PM-1188] move emergency access sql files to auth folder * [PM-1188] move emergencyaccess test files to auth folder * [PM-1188] fix emergency access test files namespace & imports * [PM-1188] move captcha files to auth folder * [PM-1188] fix captcha files namespace & imports * [PM-1188] move auth admin files into auth folder * [PM-1188] fix admin auth files namespace & imports - configure mvc to look in auth folders for views * [PM-1188] remove extra imports and formatting * [PM-1188] fix ef auth model imports * [PM-1188] fix DatabaseContextModelSnapshot paths * [PM-1188] fix grant import in ef * [PM-1188] update sqlproj * [PM-1188] move missed sqlproj files * [PM-1188] move auth ef models out of auth folder * [PM-1188] fix auth ef models namespace * [PM-1188] remove auth ef models unused imports * [PM-1188] fix imports for auth ef models * [PM-1188] fix more ef model imports * [PM-1188] fix file encodings
93 lines
2.9 KiB
C#
93 lines
2.9 KiB
C#
using Bit.Core.Auth.Enums;
|
|
using Bit.Core.Auth.Models;
|
|
using Bit.Core.Entities;
|
|
using Bit.Core.Services;
|
|
using Bit.Core.Utilities;
|
|
using Bit.Test.Common.AutoFixture;
|
|
using Bit.Test.Common.AutoFixture.Attributes;
|
|
using Microsoft.AspNetCore.Identity;
|
|
using Microsoft.Extensions.Logging;
|
|
using Microsoft.Extensions.Options;
|
|
using NSubstitute;
|
|
using Xunit;
|
|
|
|
namespace Bit.Core.Test.Auth.Identity;
|
|
|
|
[SutProviderCustomize]
|
|
public abstract class BaseTokenProviderTests<T>
|
|
where T : IUserTwoFactorTokenProvider<User>
|
|
{
|
|
public abstract TwoFactorProviderType TwoFactorProviderType { get; }
|
|
|
|
#region Helpers
|
|
protected static IEnumerable<object[]> SetupCanGenerateData(params (Dictionary<string, object> MetaData, bool ExpectedResponse)[] data)
|
|
{
|
|
return data.Select(d =>
|
|
new object[]
|
|
{
|
|
d.MetaData,
|
|
d.ExpectedResponse,
|
|
});
|
|
}
|
|
|
|
protected virtual IUserService AdditionalSetup(SutProvider<T> sutProvider, User user)
|
|
{
|
|
var userService = Substitute.For<IUserService>();
|
|
|
|
sutProvider.GetDependency<IServiceProvider>()
|
|
.GetService(typeof(IUserService))
|
|
.Returns(userService);
|
|
|
|
SetupUserService(userService, user);
|
|
|
|
return userService;
|
|
}
|
|
|
|
protected virtual void SetupUserService(IUserService userService, User user)
|
|
{
|
|
userService
|
|
.TwoFactorProviderIsEnabledAsync(TwoFactorProviderType, user)
|
|
.Returns(true);
|
|
}
|
|
|
|
protected static UserManager<User> SubstituteUserManager()
|
|
{
|
|
return new UserManager<User>(Substitute.For<IUserStore<User>>(),
|
|
Substitute.For<IOptions<IdentityOptions>>(),
|
|
Substitute.For<IPasswordHasher<User>>(),
|
|
Enumerable.Empty<IUserValidator<User>>(),
|
|
Enumerable.Empty<IPasswordValidator<User>>(),
|
|
Substitute.For<ILookupNormalizer>(),
|
|
Substitute.For<IdentityErrorDescriber>(),
|
|
Substitute.For<IServiceProvider>(),
|
|
Substitute.For<ILogger<UserManager<User>>>());
|
|
}
|
|
|
|
protected void MockDatabase(User user, Dictionary<string, object> metaData)
|
|
{
|
|
var providers = new Dictionary<TwoFactorProviderType, TwoFactorProvider>
|
|
{
|
|
[TwoFactorProviderType] = new TwoFactorProvider
|
|
{
|
|
Enabled = true,
|
|
MetaData = metaData,
|
|
},
|
|
};
|
|
|
|
user.TwoFactorProviders = JsonHelpers.LegacySerialize(providers);
|
|
}
|
|
#endregion
|
|
|
|
public virtual async Task RunCanGenerateTwoFactorTokenAsync(Dictionary<string, object> metaData, bool expectedResponse,
|
|
User user, SutProvider<T> sutProvider)
|
|
{
|
|
var userManager = SubstituteUserManager();
|
|
MockDatabase(user, metaData);
|
|
|
|
AdditionalSetup(sutProvider, user);
|
|
|
|
var response = await sutProvider.Sut.CanGenerateTwoFactorTokenAsync(userManager, user);
|
|
Assert.Equal(expectedResponse, response);
|
|
}
|
|
}
|