// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
using System;
using System.Collections.Generic;
using System.Runtime.CompilerServices;
using System.Text;
using ApprovalTests;
using ApprovalTests.Reporters;
using Microsoft.ML.CodeGenerator.Templates.Azure.Model;
using Microsoft.ML.CodeGenerator.Templates.Console;
using Microsoft.ML.TestFramework;
using Xunit;
using Xunit.Abstractions;
 
namespace Microsoft.ML.CodeGenerator.Tests
{
    [UseReporter(typeof(DiffReporter))]
    public class TemplateTest : BaseTestClass
    {
        public TemplateTest(ITestOutputHelper output) : base(output)
        {
            if (Environment.GetEnvironmentVariable("HELIX_CORRELATION_ID") != null)
            {
                Approvals.UseAssemblyLocationForApprovedFiles();
            }
        }
 
        [Fact]
        [UseReporter(typeof(DiffReporter))]
        [MethodImpl(MethodImplOptions.NoInlining)]
        public void TestPredictProgram_WithSampleData()
        {
            var predictProgram = new PredictProgram()
            {
                SampleData = new Dictionary<string, string>()
                {
                    { "key1", "\"key1\"" },
                    { "key2", "\"key2\"" },
                    { "key3", "\"key\\\"3\"" },
                },
                TaskType = "null",
                Features = new List<string>(),
                Namespace = "Namespace",
                LabelName = "LabelName",
                Separator = ','
            };
            Approvals.Verify(predictProgram.TransformText());
        }
 
        [Fact]
        [UseReporter(typeof(DiffReporter))]
        [MethodImpl(MethodImplOptions.NoInlining)]
        public void TestConsumeModel_NotAzureImage_NotObjectDetection()
        {
            var consumeModel = new ConsumeModel()
            {
                Namespace = "Namespace",
                IsAzureImage = false,
                IsAzureObjectDetection = false,
                MLNetModelName = @"mlmodel.zip",
            };
 
            Approvals.Verify(consumeModel.TransformText());
        }
 
        [Fact]
        [UseReporter(typeof(DiffReporter))]
        [MethodImpl(MethodImplOptions.NoInlining)]
        public void TestConsumeModel_AzureImage()
        {
            var consumeModel = new ConsumeModel()
            {
                Namespace = "Namespace",
                IsAzureImage = true,
                IsAzureObjectDetection = false,
                MLNetModelName = @"mlmodel.zip",
                OnnxModelName = "onnx.onnx"
            };
 
            Approvals.Verify(consumeModel.TransformText());
        }
 
        [Fact]
        [UseReporter(typeof(DiffReporter))]
        [MethodImpl(MethodImplOptions.NoInlining)]
        public void TestConsumeModel_AzureObjectDetection()
        {
            var consumeModel = new ConsumeModel()
            {
                Namespace = "Namespace",
                IsAzureImage = false,
                IsAzureObjectDetection = true,
                MLNetModelName = @"mlmodel.zip",
                OnnxModelName = "onnx.onnx"
            };
 
            Approvals.Verify(consumeModel.TransformText());
        }
 
        [Fact]
        [UseReporter(typeof(DiffReporter))]
        [MethodImpl(MethodImplOptions.NoInlining)]
        public void TestAzureImageModelOutputClass_WithLabel()
        {
            var azureImageOutput = new AzureImageModelOutputClass
            {
                Namespace = "Namespace",
                Labels = new string[] { "str1", "str2", "str3" },
                Target = CSharp.GenerateTarget.Cli
            };
 
            Approvals.Verify(azureImageOutput.TransformText());
        }
 
        [Fact]
        [UseReporter(typeof(DiffReporter))]
        [MethodImpl(MethodImplOptions.NoInlining)]
        public void TestAzureObjectDetectionModelOutputClass_WithLabel()
        {
            var azureImageOutput = new AzureObjectDetectionModelOutputClass
            {
                Namespace = "Namespace",
                Labels = new string[] { "str1", "str2", "str3" },
                Target = CSharp.GenerateTarget.Cli
            };
 
            Approvals.Verify(azureImageOutput.TransformText());
        }
 
        [Fact]
        [UseReporter(typeof(DiffReporter))]
        [MethodImpl(MethodImplOptions.NoInlining)]
        public void TestModelBuilder_noOnnx_noTestData()
        {
            var modelBuilder = new ModelBuilder()
            {
                Namespace = "Namespace",
                HasOnnxModel = false,
                Path = "Path",
                Separator = ',',
                PreTrainerTransforms = new string[] { "PreTrainerTransformer1" },
                Trainer = "Trainer",
                TaskType = "Task1",
                GeneratedUsings = "Using package1",
                AllowQuoting = true,
                AllowSparse = true,
                LabelName = "Label",
                CacheBeforeTrainer = true,
                PostTrainerTransforms = new string[] { "PostTrainerTransformer1" },
                MLNetModelName = "/path/to/model",
            };
 
            Approvals.Verify(modelBuilder.TransformText());
        }
    }
}