Function TestTrinoQueryType

backend/plugin/parser/trino/parser_test.go:39–94

(t *testing.T)

}

func TestTrinoQueryType(t *testing.T) {
	testCases := []struct {
		sql      string
		expected base.QueryType
	}{
		{
			sql:      "SELECT * FROM system.runtime.nodes;",
			expected: base.SelectInfoSchema,
		},
		{
			sql:      "SELECT * FROM users;",
			expected: base.Select,
		},
		{
			sql:      "INSERT INTO users (id, name) VALUES (1, 'John');",
			expected: base.DML,
		},
		{
			sql:      "UPDATE users SET name = 'Jane' WHERE id = 1;",
			expected: base.DML,
		},
		{
			sql:      "DELETE FROM users WHERE id = 1;",
			expected: base.DML,
		},
		{
			sql:      "CREATE TABLE users (id INT, name VARCHAR);",
			expected: base.DDL,
		},
		{
			sql:      "DROP TABLE users;",
			expected: base.DDL,
		},
		{
			sql:      "EXPLAIN SELECT * FROM users;",
			expected: base.Explain,
		},
		{
			sql:      "EXPLAIN ANALYZE SELECT * FROM users;",
			expected: base.Select, // EXPLAIN ANALYZE executes the query, so it is classified as Select.
		},
		{
			sql:      "SHOW TABLES;",
			expected: base.SelectInfoSchema,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.sql, func(t *testing.T) {
			// queryTypeFromText applies the same system-schema promotion the
			// query-span path uses, so a SELECT over system.* is SelectInfoSchema.
			queryType := queryTypeFromText(tc.sql)
			assert.Equal(t, tc.expected, queryType)
		})
	}
}

func TestMain(m *testing.M) {