kompas3d-mcp/tests/Kompas.Mcp.Tests/Integration/DrawingLeaderTests.cs

using Kompas.Mcp.Core.Documents;
using Kompas.Mcp.Core.Drawings;
using Kompas.Mcp.Core.Modeling;

namespace Kompas.Mcp.Tests.Integration;

/// <summary>Интеграция: линия-выноска с текстом на виде (API7 ISymbols2DContainer.Leaders + IBranchs).</summary>
[Trait("Category", "Integration")]
[Collection(KompasCollection.Name)]
public sealed class DrawingLeaderTests : IntegrationTestBase
{
    private readonly DocumentService _docs;
    private readonly PartModeler _modeler;
    private readonly DrawingService _drawing;

    public DrawingLeaderTests(KompasFixture fx) : base(fx)
    {
        _docs = new DocumentService(fx.Session, fx.Dispatcher);
        _modeler = new PartModeler(fx.Session, fx.Dispatcher);
        _drawing = new DrawingService(fx.Session, fx.Dispatcher);
    }

    private async Task<string> BuildAndSaveBoxAsync()
    {
        var path = TestPaths.NewFile(".m3d");
        await _docs.CreateAsync(KompasDocumentType.Part);
        try
        {
            var s = await _modeler.OpenSketchAsync(BasePlane.XOY);
            await _modeler.AddRectangleAsync(s, 0, 0, 40, 30);
            await _modeler.CloseSketchAsync(s);
            await _modeler.ExtrudeAsync(s, depth: 20);
            await _modeler.RebuildAsync();
            await _docs.SaveAsAsync(path);
        }
        finally { await _docs.CloseAsync(save: false); }
        return path;
    }

    [Fact]
    public async Task Leader_is_placed_round_trip_and_resolves_default_view()
    {
        var partFile = await BuildAndSaveBoxAsync();
        await _docs.CreateAsync(KompasDocumentType.Drawing);
        try
        {
            var views = await _drawing.CreateStandardViewsAsync(partFile, 1.0, 100, 150);
            var first = views.ViewNumbers[0];

            var before = await _drawing.GetViewLeaderCountAsync(first);
            // viewNumber=0 → первый несистемный вид (= first)
            var r = await _drawing.AddLeaderAsync(0, x: 5, y: 5, textX: 25, textY: 20,
                text: "Примечание 1", shelfDirection: ShelfDirection.Auto);
            var after = await _drawing.GetViewLeaderCountAsync(first);

            Assert.Equal(first, r.ViewNumber);
            Assert.Equal("Примечание 1", r.Value); // read-back из COM
            Assert.Equal(before + 1, after);
        }
        finally { await _docs.CloseAsync(save: false); }
    }

    [Theory]
    [InlineData(ShelfDirection.Right)]
    [InlineData(ShelfDirection.Left)]
    [InlineData(ShelfDirection.Up)]
    [InlineData(ShelfDirection.Down)]
    public async Task Leader_with_explicit_shelf_direction(ShelfDirection dir)
    {
        var partFile = await BuildAndSaveBoxAsync();
        await _docs.CreateAsync(KompasDocumentType.Drawing);
        try
        {
            var views = await _drawing.CreateStandardViewsAsync(partFile, 1.0, 100, 150);
            var target = views.ViewNumbers[0];
            var before = await _drawing.GetViewLeaderCountAsync(target);
            // Покрывает всю поверхность ShelfDirections.ToKompas (Right/Left/Up/Down) на реальном COM.
            var r = await _drawing.AddLeaderAsync(target, x: 10, y: 8, textX: 30, textY: 25,
                text: "Полка", shelfDirection: dir);
            var after = await _drawing.GetViewLeaderCountAsync(target);
            Assert.Equal(before + 1, after);
            Assert.Equal("Полка", r.Value);
        }
        finally { await _docs.CloseAsync(save: false); }
    }

    [Theory]
    [InlineData("")]
    [InlineData("   ")]
    public async Task Empty_text_throws(string text)
    {
        var partFile = await BuildAndSaveBoxAsync();
        await _docs.CreateAsync(KompasDocumentType.Drawing);
        try
        {
            await _drawing.CreateStandardViewsAsync(partFile, 1.0, 100, 150);
            await Assert.ThrowsAsync<ArgumentException>(() =>
                _drawing.AddLeaderAsync(0, x: 5, y: 5, textX: 25, textY: 20, text: text,
                    shelfDirection: ShelfDirection.Auto));
        }
        finally { await _docs.CloseAsync(save: false); }
    }

    [Fact]
    public async Task Degenerate_tip_equals_shelf_throws()
    {
        var partFile = await BuildAndSaveBoxAsync();
        await _docs.CreateAsync(KompasDocumentType.Drawing);
        try
        {
            await _drawing.CreateStandardViewsAsync(partFile, 1.0, 100, 150);
            await Assert.ThrowsAsync<ArgumentException>(() =>
                _drawing.AddLeaderAsync(0, x: 5, y: 5, textX: 5, textY: 5, text: "X",
                    shelfDirection: ShelfDirection.Auto));
        }
        finally { await _docs.CloseAsync(save: false); }
    }

    [Fact]
    public async Task No_views_throws()
    {
        await _docs.CreateAsync(KompasDocumentType.Drawing);
        try
        {
            await Assert.ThrowsAsync<InvalidOperationException>(() =>
                _drawing.AddLeaderAsync(0, x: 5, y: 5, textX: 25, textY: 20, text: "X",
                    shelfDirection: ShelfDirection.Auto));
        }
        finally { await _docs.CloseAsync(save: false); }
    }

    [Fact]
    public async Task Unknown_view_number_throws()
    {
        var partFile = await BuildAndSaveBoxAsync();
        await _docs.CreateAsync(KompasDocumentType.Drawing);
        try
        {
            await _drawing.CreateStandardViewsAsync(partFile, 1.0, 100, 150);
            await Assert.ThrowsAsync<InvalidOperationException>(() =>
                _drawing.AddLeaderAsync(999, x: 5, y: 5, textX: 25, textY: 20, text: "X",
                    shelfDirection: ShelfDirection.Auto));
        }
        finally { await _docs.CloseAsync(save: false); }
    }
}