using System.Data;
using JdeScoping.DataSync.Etl.Transformers;
using NSubstitute;

namespace JdeScoping.DataSync.Tests.Etl.Transformers;

public class RegexTransformerTests
{
    [Fact]
    public void FindReplace_RemovesPrefix()
    {
        // Arrange
        var source = CreateMockReader(
            columns: new[] { "BatchID", "Name" },
            values: new object[] { "IIS_12345", "Test" });

        var transformer = new RegexTransformer(
            columnName: "BatchID",
            pattern: "^IIS_",
            replacement: "");

        // Act
        var reader = transformer.Transform(source);
        source.Read().Returns(true);
        reader.Read();

        // Assert
        Assert.Equal("12345", reader.GetValue(0));
        Assert.Equal("Test", reader.GetValue(1)); // Other column unchanged
    }

    [Fact]
    public void MatchExtract_ExtractsFirstCaptureGroup()
    {
        // Arrange
        var source = CreateMockReader(
            columns: new[] { "Code" },
            values: new object[] { "ID_12345" });

        var transformer = new RegexTransformer(
            columnName: "Code",
            pattern: @"ID_(\d+)",
            replacement: null); // null = Match & Extract mode

        // Act
        var reader = transformer.Transform(source);
        source.Read().Returns(true);
        reader.Read();

        // Assert
        Assert.Equal("12345", reader.GetValue(0));
    }

    [Fact]
    public void MatchExtract_NoMatch_KeepOriginal()
    {
        // Arrange
        var source = CreateMockReader(
            columns: new[] { "Code" },
            values: new object[] { "UNKNOWN" });

        var transformer = new RegexTransformer(
            columnName: "Code",
            pattern: @"ID_(\d+)",
            replacement: null,
            nonMatchBehavior: NonMatchBehavior.KeepOriginal);

        // Act
        var reader = transformer.Transform(source);
        source.Read().Returns(true);
        reader.Read();

        // Assert
        Assert.Equal("UNKNOWN", reader.GetValue(0));
    }

    [Fact]
    public void MatchExtract_NoMatch_ReturnNull()
    {
        // Arrange
        var source = CreateMockReader(
            columns: new[] { "Code" },
            values: new object[] { "UNKNOWN" });

        var transformer = new RegexTransformer(
            columnName: "Code",
            pattern: @"ID_(\d+)",
            replacement: null,
            nonMatchBehavior: NonMatchBehavior.ReturnNull);

        // Act
        var reader = transformer.Transform(source);
        source.Read().Returns(true);
        reader.Read();

        // Assert
        Assert.Equal(DBNull.Value, reader.GetValue(0));
    }

    [Fact]
    public void MatchExtract_NoMatch_ReturnEmpty()
    {
        // Arrange
        var source = CreateMockReader(
            columns: new[] { "Code" },
            values: new object[] { "UNKNOWN" });

        var transformer = new RegexTransformer(
            columnName: "Code",
            pattern: @"ID_(\d+)",
            replacement: null,
            nonMatchBehavior: NonMatchBehavior.ReturnEmpty);

        // Act
        var reader = transformer.Transform(source);
        source.Read().Returns(true);
        reader.Read();

        // Assert
        Assert.Equal(string.Empty, reader.GetValue(0));
    }

    private static IDataReader CreateMockReader(string[] columns, object[] values)
    {
        var reader = Substitute.For<IDataReader>();
        reader.FieldCount.Returns(columns.Length);
        for (int i = 0; i < columns.Length; i++)
        {
            var index = i;
            reader.GetName(index).Returns(columns[index]);
            reader.GetOrdinal(columns[index]).Returns(index);
            reader.GetFieldType(index).Returns(values[index]?.GetType() ?? typeof(object));
            reader.GetValue(index).Returns(values[index]);
            reader.IsDBNull(index).Returns(values[index] == null || values[index] == DBNull.Value);
        }
        return reader;
    }
}