abp/modules/permission-management/test/Volo.Abp.PermissionManagement.Domain.Tests/Volo/Abp/PermissionManagement/PermissionTestBase.cs

using System;
using Volo.Abp.PermissionManagement.EntityFrameworkCore;

namespace Volo.Abp.PermissionManagement;

public abstract class PermissionTestBase : PermissionManagementTestBase<AbpPermissionManagementTestModule>
{
    protected virtual void UsingDbContext(Action<IPermissionManagementDbContext> action)
    {
        using (var dbContext = GetRequiredService<IPermissionManagementDbContext>())
        {
            action.Invoke(dbContext);
        }
    }

    protected virtual T UsingDbContext<T>(Func<IPermissionManagementDbContext, T> action)
    {
        using (var dbContext = GetRequiredService<IPermissionManagementDbContext>())
        {
            return action.Invoke(dbContext);
        }
    }
}