82 lines
2.7 KiB
C#
82 lines
2.7 KiB
C#
using Strata.SqlTools.SqlBreakdown.Classes;
|
|
|
|
namespace Strata.SqlTools.SqlBreakdown.Tests.SqlBreakdownCollectionTests;
|
|
|
|
/// <summary>
|
|
/// Unit tests for SqlBreakdownCollection with UPDATE statements.
|
|
/// </summary>
|
|
[TestFixture]
|
|
public class SqlBreakdownCollectionUpdateTests : SqlBreakdownCollectionTestBase
|
|
{
|
|
[Test]
|
|
public void ParseBatch_WithUpdateStatements_ParsesSuccessfully()
|
|
{
|
|
// Arrange
|
|
var collection = new SqlBreakdownCollection();
|
|
var batchSql = @"
|
|
UPDATE Users SET Name = 'John Updated', ModifiedDate = GETDATE() WHERE Id = 1
|
|
GO
|
|
UPDATE Orders SET Status = 'Shipped' WHERE OrderDate < DATEADD(day, -30, GETDATE())
|
|
GO
|
|
UPDATE Products SET Price = Price * 1.1 WHERE Category = 'Electronics'
|
|
";
|
|
|
|
// Act
|
|
collection.ParseBatch(batchSql);
|
|
|
|
// Assert
|
|
Assert.That(collection.Count, Is.EqualTo(3));
|
|
Assert.That(collection.GetRawStatementAt(0), Does.Contain("UPDATE Users"));
|
|
Assert.That(collection.GetRawStatementAt(1), Does.Contain("Status"));
|
|
Assert.That(collection.GetRawStatementAt(2), Does.Contain("Price * 1.1"));
|
|
}
|
|
|
|
[Test]
|
|
public void ParseBatch_WithUpdateFromSelectStatements_ParsesSuccessfully()
|
|
{
|
|
// Arrange
|
|
var collection = new SqlBreakdownCollection();
|
|
var batchSql = @"
|
|
UPDATE u
|
|
SET u.LastOrderDate = (SELECT MAX(OrderDate) FROM Orders WHERE UserId = u.Id)
|
|
FROM Users u
|
|
WHERE u.Id IN (SELECT DISTINCT UserId FROM Orders)
|
|
GO
|
|
UPDATE Products
|
|
SET Quantity = Quantity - 1
|
|
FROM OrderItems oi
|
|
WHERE Products.Id = oi.ProductId AND oi.OrderId = 100
|
|
";
|
|
|
|
// Act
|
|
collection.ParseBatch(batchSql);
|
|
|
|
// Assert
|
|
Assert.That(collection.Count, Is.EqualTo(2));
|
|
Assert.That(collection.GetRawStatementAt(0), Does.Contain("UPDATE u"));
|
|
Assert.That(collection.GetRawStatementAt(1), Does.Contain("Quantity - 1"));
|
|
}
|
|
|
|
[Test]
|
|
public void ParseBatch_WithUpdateWithJoinStatements_ParsesSuccessfully()
|
|
{
|
|
// Arrange
|
|
var collection = new SqlBreakdownCollection();
|
|
var batchSql = @"
|
|
UPDATE Orders
|
|
SET OrderStatus = 'Processed'
|
|
FROM Orders o
|
|
JOIN OrderItems oi ON o.OrderId = oi.OrderId
|
|
WHERE oi.Quantity > 50
|
|
";
|
|
|
|
// Act
|
|
collection.ParseBatch(batchSql);
|
|
|
|
// Assert
|
|
Assert.That(collection.Count, Is.GreaterThanOrEqualTo(1));
|
|
Assert.That(collection.GetRawStatementAt(0), Does.Contain("UPDATE Orders"));
|
|
Assert.That(collection.GetRawStatementAt(0), Does.Contain("JOIN"));
|
|
}
|
|
}
|