using Strata.SqlTools.SqlBreakdown.Classes;

namespace Strata.SqlTools.SqlBreakdown.Tests.SqlBreakdownCollectionTests;

/// <summary>
/// Unit tests for SqlBreakdownCollection with DELETE statements.
/// </summary>
[TestFixture]
public class SqlBreakdownCollectionDeleteTests : SqlBreakdownCollectionTestBase
{
    [Test]
    public void ParseBatch_WithDeleteStatements_ParsesSuccessfully()
    {
        // Arrange
        var collection = new SqlBreakdownCollection();
        var batchSql = @"
            DELETE FROM Orders WHERE OrderDate < '2019-01-01'
            GO
            DELETE FROM UserProfiles WHERE UserId NOT IN (SELECT Id FROM Users)
            GO
            DELETE FROM AuditLog WHERE LogDate < DATEADD(year, -2, GETDATE())
            ";

        // Act
        collection.ParseBatch(batchSql);

        // Assert
        Assert.That(collection.Count, Is.EqualTo(3));
        Assert.That(collection.GetRawStatementAt(0), Does.Contain("DELETE FROM Orders"));
        Assert.That(collection.GetRawStatementAt(1), Does.Contain("NOT IN"));
        Assert.That(collection.GetRawStatementAt(2), Does.Contain("DATEADD"));
    }

    [Test]
    public void ParseBatch_WithDeleteFromSelectStatements_ParsesSuccessfully()
    {
        // Arrange
        var collection = new SqlBreakdownCollection();
        var batchSql = @"
            DELETE o FROM Orders o
            WHERE NOT EXISTS (SELECT 1 FROM OrderItems oi WHERE oi.OrderId = o.OrderId)
            GO
            DELETE FROM Users
            WHERE Id IN (SELECT UserId FROM Orders WHERE OrderStatus = 'Cancelled')
            ";

        // Act
        collection.ParseBatch(batchSql);

        // Assert
        Assert.That(collection.Count, Is.EqualTo(2));
        Assert.That(collection.GetRawStatementAt(0), Does.Contain("EXISTS"));
    }

    [Test]
    public void ParseBatch_WithDeleteWithJoinStatements_ParsesSuccessfully()
    {
        // Arrange
        var collection = new SqlBreakdownCollection();
        var batchSql = @"
            DELETE Orders
            FROM Orders o
            INNER JOIN OrderStatuses os ON o.StatusId = os.Id
            WHERE os.StatusName = 'Archived'
            ";

        // Act
        collection.ParseBatch(batchSql);

        // Assert
        Assert.That(collection.Count, Is.GreaterThanOrEqualTo(1));
        Assert.That(collection.GetRawStatementAt(0), Does.Contain("DELETE Orders"));
        Assert.That(collection.GetRawStatementAt(0), Does.Contain("INNER JOIN"));
    }
}