// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
namespace Aspire.TerminalHost.Tests;
 
public class TerminalHostArgsTests
{
    [Fact]
    public void ParseAllRequiredArgsSucceeds()
    {
        var args = TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p.sock",
            "--consumer-uds", "/tmp/c.sock",
            "--control-uds", "/tmp/ctrl.sock",
        ]);
 
        Assert.Equal("/tmp/p.sock", args.ProducerUdsPath);
        Assert.Equal("/tmp/c.sock", args.ConsumerUdsPath);
        Assert.Equal("/tmp/ctrl.sock", args.ControlUdsPath);
        Assert.Equal(120, args.Columns);
        Assert.Equal(30, args.Rows);
    }
 
    [Fact]
    public void ParseAcceptsOptionalDimensions()
    {
        var args = TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p.sock",
            "--consumer-uds", "/tmp/c.sock",
            "--control-uds", "/tmp/ctrl.sock",
            "--columns", "200",
            "--rows", "50",
        ]);
 
        Assert.Equal(200, args.Columns);
        Assert.Equal(50, args.Rows);
    }
 
    [Fact]
    public void ParseAcceptsIgnoredShellForCompatibility()
    {
        var args = TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p.sock",
            "--consumer-uds", "/tmp/c.sock",
            "--control-uds", "/tmp/ctrl.sock",
            "--shell", "/bin/bash",
        ]);
 
        Assert.Equal("/tmp/p.sock", args.ProducerUdsPath);
        Assert.Equal("/tmp/c.sock", args.ConsumerUdsPath);
        Assert.Equal("/tmp/ctrl.sock", args.ControlUdsPath);
    }
 
    [Fact]
    public void ParseMissingProducerUdsThrows()
    {
        var ex = Assert.Throws<TerminalHostArgsException>(() => TerminalHostArgs.Parse([
            "--consumer-uds", "/tmp/c.sock",
            "--control-uds", "/tmp/ctrl.sock",
        ]));
 
        Assert.Contains("--producer-uds", ex.Message);
    }
 
    [Fact]
    public void ParseMissingConsumerUdsThrows()
    {
        var ex = Assert.Throws<TerminalHostArgsException>(() => TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p.sock",
            "--control-uds", "/tmp/ctrl.sock",
        ]));
 
        Assert.Contains("--consumer-uds", ex.Message);
    }
 
    [Fact]
    public void ParseMissingControlUdsThrows()
    {
        var ex = Assert.Throws<TerminalHostArgsException>(() => TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p.sock",
            "--consumer-uds", "/tmp/c.sock",
        ]));
 
        Assert.Contains("--control-uds", ex.Message);
    }
 
    [Fact]
    public void ParseDuplicateProducerUdsThrows()
    {
        // Each terminal host serves exactly one replica, so passing two producer-uds
        // entries is interpreted as a misuse of the new single-replica argument shape.
        var ex = Assert.Throws<TerminalHostArgsException>(() => TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p0.sock",
            "--producer-uds", "/tmp/p1.sock",
            "--consumer-uds", "/tmp/c.sock",
            "--control-uds", "/tmp/ctrl.sock",
        ]));
 
        Assert.Contains("--producer-uds", ex.Message);
    }
 
    [Fact]
    public void ParseDuplicateConsumerUdsThrows()
    {
        var ex = Assert.Throws<TerminalHostArgsException>(() => TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p.sock",
            "--consumer-uds", "/tmp/c0.sock",
            "--consumer-uds", "/tmp/c1.sock",
            "--control-uds", "/tmp/ctrl.sock",
        ]));
 
        Assert.Contains("--consumer-uds", ex.Message);
    }
 
    [Fact]
    public void ParseNegativeColumnsThrows()
    {
        var ex = Assert.Throws<TerminalHostArgsException>(() => TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p.sock",
            "--consumer-uds", "/tmp/c.sock",
            "--control-uds", "/tmp/ctrl.sock",
            "--columns", "-5",
        ]));
 
        Assert.Contains("--columns", ex.Message);
    }
 
    [Fact]
    public void ParseUnknownArgumentThrows()
    {
        var ex = Assert.Throws<TerminalHostArgsException>(() => TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p.sock",
            "--consumer-uds", "/tmp/c.sock",
            "--control-uds", "/tmp/ctrl.sock",
            "--bogus",
        ]));
 
        Assert.Contains("--bogus", ex.Message);
    }
 
    [Fact]
    public void ParseMissingValueForArgumentThrows()
    {
        var ex = Assert.Throws<TerminalHostArgsException>(() => TerminalHostArgs.Parse([
            "--producer-uds",
        ]));
 
        Assert.Contains("--producer-uds", ex.Message);
    }
 
    [Fact]
    public void ParseNonIntegerForColumnsThrows()
    {
        var ex = Assert.Throws<TerminalHostArgsException>(() => TerminalHostArgs.Parse([
            "--producer-uds", "/tmp/p.sock",
            "--consumer-uds", "/tmp/c.sock",
            "--control-uds", "/tmp/ctrl.sock",
            "--columns", "abc",
        ]));
 
        Assert.Contains("--columns", ex.Message);
    }
 
    [Fact]
    public void ParseNullArgsThrows()
    {
        Assert.Throws<ArgumentNullException>(() => TerminalHostArgs.Parse(null!));
    }
}