// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System;
using System.Text.Json;
using Xunit;
 
namespace Microsoft.Extensions.AI;
 
public class EmbeddingGenerationOptionsTests
{
    [Fact]
    public void Constructor_Parameterless_PropsDefaulted()
    {
        EmbeddingGenerationOptions options = new();
        Assert.Null(options.ModelId);
        Assert.Null(options.AdditionalProperties);
        Assert.Null(options.Dimensions);
 
        EmbeddingGenerationOptions clone = options.Clone();
        Assert.Null(clone.ModelId);
        Assert.Null(clone.AdditionalProperties);
        Assert.Null(clone.Dimensions);
    }
 
    [Fact]
    public void InvalidArgs_Throws()
    {
        EmbeddingGenerationOptions options = new();
        Assert.Throws<ArgumentOutOfRangeException>("value", () => options.Dimensions = 0);
        Assert.Throws<ArgumentOutOfRangeException>("value", () => options.Dimensions = -1);
    }
 
    [Fact]
    public void Properties_Roundtrip()
    {
        EmbeddingGenerationOptions options = new();
 
        AdditionalPropertiesDictionary additionalProps = new()
        {
            ["key"] = "value",
        };
 
        Func<IEmbeddingGenerator?, object?> rawRepresentationFactory = (c) => null;
 
        options.ModelId = "modelId";
        options.Dimensions = 1536;
        options.AdditionalProperties = additionalProps;
        options.RawRepresentationFactory = rawRepresentationFactory;
 
        Assert.Equal("modelId", options.ModelId);
        Assert.Equal(1536, options.Dimensions);
        Assert.Same(additionalProps, options.AdditionalProperties);
        Assert.Same(rawRepresentationFactory, options.RawRepresentationFactory);
 
        EmbeddingGenerationOptions clone = options.Clone();
        Assert.Equal("modelId", clone.ModelId);
        Assert.Equal(1536, clone.Dimensions);
        Assert.Equal(additionalProps, clone.AdditionalProperties);
        Assert.Same(rawRepresentationFactory, clone.RawRepresentationFactory);
    }
 
    [Fact]
    public void JsonSerialization_Roundtrips()
    {
        EmbeddingGenerationOptions options = new();
 
        AdditionalPropertiesDictionary additionalProps = new()
        {
            ["key"] = "value",
        };
 
        options.ModelId = "model";
        options.AdditionalProperties = additionalProps;
        options.Dimensions = 1536;
 
        string json = JsonSerializer.Serialize(options, TestJsonSerializerContext.Default.EmbeddingGenerationOptions);
 
        EmbeddingGenerationOptions? deserialized = JsonSerializer.Deserialize(json, TestJsonSerializerContext.Default.EmbeddingGenerationOptions);
        Assert.NotNull(deserialized);
 
        Assert.Equal("model", deserialized.ModelId);
        Assert.Equal(1536, deserialized.Dimensions);
 
        Assert.NotNull(deserialized.AdditionalProperties);
        Assert.Single(deserialized.AdditionalProperties);
        Assert.True(deserialized.AdditionalProperties.TryGetValue("key", out object? value));
        Assert.IsType<JsonElement>(value);
        Assert.Equal("value", ((JsonElement)value!).GetString());
    }
 
    [Fact]
    public void CopyConstructors_EnableHierarchyCloning()
    {
        OptionsB b = new()
        {
            ModelId = "test",
            A = 42,
            B = 84,
        };
 
        EmbeddingGenerationOptions clone = b.Clone();
 
        Assert.Equal("test", clone.ModelId);
        Assert.Equal(42, Assert.IsType<OptionsA>(clone, exactMatch: false).A);
        Assert.Equal(84, Assert.IsType<OptionsB>(clone, exactMatch: true).B);
    }
 
    private class OptionsA : EmbeddingGenerationOptions
    {
        public OptionsA()
        {
        }
 
        protected OptionsA(OptionsA other)
            : base(other)
        {
            A = other.A;
        }
 
        public int A { get; set; }
 
        public override EmbeddingGenerationOptions Clone() => new OptionsA(this);
    }
 
    private class OptionsB : OptionsA
    {
        public OptionsB()
        {
        }
 
        protected OptionsB(OptionsB other)
            : base(other)
        {
            B = other.B;
        }
 
        public int B { get; set; }
 
        public override EmbeddingGenerationOptions Clone() => new OptionsB(this);
    }
 
    [Fact]
    public void CopyConstructor_Null_Valid()
    {
        PassedNullToBaseOptions options = new();
        Assert.NotNull(options);
    }
 
    private class PassedNullToBaseOptions : EmbeddingGenerationOptions
    {
        public PassedNullToBaseOptions()
            : base(null)
        {
        }
    }
}