bitwarden-server/test/Infrastructure.EFIntegration.Test/AutoFixture/Relays/MaxLengthStringRelay.cs

using System.ComponentModel.DataAnnotations;
using System.Reflection;
using AutoFixture.Kernel;

namespace Bit.Infrastructure.EFIntegration.Test.AutoFixture.Relays;

// Creates a string the same length as any availible MaxLength data annotation
// Modified version of the StringLenfthRelay provided by AutoFixture
// https://github.com/AutoFixture/AutoFixture/blob/master/Src/AutoFixture/DataAnnotations/StringLengthAttributeRelay.cs
public class MaxLengthStringRelay : ISpecimenBuilder
{
    public object Create(object request, ISpecimenContext context)
    {
        if (request == null)
        {
            return new NoSpecimen();
        }

        if (context == null)
        {
            throw new ArgumentNullException(nameof(context));
        }

        var p = request as PropertyInfo;
        if (p == null)
        {
            return new NoSpecimen();
        }

        var a = (MaxLengthAttribute)p.GetCustomAttributes(typeof(MaxLengthAttribute), false).SingleOrDefault();

        if (a == null)
        {
            return new NoSpecimen();
        }

        return context.Resolve(new ConstrainedStringRequest(a.Length, a.Length));
    }
}