You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
210 lines
6.7 KiB
210 lines
6.7 KiB
using System;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Abp;
|
|
using Abp.Authorization.Users;
|
|
using Abp.Events.Bus;
|
|
using Abp.Events.Bus.Entities;
|
|
using Abp.MultiTenancy;
|
|
using Abp.Runtime.Session;
|
|
using Abp.TestBase;
|
|
using MeetingSchedule.Authorization.Users;
|
|
using MeetingSchedule.EntityFrameworkCore;
|
|
using MeetingSchedule.EntityFrameworkCore.Seed.Host;
|
|
using MeetingSchedule.EntityFrameworkCore.Seed.Tenants;
|
|
using MeetingSchedule.MultiTenancy;
|
|
|
|
namespace MeetingSchedule.Tests
|
|
{
|
|
public abstract class MeetingScheduleTestBase : AbpIntegratedTestBase<MeetingScheduleTestModule>
|
|
{
|
|
protected MeetingScheduleTestBase()
|
|
{
|
|
void NormalizeDbContext(MeetingScheduleDbContext context)
|
|
{
|
|
context.EntityChangeEventHelper = NullEntityChangeEventHelper.Instance;
|
|
context.EventBus = NullEventBus.Instance;
|
|
context.SuppressAutoSetTenantId = true;
|
|
}
|
|
|
|
// Seed initial data for host
|
|
AbpSession.TenantId = null;
|
|
UsingDbContext(context =>
|
|
{
|
|
NormalizeDbContext(context);
|
|
new InitialHostDbBuilder(context).Create();
|
|
new DefaultTenantBuilder(context).Create();
|
|
});
|
|
|
|
// Seed initial data for default tenant
|
|
AbpSession.TenantId = 1;
|
|
UsingDbContext(context =>
|
|
{
|
|
NormalizeDbContext(context);
|
|
new TenantRoleAndUserBuilder(context, 1).Create();
|
|
});
|
|
|
|
LoginAsDefaultTenantAdmin();
|
|
}
|
|
|
|
#region UsingDbContext
|
|
|
|
protected IDisposable UsingTenantId(int? tenantId)
|
|
{
|
|
var previousTenantId = AbpSession.TenantId;
|
|
AbpSession.TenantId = tenantId;
|
|
return new DisposeAction(() => AbpSession.TenantId = previousTenantId);
|
|
}
|
|
|
|
protected void UsingDbContext(Action<MeetingScheduleDbContext> action)
|
|
{
|
|
UsingDbContext(AbpSession.TenantId, action);
|
|
}
|
|
|
|
protected Task UsingDbContextAsync(Func<MeetingScheduleDbContext, Task> action)
|
|
{
|
|
return UsingDbContextAsync(AbpSession.TenantId, action);
|
|
}
|
|
|
|
protected T UsingDbContext<T>(Func<MeetingScheduleDbContext, T> func)
|
|
{
|
|
return UsingDbContext(AbpSession.TenantId, func);
|
|
}
|
|
|
|
protected Task<T> UsingDbContextAsync<T>(Func<MeetingScheduleDbContext, Task<T>> func)
|
|
{
|
|
return UsingDbContextAsync(AbpSession.TenantId, func);
|
|
}
|
|
|
|
protected void UsingDbContext(int? tenantId, Action<MeetingScheduleDbContext> action)
|
|
{
|
|
using (UsingTenantId(tenantId))
|
|
{
|
|
using (var context = LocalIocManager.Resolve<MeetingScheduleDbContext>())
|
|
{
|
|
action(context);
|
|
context.SaveChanges();
|
|
}
|
|
}
|
|
}
|
|
|
|
protected async Task UsingDbContextAsync(int? tenantId, Func<MeetingScheduleDbContext, Task> action)
|
|
{
|
|
using (UsingTenantId(tenantId))
|
|
{
|
|
using (var context = LocalIocManager.Resolve<MeetingScheduleDbContext>())
|
|
{
|
|
await action(context);
|
|
await context.SaveChangesAsync();
|
|
}
|
|
}
|
|
}
|
|
|
|
protected T UsingDbContext<T>(int? tenantId, Func<MeetingScheduleDbContext, T> func)
|
|
{
|
|
T result;
|
|
|
|
using (UsingTenantId(tenantId))
|
|
{
|
|
using (var context = LocalIocManager.Resolve<MeetingScheduleDbContext>())
|
|
{
|
|
result = func(context);
|
|
context.SaveChanges();
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
protected async Task<T> UsingDbContextAsync<T>(int? tenantId, Func<MeetingScheduleDbContext, Task<T>> func)
|
|
{
|
|
T result;
|
|
|
|
using (UsingTenantId(tenantId))
|
|
{
|
|
using (var context = LocalIocManager.Resolve<MeetingScheduleDbContext>())
|
|
{
|
|
result = await func(context);
|
|
await context.SaveChangesAsync();
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Login
|
|
|
|
protected void LoginAsHostAdmin()
|
|
{
|
|
LoginAsHost(AbpUserBase.AdminUserName);
|
|
}
|
|
|
|
protected void LoginAsDefaultTenantAdmin()
|
|
{
|
|
LoginAsTenant(AbpTenantBase.DefaultTenantName, AbpUserBase.AdminUserName);
|
|
}
|
|
|
|
protected void LoginAsHost(string userName)
|
|
{
|
|
AbpSession.TenantId = null;
|
|
|
|
var user =
|
|
UsingDbContext(
|
|
context =>
|
|
context.Users.FirstOrDefault(u => u.TenantId == AbpSession.TenantId && u.UserName == userName));
|
|
if (user == null)
|
|
{
|
|
throw new Exception("There is no user: " + userName + " for host.");
|
|
}
|
|
|
|
AbpSession.UserId = user.Id;
|
|
}
|
|
|
|
protected void LoginAsTenant(string tenancyName, string userName)
|
|
{
|
|
var tenant = UsingDbContext(context => context.Tenants.FirstOrDefault(t => t.TenancyName == tenancyName));
|
|
if (tenant == null)
|
|
{
|
|
throw new Exception("There is no tenant: " + tenancyName);
|
|
}
|
|
|
|
AbpSession.TenantId = tenant.Id;
|
|
|
|
var user =
|
|
UsingDbContext(
|
|
context =>
|
|
context.Users.FirstOrDefault(u => u.TenantId == AbpSession.TenantId && u.UserName == userName));
|
|
if (user == null)
|
|
{
|
|
throw new Exception("There is no user: " + userName + " for tenant: " + tenancyName);
|
|
}
|
|
|
|
AbpSession.UserId = user.Id;
|
|
}
|
|
|
|
#endregion
|
|
|
|
/// <summary>
|
|
/// Gets current user if <see cref="IAbpSession.UserId"/> is not null.
|
|
/// Throws exception if it's null.
|
|
/// </summary>
|
|
protected async Task<User> GetCurrentUserAsync()
|
|
{
|
|
var userId = AbpSession.GetUserId();
|
|
return await UsingDbContext(context => context.Users.SingleAsync(u => u.Id == userId));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets current tenant if <see cref="IAbpSession.TenantId"/> is not null.
|
|
/// Throws exception if there is no current tenant.
|
|
/// </summary>
|
|
protected async Task<Tenant> GetCurrentTenantAsync()
|
|
{
|
|
var tenantId = AbpSession.GetTenantId();
|
|
return await UsingDbContext(context => context.Tenants.SingleAsync(t => t.Id == tenantId));
|
|
}
|
|
}
|
|
}
|