bitwarden-server/bitwarden_license/test/CmmCore.Test/AutoFixture/ProviderUserFixtures.cs

using System.Reflection;
using AutoFixture;
using AutoFixture.Xunit2;
using Bit.Core.Enums.Provider;

namespace Bit.CommCore.Test.AutoFixture.ProviderUserFixtures
{
    internal class ProviderUser : ICustomization
    {
        public ProviderUserStatusType Status { get; set; }
        public ProviderUserType Type { get; set; }

        public ProviderUser(ProviderUserStatusType status, ProviderUserType type)
        {
            Status = status;
            Type = type;
        }

        public void Customize(IFixture fixture)
        {
            fixture.Customize<Core.Models.Table.Provider.ProviderUser>(composer => composer
                .With(o => o.Type, Type)
                .With(o => o.Status, Status));
        }
    }

    public class ProviderUserAttribute : CustomizeAttribute
    {
        private readonly ProviderUserStatusType _status;
        private readonly ProviderUserType _type;

        public ProviderUserAttribute(
            ProviderUserStatusType status = ProviderUserStatusType.Confirmed,
            ProviderUserType type = ProviderUserType.ProviderAdmin)
        {
            _status = status;
            _type = type;
        }

        public override ICustomization GetCustomization(ParameterInfo parameter)
        {
            return new ProviderUser(_status, _type);
        }
    }
}