103 lines
5.1 KiB
C#
103 lines
5.1 KiB
C#
// Copyright (c) Microsoft Open Technologies, Inc. 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.Threading.Tasks;
|
|
using Microsoft.AspNet.Identity.Test;
|
|
using Microsoft.Data.Entity;
|
|
using Microsoft.Data.Entity.InMemory;
|
|
using Microsoft.Framework.DependencyInjection;
|
|
using Microsoft.Framework.DependencyInjection.Fallback;
|
|
using Xunit;
|
|
|
|
namespace Microsoft.AspNet.Identity.Entity.Test
|
|
{
|
|
public class RoleStoreTest
|
|
{
|
|
[Fact]
|
|
public async Task CanCreateUsingAddRoleManager()
|
|
{
|
|
var services = new ServiceCollection();
|
|
services.AddEntityFramework().AddInMemoryStore();
|
|
// TODO: this should construct a new instance of InMemoryStore
|
|
var store = new EntityRoleStore<EntityRole>(new IdentityContext());
|
|
services.AddIdentity<EntityUser, EntityRole>(s =>
|
|
{
|
|
s.AddRoleStore(() => store);
|
|
});
|
|
|
|
var provider = services.BuildServiceProvider();
|
|
var manager = provider.GetService<RoleManager<EntityRole>>();
|
|
Assert.NotNull(manager);
|
|
IdentityResultAssert.IsSuccess(await manager.CreateAsync(new EntityRole("arole")));
|
|
}
|
|
[Fact]
|
|
public async Task CanCreateRoleWithSingletonManager()
|
|
{
|
|
var services = new ServiceCollection();
|
|
services.AddEntityFramework().AddInMemoryStore();
|
|
services.AddTransient<IdentityContext>();
|
|
services.AddTransient<IRoleStore<EntityRole>, EntityRoleStore<EntityRole>>();
|
|
services.AddSingleton<RoleManager<EntityRole>>();
|
|
var provider = services.BuildServiceProvider();
|
|
var manager = provider.GetService<RoleManager<EntityRole>>();
|
|
Assert.NotNull(manager);
|
|
IdentityResultAssert.IsSuccess(await manager.CreateAsync(new EntityRole("someRole")));
|
|
}
|
|
|
|
[Fact]
|
|
public async Task RoleStoreMethodsThrowWhenDisposedTest()
|
|
{
|
|
var store = new EntityRoleStore<EntityRole>(new IdentityContext());
|
|
store.Dispose();
|
|
await Assert.ThrowsAsync<ObjectDisposedException>(async () => await store.FindByIdAsync(null));
|
|
await Assert.ThrowsAsync<ObjectDisposedException>(async () => await store.FindByNameAsync(null));
|
|
await Assert.ThrowsAsync<ObjectDisposedException>(async () => await store.GetRoleIdAsync(null));
|
|
await Assert.ThrowsAsync<ObjectDisposedException>(async () => await store.GetRoleNameAsync(null));
|
|
await Assert.ThrowsAsync<ObjectDisposedException>(async () => await store.SetRoleNameAsync(null, null));
|
|
await Assert.ThrowsAsync<ObjectDisposedException>(async () => await store.CreateAsync(null));
|
|
await Assert.ThrowsAsync<ObjectDisposedException>(async () => await store.UpdateAsync(null));
|
|
await Assert.ThrowsAsync<ObjectDisposedException>(async () => await store.DeleteAsync(null));
|
|
}
|
|
|
|
[Fact]
|
|
public async Task RoleStorePublicNullCheckTest()
|
|
{
|
|
Assert.Throws<ArgumentNullException>("context", () => new EntityRoleStore<EntityRole>(null));
|
|
var store = new EntityRoleStore<EntityRole>(new IdentityContext());
|
|
await Assert.ThrowsAsync<ArgumentNullException>("role", async () => await store.GetRoleIdAsync(null));
|
|
await Assert.ThrowsAsync<ArgumentNullException>("role", async () => await store.GetRoleNameAsync(null));
|
|
await Assert.ThrowsAsync<ArgumentNullException>("role", async () => await store.SetRoleNameAsync(null, null));
|
|
await Assert.ThrowsAsync<ArgumentNullException>("role", async () => await store.CreateAsync(null));
|
|
await Assert.ThrowsAsync<ArgumentNullException>("role", async () => await store.UpdateAsync(null));
|
|
await Assert.ThrowsAsync<ArgumentNullException>("role", async () => await store.DeleteAsync(null));
|
|
}
|
|
|
|
[Fact]
|
|
public async Task CanUpdateRoleName()
|
|
{
|
|
var manager = TestIdentityFactory.CreateRoleManager();
|
|
var role = new EntityRole("UpdateRoleName");
|
|
IdentityResultAssert.IsSuccess(await manager.CreateAsync(role));
|
|
Assert.Null(await manager.FindByNameAsync("New"));
|
|
role.Name = "New";
|
|
IdentityResultAssert.IsSuccess(await manager.UpdateAsync(role));
|
|
Assert.NotNull(await manager.FindByNameAsync("New"));
|
|
Assert.Null(await manager.FindByNameAsync("UpdateAsync"));
|
|
}
|
|
|
|
[Fact]
|
|
public async Task CanSetUserName()
|
|
{
|
|
var manager = TestIdentityFactory.CreateRoleManager();
|
|
var role = new EntityRole("UpdateRoleName");
|
|
IdentityResultAssert.IsSuccess(await manager.CreateAsync(role));
|
|
Assert.Null(await manager.FindByNameAsync("New"));
|
|
IdentityResultAssert.IsSuccess(await manager.SetRoleNameAsync(role, "New"));
|
|
Assert.NotNull(await manager.FindByNameAsync("New"));
|
|
Assert.Null(await manager.FindByNameAsync("UpdateAsync"));
|
|
}
|
|
|
|
}
|
|
}
|