style: format test files with Black

Apply Black formatting to:
- tests/unit/test_connection.py
- tests/unit/test_schema.py
- tests/unit/test_executor.py
- tests/integration/test_database.py

Author: Sakeeb91

tests/integration/test_database.py

@@ -27,46 +27,54 @@ async def test_engine() -> AsyncEngine:
     # Create test tables
     async with engine.begin() as conn:
         await conn.execute(
-            text("""
+            text(
+                """
             CREATE TABLE customers (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 name VARCHAR(100) NOT NULL,
                 email VARCHAR(255),
                 state VARCHAR(50)
             )
-        """)
+        """
+            )
         )
 
         await conn.execute(
-            text("""
+            text(
+                """
             CREATE TABLE orders (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 customer_id INTEGER NOT NULL,
                 amount DECIMAL(10, 2) NOT NULL,
                 order_date DATE NOT NULL,
                 FOREIGN KEY (customer_id) REFERENCES customers(id)
             )
-        """)
+        """
+            )
         )
 
         # Insert test data
         await conn.execute(
-            text("""
+            text(
+                """
             INSERT INTO customers (name, email, state) VALUES
                 ('Alice', '[email protected]', 'California'),
                 ('Bob', '[email protected]', 'New York'),
                 ('Charlie', '[email protected]', 'California')
-        """)
+        """
+            )
         )
 
         await conn.execute(
-            text("""
+            text(
+                """
             INSERT INTO orders (customer_id, amount, order_date) VALUES
                 (1, 100.50, '2024-01-15'),
                 (1, 200.00, '2024-02-20'),
                 (2, 150.75, '2024-01-20'),
                 (3, 75.25, '2024-03-01')
-        """)
+        """
+            )
         )
 
     yield engine
@@ -356,13 +364,15 @@ async def test_join_query(self, test_session: AsyncSession) -> None:
         """Test JOIN query."""
         executor = SafeQueryExecutor(test_session)
 
-        result = await executor.execute("""
+        result = await executor.execute(
+            """
             SELECT c.name, SUM(o.amount) as total_orders
             FROM customers c
             JOIN orders o ON c.id = o.customer_id
             GROUP BY c.id
             ORDER BY total_orders DESC
-        """)
+        """
+        )
 
         assert result.success is True
         assert result.row_count == 3
@@ -376,7 +386,8 @@ async def test_cte_query(self, test_session: AsyncSession) -> None:
         """Test Common Table Expression (CTE) query."""
         executor = SafeQueryExecutor(test_session)
 
-        result = await executor.execute("""
+        result = await executor.execute(
+            """
             WITH customer_totals AS (
                 SELECT customer_id, SUM(amount) as total
                 FROM orders
@@ -385,7 +396,8 @@ async def test_cte_query(self, test_session: AsyncSession) -> None:
             SELECT c.name, ct.total
             FROM customers c
             JOIN customer_totals ct ON c.id = ct.customer_id
-        """)
+        """
+        )
 
         assert result.success is True
         assert result.row_count == 3
@@ -468,4 +480,3 @@ async def test_schema_to_query_flow(self, test_engine: AsyncEngine) -> None:
             for row in result.rows:
                 assert "name" in row
                 assert "order_count" in row
-

tests/unit/test_connection.py

@@ -219,9 +219,7 @@ def reset_global_manager(self) -> None:
         db.connection._db_manager = None
 
     @pytest.mark.asyncio
-    async def test_get_database_creates_manager(
-        self, mock_settings: MagicMock
-    ) -> None:
+    async def test_get_database_creates_manager(self, mock_settings: MagicMock) -> None:
         """Test get_database creates and initializes manager."""
         with patch("db.connection.get_settings", return_value=mock_settings):
             with patch("db.connection.create_async_engine") as mock_engine:
@@ -267,4 +265,3 @@ async def test_close_database_when_not_initialized(self) -> None:
         """Test close_database when no manager exists."""
         # Should not raise any errors
         await close_database()
-

tests/unit/test_executor.py

@@ -326,9 +326,7 @@ async def test_execute_explain_invalid_query(self, mock_session: MagicMock) -> N
     @pytest.mark.asyncio
     async def test_execute_explain_failure(self, mock_session: MagicMock) -> None:
         """Test EXPLAIN failure returns error info."""
-        mock_session.execute = AsyncMock(
-            side_effect=Exception("EXPLAIN not supported")
-        )
+        mock_session.execute = AsyncMock(side_effect=Exception("EXPLAIN not supported"))
 
         executor = SafeQueryExecutor(mock_session)
         result = await executor.execute_explain("SELECT * FROM users")
@@ -405,4 +403,3 @@ def test_injection_patterns_coverage(self) -> None:
 
         for pattern in INJECTION_PATTERNS:
             re.compile(pattern)  # Should not raise
-

tests/unit/test_schema.py

@@ -434,4 +434,3 @@ async def test_get_sample_data_error(self) -> None:
 
         # Should return empty list on error
         assert result == []
-