Function TestJDBCDatabaseMetaDataQueries

tests/integration/jdbc_test.go:191–249

TestJDBCDatabaseMetaDataQueries tests queries similar to what JDBC DatabaseMetaData methods generate (getTables, getSchemas, etc.).

(t *testing.T)

// TestJDBCDatabaseMetaDataQueries tests queries similar to what JDBC
// DatabaseMetaData methods generate (getTables, getSchemas, etc.).
func TestJDBCDatabaseMetaDataQueries(t *testing.T) {
	// These are representative of what the PostgreSQL JDBC driver generates
	// for DatabaseMetaData calls
	metadataQueries := []struct {
		name string
		sql  string
	}{
		{
			"getTables",
			`SELECT NULL AS TABLE_CAT, n.nspname AS TABLE_SCHEM, c.relname AS TABLE_NAME,
			 CASE c.relkind WHEN 'r' THEN 'TABLE' WHEN 'v' THEN 'VIEW' END AS TABLE_TYPE
			 FROM pg_catalog.pg_class c
			 LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
			 WHERE c.relkind IN ('r','v')
			 AND n.nspname NOT IN ('pg_catalog', 'information_schema')
			 ORDER BY TABLE_SCHEM, TABLE_NAME
			 LIMIT 20`,
		},
		{
			"getSchemas",
			`SELECT nspname AS TABLE_SCHEM, NULL AS TABLE_CATALOG
			 FROM pg_catalog.pg_namespace
			 WHERE nspname NOT LIKE 'pg_%' AND nspname != 'information_schema'
			 ORDER BY TABLE_SCHEM`,
		},
		{
			"getCatalogs",
			`SELECT datname AS TABLE_CAT FROM pg_catalog.pg_database ORDER BY 1`,
		},
		{
			"getTypeInfo",
			`SELECT typname FROM pg_catalog.pg_type LIMIT 20`,
		},
	}

	for _, q := range metadataQueries {
		t.Run(q.name, func(t *testing.T) {
			ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
			defer cancel()

			conn := pgxConnect(t)
			defer func() { _ = conn.Close(ctx) }()

			rows, err := conn.Query(ctx, q.sql)
			if err != nil {
				t.Fatalf("Query failed: %v", err)
			}
			count := 0
			for rows.Next() {
				count++
			}
			if err := rows.Err(); err != nil {
				t.Fatalf("Row iteration error: %v", err)
			}
			rows.Close()
			t.Logf("%s returned %d rows", q.name, count)
		})
	}