chore: initial git load of code space

tests/Strata.SqlTools.Markdown.Tests/PostgreSql/SqlStatementGeneratorTests.cs

@@ -0,0 +1,179 @@
+using Strata.SqlTools.Breakdowns.PostgreSql;
+using Strata.SqlTools.Markdown.PostgreSql;
+
+namespace Strata.SqlTools.Markdown.Tests.PostgreSql;
+
+[TestFixture]
+public class SqlStatementGeneratorTests
+{
+    private SqlStatementGenerator _generator = null!;
+
+    [SetUp]
+    public void Setup()
+    {
+        _generator = new SqlStatementGenerator();
+    }
+
+    [Test]
+    public void GenerateSequenceDiagram_PostgreSqlQuery_GeneratesValidDiagram()
+    {
+        // Arrange
+        var query = QueryBreakdown.Parse(@"
+            SELECT user_id, user_name, email
+            FROM users
+            WHERE status = 'ACTIVE'
+        ");
+
+        // Act
+        var result = _generator.GenerateSequenceDiagram(query, "User Query Flow");
+
+        // Assert
+        Assert.That(result, Does.Contain("```mermaid"));
+        Assert.That(result, Does.Contain("sequenceDiagram"));
+        Assert.That(result, Does.Contain("### User Query Flow"));
+        Assert.That(result, Does.Contain("```"));
+    }
+
+    [Test]
+    public void GenerateSequenceDiagram_WithPostgreSqlCTE_ShowsMainStatement()
+    {
+        // Arrange
+        var query = QueryBreakdown.Parse(@"
+            WITH active_customers AS (
+                SELECT customer_id, customer_name
+                FROM customers
+                WHERE status = 'ACTIVE'
+            )
+            SELECT c.customer_id, c.customer_name, COUNT(o.order_id) AS order_count
+            FROM active_customers c
+            LEFT JOIN orders o ON c.customer_id = o.customer_id
+            GROUP BY c.customer_id, c.customer_name
+        ");
+
+        // Act
+        var result = _generator.GenerateSequenceDiagram(query);
+
+        // Assert
+        Assert.That(result, Does.Contain("Execute Main Statement"));
+        Assert.That(result, Does.Contain("activate DB"));
+        Assert.That(result, Does.Contain("deactivate DB"));
+    }
+
+    [Test]
+    public void GenerateSequenceDiagram_ComplexPostgreSqlQuery_GeneratesSequenceDiagram()
+    {
+        // Arrange
+        var query = QueryBreakdown.Parse(@"
+            SELECT region, product, SUM(sales) AS total_sales
+            FROM sales_data
+            WHERE sale_date >= $1
+            GROUP BY region, product
+            HAVING SUM(sales) > 1000
+            ORDER BY total_sales DESC
+        ");
+
+        // Act
+        var result = _generator.GenerateSequenceDiagram(query, "Sales Analysis");
+
+        // Assert
+        Assert.That(result, Does.Contain("### Sales Analysis"));
+        Assert.That(result, Does.Contain("sequenceDiagram"));
+        Assert.That(result, Does.Contain("Execute Main Statement"));
+        Assert.That(result, Does.Contain("activate DB"));
+        Assert.That(result, Does.Contain("Return Results"));
+    }
+
+    [Test]
+    public void GenerateEntityRelationshipDiagram_PostgreSqlTables_GeneratesERDiagram()
+    {
+        // Arrange - PostgreSQL typically uses lowercase table names
+        var tables = new[] { "customers", "orders", "order_items", "products" };
+
+        // Act
+        var result = _generator.GenerateEntityRelationshipDiagram(tables, "E-Commerce Schema");
+
+        // Assert
+        Assert.That(result, Does.Contain("```mermaid"));
+        Assert.That(result, Does.Contain("erDiagram"));
+        Assert.That(result, Does.Contain("### E-Commerce Schema"));
+        Assert.That(result, Does.Contain("customers"));
+        Assert.That(result, Does.Contain("orders"));
+        Assert.That(result, Does.Contain("order_items"));
+        Assert.That(result, Does.Contain("products"));
+    }
+
+    [Test]
+    public void GenerateEntityRelationshipDiagram_WithDoubleQuotedNames_HandlesQuotedIdentifiers()
+    {
+        // Arrange - PostgreSQL case-sensitive identifiers
+        var tables = new[] { "\"Customers\"", "\"Orders\"", "\"OrderItems\"" };
+
+        // Act
+        var result = _generator.GenerateEntityRelationshipDiagram(tables);
+
+        // Assert
+        Assert.That(result, Does.Contain("erDiagram"));
+        Assert.That(result, Does.Contain("```"));
+    }
+
+    [Test]
+    public void GenerateEntityRelationshipDiagram_WithSchemaQualifiedNames_HandlesSchemaDotTable()
+    {
+        // Arrange - PostgreSQL schema.table qualified names
+        var tables = new[] { "public.users", "sales.orders", "inventory.products" };
+
+        // Act
+        var result = _generator.GenerateEntityRelationshipDiagram(tables);
+
+        // Assert
+        Assert.That(result, Does.Contain("erDiagram"));
+        Assert.That(result, Does.Contain("```"));
+    }
+
+    [Test]
+    public void GenerateEntityRelationshipDiagram_SingleTable_GeneratesDiagram()
+    {
+        // Arrange
+        var tables = new[] { "users" };
+
+        // Act
+        var result = _generator.GenerateEntityRelationshipDiagram(tables);
+
+        // Assert
+        Assert.That(result, Does.Contain("erDiagram"));
+        Assert.That(result, Does.Contain("users"));
+    }
+
+    [Test]
+    public void GenerateSequenceDiagram_NullTitle_GeneratesWithoutTitle()
+    {
+        // Arrange
+        var query = new QueryBreakdown("order_id", "orders");
+
+        // Act
+        var result = _generator.GenerateSequenceDiagram(query, null);
+
+        // Assert
+        Assert.That(result, Does.Not.Contain("###"));
+        Assert.That(result, Does.Contain("sequenceDiagram"));
+    }
+
+    [Test]
+    public void GenerateSequenceDiagram_WithPositionalParameters_HandlesParametrizedQuery()
+    {
+        // Arrange - PostgreSQL positional parameters $1, $2
+        var query = QueryBreakdown.Parse(@"
+            SELECT product_id, product_name, price
+            FROM products
+            WHERE category_id = $1 AND price > $2
+            ORDER BY price DESC
+        ");
+
+        // Act
+        var result = _generator.GenerateSequenceDiagram(query);
+
+        // Assert
+        Assert.That(result, Does.Contain("sequenceDiagram"));
+        Assert.That(result, Does.Contain("Execute Main Statement"));
+    }
+}