recyclarr/src/Recyclarr.TrashLib.Tests/Config/Parsing/ConfigurationLoaderEnvVarTest.cs

using FluentAssertions;
using NSubstitute;
using NUnit.Framework;
using Recyclarr.Cli.TestLibrary;
using Recyclarr.Common;
using Recyclarr.TrashLib.Config;
using Recyclarr.TrashLib.Config.Parsing;
using Recyclarr.TrashLib.Services.Sonarr.Config;

namespace Recyclarr.TrashLib.Tests.Config.Parsing;

[TestFixture]
[Parallelizable(ParallelScope.All)]
public class ConfigurationLoaderEnvVarTest : IntegrationFixture
{
    [Test]
    public void Test_successful_environment_variable_loading()
    {
        var env = Resolve<IEnvironment>();
        env.GetEnvironmentVariable("SONARR_API_KEY").Returns("the_api_key");
        env.GetEnvironmentVariable("SONARR_URL").Returns("the_url");

        var sut = Resolve<ConfigurationLoader>();

        const string testYml = @"
sonarr:
  instance:
    api_key: !env_var SONARR_API_KEY
    base_url: !env_var SONARR_URL http://sonarr:1233
";

        var configCollection = sut.LoadFromStream(new StringReader(testYml));

        var config = configCollection.Get<SonarrConfiguration>(SupportedServices.Sonarr);
        config.Should().BeEquivalentTo(new[]
        {
            new
            {
                BaseUrl = "the_url",
                ApiKey = "the_api_key"
            }
        });
    }

    [Test]
    public void Use_default_value_if_env_var_not_defined()
    {
        var sut = Resolve<ConfigurationLoader>();

        const string testYml = @"
sonarr:
  instance:
    base_url: !env_var SONARR_URL http://sonarr:1233
    api_key: value
";

        var configCollection = sut.LoadFromStream(new StringReader(testYml));

        var config = configCollection.Get<SonarrConfiguration>(SupportedServices.Sonarr);
        config.Should().BeEquivalentTo(new[]
        {
            new
            {
                BaseUrl = "http://sonarr:1233"
            }
        });
    }

    [Test]
    public void Default_value_with_spaces_is_allowed()
    {
        var env = Resolve<IEnvironment>();
        env.GetEnvironmentVariable("SONARR_URL").Returns("");

        var sut = Resolve<ConfigurationLoader>();

        const string testYml = @"
sonarr:
  instance:
    base_url: !env_var SONARR_URL some value
    api_key: value
";

        var configCollection = sut.LoadFromStream(new StringReader(testYml));

        var config = configCollection.Get<SonarrConfiguration>(SupportedServices.Sonarr);
        config.Should().BeEquivalentTo(new[]
        {
            new
            {
                BaseUrl = "some value"
            }
        });
    }

    [Test]
    public void Quotation_characters_are_stripped_from_default_value()
    {
        var env = Resolve<IEnvironment>();
        env.GetEnvironmentVariable("SONARR_URL").Returns("");

        var sut = Resolve<ConfigurationLoader>();

        const string testYml = @"
sonarr:
  instance:
    base_url: !env_var SONARR_URL ""the url""
    api_key: !env_var SONARR_API 'the key'
";

        var configCollection = sut.LoadFromStream(new StringReader(testYml));

        var config = configCollection.Get<SonarrConfiguration>(SupportedServices.Sonarr);
        config.Should().BeEquivalentTo(new[]
        {
            new
            {
                BaseUrl = "the url",
                ApiKey = "the key"
            }
        });
    }

    [Test]
    public void Multiple_spaces_between_default_and_env_var_work()
    {
        var sut = Resolve<ConfigurationLoader>();

        const string testYml = @"
sonarr:
  instance:
    base_url: !env_var SONARR_URL    some value
    api_key: value
";

        var configCollection = sut.LoadFromStream(new StringReader(testYml));

        var config = configCollection.Get<SonarrConfiguration>(SupportedServices.Sonarr);
        config.Should().BeEquivalentTo(new[]
        {
            new
            {
                BaseUrl = "some value"
            }
        });
    }

    [Test]
    public void Tab_characters_are_stripped()
    {
        var sut = Resolve<ConfigurationLoader>();

        const string testYml = $@"
sonarr:
  instance:
    base_url: !env_var SONARR_URL {"\t"}some value
    api_key: value
";

        var configCollection = sut.LoadFromStream(new StringReader(testYml));

        var config = configCollection.Get<SonarrConfiguration>(SupportedServices.Sonarr);
        config.Should().BeEquivalentTo(new[]
        {
            new
            {
                BaseUrl = "some value"
            }
        });
    }
}