Method test_edge_cases

tests/_ast/test_sql_utils.py:209–233

Test edge cases and unusual inputs.

(self)

        assert classify_sql_statement(complex_dql) == "DQL"

    def test_edge_cases(self):
        """Test edge cases and unusual inputs."""
        # Empty string
        assert classify_sql_statement("") == "unknown"

        # Whitespace only
        assert classify_sql_statement("   ") == "unknown"

        # Invalid SQL
        assert classify_sql_statement("INVALID SQL STATEMENT") == "unknown"
        assert (
            classify_sql_statement("SELECT * FROM") == "unknown"
        )  # Incomplete statement

        # SQL with comments
        assert (
            classify_sql_statement("-- This is a comment\nSELECT * FROM users")
            == "DQL"
        )
        assert (
            classify_sql_statement(
                "/* Multi-line comment */\nCREATE TABLE users (id INT)"
            )
            == "DDL"
        )

    def test_multiple_statements(self):
        """Test that the function handles the first statement in a batch."""