abp/modules/permission-management/test/Volo.Abp.PermissionManagement.TestBase/Volo/Abp/PermissionManagement/PermissionTestDataBuilder.cs

using System;
using System.Threading.Tasks;
using Volo.Abp.Authorization.Permissions;
using Volo.Abp.DependencyInjection;
using Volo.Abp.Guids;

namespace Volo.Abp.PermissionManagement
{
    public class PermissionTestDataBuilder : ITransientDependency
    {
        public static Guid User1Id { get; } = Guid.NewGuid();
        public static Guid User2Id { get; } = Guid.NewGuid();

        private readonly IPermissionGrantRepository _permissionGrantRepository;
        private readonly IGuidGenerator _guidGenerator;

        public PermissionTestDataBuilder(IGuidGenerator guidGenerator, IPermissionGrantRepository permissionGrantRepository)
        {
            _guidGenerator = guidGenerator;
            _permissionGrantRepository = permissionGrantRepository;
        }

        public async Task BuildAsync()
        {
            await _permissionGrantRepository.InsertAsync(
                new PermissionGrant(
                    _guidGenerator.Create(),
                    "MyPermission1",
                    UserPermissionValueProvider.ProviderName,
                    User1Id.ToString()
                )
            );

            await _permissionGrantRepository.InsertAsync(
                new PermissionGrant(
                    _guidGenerator.Create(),
                    "MyDisabledPermission1",
                    UserPermissionValueProvider.ProviderName,
                    User1Id.ToString()
                )
            );

            await _permissionGrantRepository.InsertAsync(
                new PermissionGrant(
                    _guidGenerator.Create(),
                    "MyPermission3",
                    UserPermissionValueProvider.ProviderName,
                    User1Id.ToString()
                )
            );
        }
    }
}