//-----------------------------------------------------------------------------
// <copyright file="TestXRayOptions.cs" company="Amazon.com">
//      Copyright 2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
//      Licensed under the Apache License, Version 2.0 (the "License").
//      You may not use this file except in compliance with the License.
//      A copy of the License is located at
//
//      http://aws.amazon.com/apache2.0
//
//      or in the "license" file accompanying this file. This file is distributed
//      on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
//      express or implied. See the License for the specific language governing
//      permissions and limitations under the License.
// </copyright>
//-----------------------------------------------------------------------------

using Microsoft.VisualStudio.TestTools.UnitTesting;
using Amazon.XRay.Recorder.Core.Internal.Utils;
using Microsoft.Extensions.Configuration;
using System.IO;
using Amazon.XRay.Recorder.Core;
using Amazon.XRay.Recorder.Core.Internal.Emitters;
using static Amazon.XRay.Recorder.UnitTests.AwsXrayRecorderTests;
using System;

namespace Amazon.XRay.Recorder.UnitTests
{
    [TestClass]
    public class TestXRayOptions
    {
        private const string DisableXRayTracingKey = "DisableXRayTracing";

        private static XRayOptions _xRayOptions;

        private static string PREFIX = $"JSONs{Path.DirectorySeparatorChar}Appsetting{Path.DirectorySeparatorChar}";

        [TestCleanup]
        public void TestCleanup()
        {
            _xRayOptions = new XRayOptions();
            Environment.SetEnvironmentVariable(AWSXRayRecorder.EnvironmentVariableContextMissingStrategy, null);
            AWSXRayRecorder.Instance.Dispose();
        }

        [TestMethod]
        public void TestDisableXRayTracingKeyTrue()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("DisabledXRayTrue.json");

            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);
            Assert.IsTrue(_xRayOptions.IsXRayTracingDisabled);
            Assert.IsNull(_xRayOptions.AwsServiceHandlerManifest);
            Assert.IsNull(_xRayOptions.PluginSetting);
            Assert.IsNull(_xRayOptions.SamplingRuleManifest);
        }

        [TestMethod]
        public void TestDisableXRayTracingKeyFalse()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("DisabledXRayFalse.json");

            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);
            Assert.IsFalse(_xRayOptions.IsXRayTracingDisabled);
            Assert.IsNull(_xRayOptions.AwsServiceHandlerManifest);
            Assert.IsNull(_xRayOptions.PluginSetting);
            Assert.IsNull(_xRayOptions.SamplingRuleManifest);
        }

        [TestMethod]
        public void TestDisableXRayTracingKeyMissing()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("DisabledXRayMissing.json");

            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);
            Assert.IsFalse(_xRayOptions.IsXRayTracingDisabled);
            Assert.IsNull(_xRayOptions.AwsServiceHandlerManifest);
            Assert.IsNull(_xRayOptions.PluginSetting);
            Assert.IsNull(_xRayOptions.SamplingRuleManifest);
        }

        [TestMethod]
        public void TestDisableXRayTracingKeyInvalid()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("DisabledXRayInvalid.json");

            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);
            Assert.IsFalse(_xRayOptions.IsXRayTracingDisabled);
            Assert.IsNull(_xRayOptions.AwsServiceHandlerManifest);
            Assert.IsNull(_xRayOptions.PluginSetting);
            Assert.IsNull(_xRayOptions.SamplingRuleManifest);
        }

        [TestMethod]
        public void TestConfigurationIsNull()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("NoXRaySection.json");

            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);
            Assert.IsFalse(_xRayOptions.IsXRayTracingDisabled);
            Assert.IsNull(_xRayOptions.AwsServiceHandlerManifest);
            Assert.IsNull(_xRayOptions.PluginSetting);
            Assert.IsNull(_xRayOptions.SamplingRuleManifest);
        }

        [TestMethod]
        public void TestXRaySectionMissing() // No "Xray" section in the json
        {
            IConfiguration configuration = null;

            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);
            Assert.IsFalse(_xRayOptions.IsXRayTracingDisabled);
            Assert.IsNull(_xRayOptions.AwsServiceHandlerManifest);
            Assert.IsNull(_xRayOptions.PluginSetting);
            Assert.IsNull(_xRayOptions.SamplingRuleManifest);
        }

        [TestMethod]
        public void TestInitializeInstance()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("DisabledXRayTrue.json");

            AWSXRayRecorder.InitializeInstance(configuration);
            _xRayOptions = AWSXRayRecorder.Instance.XRayOptions;

            Assert.IsTrue(_xRayOptions.IsXRayTracingDisabled);
            Assert.IsNull(_xRayOptions.AwsServiceHandlerManifest);
            Assert.IsNull(_xRayOptions.PluginSetting);
            Assert.IsNull(_xRayOptions.SamplingRuleManifest);
            Assert.IsFalse(_xRayOptions.UseRuntimeErrors);

            Assert.AreEqual(typeof(UdpSegmentEmitter), AWSXRayRecorder.Instance.Emitter.GetType()); // Default emitter set

            AWSXRayRecorder.Instance.Dispose();
        }

        [TestMethod]
        public void TestInitializeInstanceWithRecorder1()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("DisabledXRayTrue.json");

            AWSXRayRecorder recorder = BuildAWSXRayRecorder(new TestSamplingStrategy());

            AWSXRayRecorder.InitializeInstance(configuration, recorder);
            _xRayOptions = recorder.XRayOptions;
            Assert.IsTrue(_xRayOptions.IsXRayTracingDisabled);
            Assert.IsNull(_xRayOptions.AwsServiceHandlerManifest);
            Assert.IsNull(_xRayOptions.PluginSetting);
            Assert.IsNull(_xRayOptions.SamplingRuleManifest);
            Assert.IsFalse(_xRayOptions.UseRuntimeErrors);

            Assert.AreEqual(AWSXRayRecorder.Instance.SamplingStrategy, recorder.SamplingStrategy); // Custom recorder set in TraceContext
            Assert.AreEqual(typeof(UdpSegmentEmitter), recorder.Emitter.GetType()); // Default emitter set
            recorder.Dispose();
        }

        [TestMethod]
        public void TestInitializeInstanceWithRecorder2()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("DisabledXRayTrue.json");

            AWSXRayRecorder recorder = BuildAWSXRayRecorder(new TestSamplingStrategy(), new DummyEmitter());

            AWSXRayRecorder.InitializeInstance(configuration, recorder);
            _xRayOptions = recorder.XRayOptions;
            Assert.IsTrue(_xRayOptions.IsXRayTracingDisabled);
            Assert.IsNull(_xRayOptions.AwsServiceHandlerManifest);
            Assert.IsNull(_xRayOptions.PluginSetting);
            Assert.IsNull(_xRayOptions.SamplingRuleManifest);
            Assert.IsFalse(_xRayOptions.UseRuntimeErrors);

            Assert.AreEqual(AWSXRayRecorder.Instance.SamplingStrategy, recorder.SamplingStrategy); // Custom recorder set in TraceContext
            Assert.AreEqual(typeof(DummyEmitter), recorder.Emitter.GetType()); // custom emitter set
            recorder.Dispose();
        }

        [TestMethod]
        public void TestUseRuntimeErrorsFalse()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("UseRuntimeErrorsFalse.json");

            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);

            AWSXRayRecorder.InitializeInstance(configuration);

            Assert.IsFalse(_xRayOptions.UseRuntimeErrors);
            Assert.AreEqual(AWSXRayRecorder.Instance.ContextMissingStrategy, Core.Strategies.ContextMissingStrategy.LOG_ERROR);
        }

        [TestMethod]
        public void TestUseRuntimeErrorsTrue()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("UseRuntimeErrorsTrue.json");

            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);

            AWSXRayRecorder.InitializeInstance(configuration);

            Assert.IsTrue(_xRayOptions.UseRuntimeErrors);
            Assert.AreEqual(AWSXRayRecorder.Instance.ContextMissingStrategy, Core.Strategies.ContextMissingStrategy.RUNTIME_ERROR);
        }

        [TestMethod]
        public void TestUseRuntimeErrorsDefaultsFalse_WhenNotSpecifiedInJson()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("DisabledXRayMissing.json");

            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);

            AWSXRayRecorder.InitializeInstance(configuration);

            Assert.IsFalse(_xRayOptions.UseRuntimeErrors);
            Assert.AreEqual(AWSXRayRecorder.Instance.ContextMissingStrategy, Core.Strategies.ContextMissingStrategy.LOG_ERROR);
        }

        [TestMethod]
        public void TestCollectSqlQueriesFalse_WhenNotSpecifiedInJson()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("DisabledXRayMissing.json");
            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);
            AWSXRayRecorder.InitializeInstance(configuration);

            Assert.IsFalse(_xRayOptions.CollectSqlQueries);
        }

        [TestMethod]
        public void TestCollecSqlQueriesFalse()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("CollectSqlQueriesFalse.json");
            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);
            AWSXRayRecorder.InitializeInstance(configuration);

            Assert.IsFalse(_xRayOptions.CollectSqlQueries);
        }

        [TestMethod]
        public void TestCollecSqlQueriesTrue()
        {
            IConfiguration configuration = TestXRayOptions.BuildConfiguration("CollectSqlQueriesTrue.json");
            _xRayOptions = XRayConfiguration.GetXRayOptions(configuration);
            AWSXRayRecorder.InitializeInstance(configuration);

            Assert.IsTrue(_xRayOptions.CollectSqlQueries);
        }

        // Creating custom configuration
        private static IConfiguration BuildConfiguration(string path)
        {
            IConfigurationBuilder builder = new ConfigurationBuilder()
                              .SetBasePath(Directory.GetCurrentDirectory())
                              .AddJsonFile(PREFIX + path);
            IConfiguration configuration = builder.Build();
            return configuration;
        }
    }
}