bitwarden-server/test/Common/AutoFixture/Attributes/EnvironmentDataAttribute.cs

using System.Reflection;
using Xunit.Sdk;

namespace Bit.Test.Common.AutoFixture.Attributes;

/// <summary>
/// Used for collecting data from environment useful for when we want to test an integration with another service and
/// it might require an api key or other piece of sensitive data that we don't want slipping into the wrong hands.
/// </summary>
/// <remarks>
/// It probably should be refactored to support fixtures and other customization so it can more easily be used in conjunction
/// with more parameters.  Currently it attempt to match environment variable names to values of the parameter type in that positions.
/// It will start from the first parameter and go for each supplied name.
/// </remarks>
public class EnvironmentDataAttribute : DataAttribute
{
    private readonly string[] _environmentVariableNames;

    public EnvironmentDataAttribute(params string[] environmentVariableNames)
    {
        _environmentVariableNames = environmentVariableNames;
    }

    public override IEnumerable<object[]> GetData(MethodInfo testMethod)
    {
        var methodParameters = testMethod.GetParameters();

        if (methodParameters.Length < _environmentVariableNames.Length)
        {
            throw new ArgumentException($"The target test method only has {methodParameters.Length} arguments but you supplied {_environmentVariableNames.Length}");
        }

        var values = new object[_environmentVariableNames.Length];

        for (var i = 0; i < _environmentVariableNames.Length; i++)
        {
            values[i] = Convert.ChangeType(Environment.GetEnvironmentVariable(_environmentVariableNames[i]), methodParameters[i].ParameterType);
        }

        return new[] { values };
    }
}