101 lines
4.5 KiB
C#
101 lines
4.5 KiB
C#
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.Builder;
|
|
using Microsoft.Extensions.Logging;
|
|
using Microsoft.Extensions.Options;
|
|
using Moq;
|
|
|
|
namespace Microsoft.AspNetCore.Identity.Test
|
|
{
|
|
public static class MockHelpers
|
|
{
|
|
public static StringBuilder LogMessage = new StringBuilder();
|
|
|
|
public static Mock<UserManager<TUser>> MockUserManager<TUser>() where TUser : class
|
|
{
|
|
var store = new Mock<IUserStore<TUser>>();
|
|
var mgr = new Mock<UserManager<TUser>>(store.Object, null, null, null, null, null, null, null, null);
|
|
mgr.Object.UserValidators.Add(new UserValidator<TUser>());
|
|
mgr.Object.PasswordValidators.Add(new PasswordValidator<TUser>());
|
|
return mgr;
|
|
}
|
|
|
|
public static Mock<RoleManager<TRole>> MockRoleManager<TRole>(IRoleStore<TRole> store = null) where TRole : class
|
|
{
|
|
store = store ?? new Mock<IRoleStore<TRole>>().Object;
|
|
var roles = new List<IRoleValidator<TRole>>();
|
|
roles.Add(new RoleValidator<TRole>());
|
|
return new Mock<RoleManager<TRole>>(store, roles, new UpperInvariantLookupNormalizer(),
|
|
new IdentityErrorDescriber(), null);
|
|
}
|
|
|
|
public static Mock<ILogger<T>> MockILogger<T>(StringBuilder logStore = null) where T : class
|
|
{
|
|
logStore = logStore ?? LogMessage;
|
|
var logger = new Mock<ILogger<T>>();
|
|
logger.Setup(x => x.Log(It.IsAny<LogLevel>(), It.IsAny<EventId>(), It.IsAny<object>(),
|
|
It.IsAny<Exception>(), It.IsAny<Func<object, Exception, string>>()))
|
|
.Callback((LogLevel logLevel, EventId eventId, object state, Exception exception, Func<object, Exception, string> formatter) =>
|
|
{
|
|
if (formatter == null)
|
|
{
|
|
logStore.Append(state.ToString());
|
|
}
|
|
else
|
|
{
|
|
logStore.Append(formatter(state, exception));
|
|
}
|
|
logStore.Append(" ");
|
|
});
|
|
logger.Setup(x => x.BeginScope(It.IsAny<object>())).Callback((object state) =>
|
|
{
|
|
logStore.Append(state.ToString());
|
|
logStore.Append(" ");
|
|
});
|
|
logger.Setup(x => x.IsEnabled(LogLevel.Debug)).Returns(true);
|
|
logger.Setup(x => x.IsEnabled(LogLevel.Warning)).Returns(true);
|
|
|
|
return logger;
|
|
}
|
|
|
|
public static UserManager<TUser> TestUserManager<TUser>(IUserStore<TUser> store = null) where TUser : class
|
|
{
|
|
store = store ?? new Mock<IUserStore<TUser>>().Object;
|
|
var options = new Mock<IOptions<IdentityOptions>>();
|
|
var idOptions = new IdentityOptions();
|
|
idOptions.Lockout.AllowedForNewUsers = false;
|
|
options.Setup(o => o.Value).Returns(idOptions);
|
|
var userValidators = new List<IUserValidator<TUser>>();
|
|
var validator = new Mock<IUserValidator<TUser>>();
|
|
userValidators.Add(validator.Object);
|
|
var pwdValidators = new List<PasswordValidator<TUser>>();
|
|
pwdValidators.Add(new PasswordValidator<TUser>());
|
|
var userManager = new UserManager<TUser>(store, options.Object, new PasswordHasher<TUser>(),
|
|
userValidators, pwdValidators, new UpperInvariantLookupNormalizer(),
|
|
new IdentityErrorDescriber(), null,
|
|
new Mock<ILogger<UserManager<TUser>>>().Object);
|
|
validator.Setup(v => v.ValidateAsync(userManager, It.IsAny<TUser>()))
|
|
.Returns(Task.FromResult(IdentityResult.Success)).Verifiable();
|
|
return userManager;
|
|
}
|
|
|
|
public static RoleManager<TRole> TestRoleManager<TRole>(IRoleStore<TRole> store = null) where TRole : class
|
|
{
|
|
store = store ?? new Mock<IRoleStore<TRole>>().Object;
|
|
var roles = new List<IRoleValidator<TRole>>();
|
|
roles.Add(new RoleValidator<TRole>());
|
|
return new AspNetRoleManager<TRole>(store, roles,
|
|
new UpperInvariantLookupNormalizer(),
|
|
new IdentityErrorDescriber(),
|
|
null,
|
|
null);
|
|
}
|
|
|
|
}
|
|
}
|