public async Task ShouldReturnAllUsersWithAdminRole()
{
DocumentDbIdentityRole adminRole = DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName();
DocumentDbUserStore <DocumentDbIdentityUser> store = CreateUserStore();
DocumentDbIdentityUser <DocumentDbIdentityRole> firstAdmin = DocumentDbIdentityUserBuilder.Create().WithId().WithNormalizedUserName().AddRole(adminRole).AddRole();
DocumentDbIdentityUser <DocumentDbIdentityRole> secondAdmin = DocumentDbIdentityUserBuilder.Create().WithId().WithNormalizedUserName().AddRole(adminRole).AddRole().AddRole();
DocumentDbIdentityUser <DocumentDbIdentityRole> thirdAdmin = DocumentDbIdentityUserBuilder.Create().WithId().WithNormalizedUserName().AddRole(adminRole);
CreateDocument(firstAdmin);
CreateDocument(secondAdmin);
CreateDocument(DocumentDbIdentityUserBuilder.Create().AddRole().AddRole());
CreateDocument(DocumentDbIdentityUserBuilder.Create().AddRole().AddRole().AddRole());
CreateDocument(thirdAdmin);
CreateDocument(DocumentDbIdentityUserBuilder.Create());
CreateDocument(DocumentDbIdentityUserBuilder.Create().AddRole());
IList <DocumentDbIdentityUser> adminUsers = await store.GetUsersInRoleAsync(adminRole.NormalizedName, CancellationToken.None);
Assert.Collection(
adminUsers,
u => u.Id.Equals(firstAdmin.Id),
u => u.Id.Equals(secondAdmin.Id),
u => u.Id.Equals(thirdAdmin.Id));
}
async Task AddFacultyRoleAsync()
{
DocumentDbIdentityRole role = new DocumentDbIdentityRole()
{
Name = config["DataStore:Containers:Identities:Roles:Faculty:Name"]
};
await roleManager.CreateAsync(role);
}
public async Task ShouldReturnNormalizedNameOfRole()
{
DocumentDbRoleStore <DocumentDbIdentityRole> store = CreateRoleStore();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithId().WithNormalizedRoleName();
string result = await store.GetNormalizedRoleNameAsync(targetRole, CancellationToken.None);
Assert.Equal(targetRole.NormalizedName, result);
}
public async Task ShouldThrowExceptionOnAddingUserToNonexistantRole()
{
DocumentDbUserStore <DocumentDbIdentityUser <DocumentDbIdentityRole>, DocumentDbIdentityRole> userStore = CreateUserStore();
DocumentDbIdentityUser <DocumentDbIdentityRole> targetUser = DocumentDbIdentityUserBuilder.Create();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithId().WithNormalizedRoleName();
string roleName = Guid.NewGuid().ToString();
// Add the created sample data role to the user
await Assert.ThrowsAsync(typeof(ArgumentException), async() => await userStore.AddToRoleAsync(targetUser, targetRole.Name, CancellationToken.None));
}
public async Task ShouldReturnUserIsInRole()
{
DocumentDbUserStore <DocumentDbIdentityUser> userStore = CreateUserStore();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName();
DocumentDbIdentityUser targetUser = DocumentDbIdentityUserBuilder.Create().AddRole(targetRole).AddRole();
bool isInRole = await userStore.IsInRoleAsync(targetUser, targetRole.NormalizedName, CancellationToken.None);
Assert.True(isInRole);
}
public async Task ShouldSetNewNormalizedRoleName()
{
DocumentDbRoleStore <DocumentDbIdentityRole> store = CreateRoleStore();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithId().WithNormalizedRoleName();
string newNormalizedRoleName = Guid.NewGuid().ToString();
await store.SetNormalizedRoleNameAsync(targetRole, newNormalizedRoleName, CancellationToken.None);
Assert.Equal(targetRole.NormalizedName, newNormalizedRoleName);
}
public async Task ShouldRemoveClaimFromRole()
{
Claim claimToRemove = new Claim(Guid.NewGuid().ToString(), Guid.NewGuid().ToString());
DocumentDbRoleStore <DocumentDbIdentityRole> store = CreateRoleStore();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithId().AddClaim().AddClaim(claimToRemove).AddClaim();
await store.RemoveClaimAsync(targetRole, claimToRemove, CancellationToken.None);
Assert.DoesNotContain(targetRole.Claims, c => c.Type.Equals(claimToRemove.Type));
}
public async Task ShouldReturnUserIsInRole()
{
DocumentDbIdentityRole role = DocumentDbIdentityRoleBuilder.Create();
DocumentDbUserStore <DocumentDbIdentityUser <DocumentDbIdentityRole>, DocumentDbIdentityRole> store = CreateUserStore();
DocumentDbIdentityUser <DocumentDbIdentityRole> user = DocumentDbIdentityUserBuilder.Create().WithId().AddRole(role).AddRole().AddRole();
bool result = await store.IsInRoleAsync(user, role.Name, CancellationToken.None);
Assert.True(result);
}
public async Task ShouldAddClaimToRole()
{
Claim newClaim = new Claim(Guid.NewGuid().ToString(), Guid.NewGuid().ToString());
DocumentDbRoleStore <DocumentDbIdentityRole> store = CreateRoleStore();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithId();
await store.AddClaimAsync(targetRole, newClaim);
Assert.Contains(targetRole.Claims, c => c.Type.Equals(newClaim.Type));
}
public async Task ShouldReturnUserIsNotInRoleWhenPassingNotNormalizedRoleNameToIsInRole()
{
DocumentDbUserStore <DocumentDbIdentityUser> userStore = CreateUserStore();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName();
DocumentDbIdentityUser targetUser = DocumentDbIdentityUserBuilder.Create().AddRole(targetRole).AddRole();
// Pass not normalized name which should lead to not locating the target role
bool isInRole = await userStore.IsInRoleAsync(targetUser, targetRole.Name, CancellationToken.None);
Assert.False(isInRole);
}
public async Task ShouldThrowExceptionWhenPassingNotNormalizedNameToAddToRole()
{
DocumentDbUserStore <DocumentDbIdentityUser> userStore = CreateUserStore();
DocumentDbRoleStore <DocumentDbIdentityRole> roleStore = CreateRoleStore();
DocumentDbIdentityUser targetUser = DocumentDbIdentityUserBuilder.Create();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithId().WithNormalizedRoleName();
// Create sample data role
await roleStore.CreateAsync(targetRole, CancellationToken.None);
// Add the user to the created role, but pass the not normalized name, expecting an exception
await Assert.ThrowsAsync(typeof(ArgumentException), async() => await userStore.AddToRoleAsync(targetUser, targetRole.Name, CancellationToken.None));
}
public async Task ShouldThrowExceptionOnAddingUserToNonexistantRole()
{
DocumentDbUserStore <DocumentDbIdentityUser> userStore = CreateUserStore();
DocumentDbRoleStore <DocumentDbIdentityRole> roleStore = CreateRoleStore();
DocumentDbIdentityUser targetUser = DocumentDbIdentityUserBuilder.Create();
DocumentDbIdentityRole someNotTargetedRole = DocumentDbIdentityRoleBuilder.Create().WithId().WithNormalizedRoleName();
// Create a role so there is a differently named role in the store
await roleStore.CreateAsync(someNotTargetedRole, CancellationToken.None);
// Add the user to a role name different than the role created before, expecting an exception
await Assert.ThrowsAsync(typeof(ArgumentException), async() => await userStore.AddToRoleAsync(targetUser, "NotExistantRole", CancellationToken.None));
}
public async Task ShouldReturnNoUsersWithAdminRoleWhenPassingNotNormalizedRoleNameToGetUsersInRole()
{
DocumentDbIdentityRole adminRole = DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName();
DocumentDbUserStore <DocumentDbIdentityUser> store = CreateUserStore();
DocumentDbIdentityUser <DocumentDbIdentityRole> firstAdmin = DocumentDbIdentityUserBuilder.Create().WithId().WithNormalizedUserName().AddRole(adminRole).AddRole();
CreateDocument(firstAdmin);
CreateDocument(DocumentDbIdentityUserBuilder.Create().AddRole().AddRole());
IList <DocumentDbIdentityUser> adminUsers = await store.GetUsersInRoleAsync(adminRole.Name, CancellationToken.None);
Assert.Empty(adminUsers);
}
public async Task ShouldRemoveRoleFromUser()
{
DocumentDbUserStore <DocumentDbIdentityUser> userStore = CreateUserStore();
DocumentDbIdentityRole firstRole = DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName();
DocumentDbIdentityRole secondRole = DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName();
DocumentDbIdentityUser targetUser = DocumentDbIdentityUserBuilder.Create().AddRole(firstRole).AddRole(secondRole);
// Remove the second role
await userStore.RemoveFromRoleAsync(targetUser, secondRole.NormalizedName, CancellationToken.None);
// Assert second role has been removed while first one is still there
Assert.DoesNotContain(targetUser.Roles, r => r.NormalizedName.Equals(secondRole.NormalizedName));
Assert.Contains(targetUser.Roles, r => r.NormalizedName.Equals(firstRole.NormalizedName));
}
public DocumentDbIdentityUserBuilder AddRole(DocumentDbIdentityRole role = null)
{
if (role == null)
{
role = new DocumentDbIdentityRole()
{
Id = Guid.NewGuid().ToString(),
Name = Guid.NewGuid().ToString()
};
}
identityUser.Roles.Add(role);
return(this);
}
public async Task ShouldCreateNewRoleInDatabase()
{
DocumentDbIdentityRole newRole = DocumentDbIdentityRoleBuilder.Create().WithId();
DocumentDbRoleStore <DocumentDbIdentityRole> store = CreateRoleStore();
// Create the new role
IdentityResult result = store.CreateAsync(newRole, CancellationToken.None).Result;
// Get it again from the DB to check if it was created correctly
DocumentDbIdentityRole queriedRole = await store.FindByIdAsync(newRole.Id, CancellationToken.None);
Assert.True(result.Succeeded);
Assert.Equal(queriedRole, newRole, new DocumentDbIdentityRoleComparer());
}
public async Task ShouldDeleteRoleFromDb()
{
DocumentDbRoleStore <DocumentDbIdentityRole> store = CreateRoleStore();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithId();
// Create sample data in DB
CreateDocument(DocumentDbIdentityRoleBuilder.Create());
CreateDocument(DocumentDbIdentityRoleBuilder.Create());
CreateDocument(targetRole);
CreateDocument(DocumentDbIdentityRoleBuilder.Create());
IdentityResult result = await store.DeleteAsync(targetRole, CancellationToken.None);
Assert.True(result.Succeeded);
}
public async Task ShouldReturnRoleByName()
{
DocumentDbRoleStore <DocumentDbIdentityRole> store = CreateRoleStore();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithId().WithNormalizedRoleName();
// Create sample data in DB
CreateDocument(DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName());
CreateDocument(DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName());
CreateDocument(targetRole);
CreateDocument(DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName());
DocumentDbIdentityRole queriedRole = await store.FindByNameAsync(targetRole.NormalizedName, CancellationToken.None);
Assert.Equal(targetRole.Id, queriedRole.Id);
}
public async Task ShouldAddUserToRole()
{
DocumentDbUserStore <DocumentDbIdentityUser> userStore = CreateUserStore();
DocumentDbRoleStore <DocumentDbIdentityRole> roleStore = CreateRoleStore();
DocumentDbIdentityUser targetUser = DocumentDbIdentityUserBuilder.Create();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create("RoleName").WithId().WithNormalizedRoleName();
// Create sample data role
await roleStore.CreateAsync(targetRole, CancellationToken.None);
// Add the created sample data role to the user
await userStore.AddToRoleAsync(targetUser, targetRole.NormalizedName, CancellationToken.None);
Assert.Contains(targetUser.Roles, r => r.NormalizedName.Equals(targetRole.NormalizedName));
}
public async Task ShouldNotRemoveRoleFromUserWhenPassingNotNormalizedRoleNameToRemoveFromRole()
{
DocumentDbUserStore <DocumentDbIdentityUser> userStore = CreateUserStore();
DocumentDbIdentityRole firstRole = DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName();
DocumentDbIdentityRole secondRole = DocumentDbIdentityRoleBuilder.Create().WithNormalizedRoleName();
DocumentDbIdentityUser targetUser = DocumentDbIdentityUserBuilder.Create().AddRole(firstRole).AddRole(secondRole);
// Try remove the second role with a not normalized role name
await userStore.RemoveFromRoleAsync(targetUser, secondRole.Name, CancellationToken.None);
// Assert both roles are still here, as lookup without normalized name should have failed
Assert.Collection(targetUser.Roles,
r => r.NormalizedName.Equals(firstRole.NormalizedName),
r => r.NormalizedName.Equals(secondRole.NormalizedName));
}
public async Task ShouldReturnQueriedClaimFromRole()
{
string firstClaimType = Guid.NewGuid().ToString();
string secondClaimType = Guid.NewGuid().ToString();
string thirdClaimType = Guid.NewGuid().ToString();
DocumentDbRoleStore <DocumentDbIdentityRole> store = CreateRoleStore();
DocumentDbIdentityRole targetRole = DocumentDbIdentityRoleBuilder.Create().WithId().AddClaim(firstClaimType).AddClaim(secondClaimType).AddClaim(thirdClaimType);
IList <Claim> returnedClaims = await store.GetClaimsAsync(targetRole, CancellationToken.None);
Assert.Collection(
returnedClaims,
c => c.Type.Equals(firstClaimType),
c => c.Type.Equals(secondClaimType),
c => c.Type.Equals(thirdClaimType));
}
public async Task ShouldUpdateExistingRoleInDatabase()
{
DocumentDbRoleStore <DocumentDbIdentityRole> store = CreateRoleStore();
DocumentDbIdentityRole existingRole = DocumentDbIdentityRoleBuilder.Create().WithId();
// Create sample data in DB
CreateDocument(existingRole);
// Change property to upate on sample data and call the update mehtod
existingRole.Name = Guid.NewGuid().ToString();
IdentityResult result = await store.UpdateAsync(existingRole, CancellationToken.None);
// Get it again from the DB to check if it was created correctly
DocumentDbIdentityRole queriedRole = await store.FindByIdAsync(existingRole.Id, CancellationToken.None);
Assert.True(result.Succeeded);
Assert.Equal(existingRole, queriedRole, new DocumentDbIdentityRoleComparer());
}
public DocumentDbIdentityUserBuilder AddRole(DocumentDbIdentityRole role = null)
{
if (role == null)
{
LookupNormalizer normalizer = new LookupNormalizer();
string newRoleName = Guid.NewGuid().ToString().ToUpper();
role = new DocumentDbIdentityRole()
{
Id = Guid.NewGuid().ToString().ToUpper(),
Name = newRoleName,
NormalizedName = normalizer.Normalize(newRoleName)
};
}
else
{
identityUser.Roles.Add(role);
}
return(this);
}
public DocumentDbIdentityRoleBuilder(DocumentDbIdentityRole identityRole)
{
this.identityRole = identityRole;
}