// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Text.Json;
using Xunit;
 
namespace Microsoft.Extensions.AI;
 
public class TextContentTests
{
    [Theory]
    [InlineData(null)]
    [InlineData("")]
    [InlineData("text")]
    public void Constructor_String_PropsDefault(string? text)
    {
        TextContent c = new(text);
        Assert.Null(c.RawRepresentation);
        Assert.Null(c.AdditionalProperties);
        Assert.Equal(text ?? string.Empty, c.Text);
    }
 
    [Fact]
    public void Constructor_PropsRoundtrip()
    {
        TextContent c = new(null);
 
        Assert.Null(c.RawRepresentation);
        object raw = new();
        c.RawRepresentation = raw;
        Assert.Same(raw, c.RawRepresentation);
 
        Assert.Null(c.AdditionalProperties);
        AdditionalPropertiesDictionary props = new() { { "key", "value" } };
        c.AdditionalProperties = props;
        Assert.Same(props, c.AdditionalProperties);
 
        Assert.Equal(string.Empty, c.Text);
        c.Text = "text";
        Assert.Equal("text", c.Text);
        Assert.Equal("text", c.ToString());
 
        c.Text = null;
        Assert.Equal(string.Empty, c.Text);
        Assert.Equal(string.Empty, c.ToString());
 
        c.Text = string.Empty;
        Assert.Equal(string.Empty, c.Text);
        Assert.Equal(string.Empty, c.ToString());
    }
 
    [Fact]
    public void Serialization_Roundtrips()
    {
        var content = new TextContent("Hello, world!");
 
        var json = JsonSerializer.Serialize(content, AIJsonUtilities.DefaultOptions);
        var deserializedContent = JsonSerializer.Deserialize<TextContent>(json, AIJsonUtilities.DefaultOptions);
 
        Assert.NotNull(deserializedContent);
        Assert.Equal(content.Text, deserializedContent.Text);
    }
 
    [Fact]
    public void JsonDeserialization_KnownPayload()
    {
        const string Json = """
            {
              "$type": "text",
              "text": "Hello, world!",
              "additionalProperties": {
                "key1": "value1"
              }
            }
            """;
 
        AIContent? result = JsonSerializer.Deserialize<AIContent>(Json, AIJsonUtilities.DefaultOptions);
 
        Assert.NotNull(result);
        var textContent = Assert.IsType<TextContent>(result);
        Assert.Equal("Hello, world!", textContent.Text);
        Assert.NotNull(textContent.AdditionalProperties);
        Assert.Single(textContent.AdditionalProperties);
        Assert.Equal("value1", textContent.AdditionalProperties["key1"]?.ToString());
    }
}