abp/modules/feature-management/test/Volo.Abp.FeatureManagement.TestBase/Volo/Abp/FeatureManagement/FeatureManagementTestDataBuilder.cs

using Volo.Abp.DependencyInjection;
using Volo.Abp.Features;
using Volo.Abp.Guids;

namespace Volo.Abp.FeatureManagement
{
    public class FeatureManagementTestDataBuilder : ITransientDependency
    {
        private readonly IFeatureValueRepository _featureValueRepository;
        private readonly IGuidGenerator _guidGenerator;
        private readonly FeatureManagementTestData _testData;

        public FeatureManagementTestDataBuilder(
            IGuidGenerator guidGenerator,
            FeatureManagementTestData testData, 
            IFeatureValueRepository featureValueRepository)
        {
            _guidGenerator = guidGenerator;
            _testData = testData;
            _featureValueRepository = featureValueRepository;
        }

        public void Build()
        {
            #region "Regular" edition features

            //SocialLogins
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.SocialLogins,
                    true.ToString().ToLowerInvariant(),
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Regular.ToString("N")
                )
            );

            //UserCount
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.UserCount,
                    "10",
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Regular.ToString("N")
                )
            );

            //ProjectCount
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.ProjectCount,
                    "1",
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Regular.ToString("N")
                )
            );

            #endregion

            #region "Enterprise" edition features

            //SocialLogins
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.SocialLogins,
                    true.ToString().ToLowerInvariant(),
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Enterprise.ToString("N")
                )
            );

            //EmailSupport
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.EmailSupport,
                    true.ToString().ToLowerInvariant(),
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Enterprise.ToString("N")
                )
            );

            //UserCount
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.UserCount,
                    "20",
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Enterprise.ToString("N")
                )
            );

            //ProjectCount
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.ProjectCount,
                    "3",
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Enterprise.ToString("N")
                )
            );

            //BackupCount
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.BackupCount,
                    "5",
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Enterprise.ToString("N")
                )
            );

            #endregion

            #region "Ultimate" edition features

            //SocialLogins
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.SocialLogins,
                    true.ToString().ToLowerInvariant(),
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Ultimate.ToString("N")
                )
            );

            //EmailSupport
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.EmailSupport,
                    true.ToString().ToLowerInvariant(),
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Ultimate.ToString("N")
                )
            );

            //EmailSupport
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.DailyAnalysis,
                    true.ToString().ToLowerInvariant(),
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Ultimate.ToString("N")
                )
            );

            //UserCount
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.UserCount,
                    "100",
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Ultimate.ToString("N")
                )
            );

            //ProjectCount
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.ProjectCount,
                    "10",
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Ultimate.ToString("N")
                )
            );

            //BackupCount
            _featureValueRepository.Insert(
                new FeatureValue(
                    _guidGenerator.Create(),
                    TestFeatureDefinitionProvider.BackupCount,
                    "10",
                    EditionFeatureValueProvider.ProviderName,
                    TestEditionIds.Ultimate.ToString("N")
                )
            );

            #endregion
        }
    }
}