Function TestConnectionRegistry

server/conn_test.go:3853–3906

(t *testing.T)

		{"multiline", "SELECT pid\nFROM pg_stat_activity\nWHERE state = 'active'", true},
		{"tab before table", "SELECT * FROM\tpg_stat_activity", true},
		{"multiple spaces", "SELECT * FROM   pg_stat_activity", true},

		// Should not match
		{"no FROM", "SELECT 'pg_stat_activity'", false},
		{"in string literal", "SELECT 'text pg_stat_activity text'", false},
		{"different table", "SELECT * FROM pg_stat_statements", false},
		{"substring match", "SELECT * FROM pg_stat_activity_detail", false},
		{"prefix only", "SELECT * FROM my_pg_stat_activity", false},
		{"empty query", "", false},
		{"INSERT", "INSERT INTO pg_stat_activity VALUES (1)", false},
		{"CREATE VIEW", "CREATE VIEW v AS SELECT * FROM pg_stat_activity", true}, // this references FROM, so it matches
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := matchPgStatActivityQuery(tt.query)
			if got != tt.want {
				t.Errorf("matchPgStatActivityQuery(%q) = %v, want %v", tt.query, got, tt.want)
			}
		})
	}
}

func TestPgStatActivityColumnsAndOIDs(t *testing.T) {
	// Verify that pgStatActivityColumns and pgStatActivityTypeOIDs are consistent
	if len(pgStatActivityTypeOIDs) != len(pgStatActivityColumns) {
		t.Fatalf("pgStatActivityTypeOIDs length %d != pgStatActivityColumns length %d",
			len(pgStatActivityTypeOIDs), len(pgStatActivityColumns))
	}

	for i, col := range pgStatActivityColumns {
		if pgStatActivityTypeOIDs[i] != col.oid {
			t.Errorf("pgStatActivityTypeOIDs[%d] = %d, want %d (column %s)",
				i, pgStatActivityTypeOIDs[i], col.oid, col.name)
		}
	}

	// Verify expected columns are present
	expectedColumns := map[string]bool{
		"datid": false, "datname": false, "pid": false, "usesysid": false,
		"usename": false, "application_name": false, "client_addr": false,
		"client_port": false, "backend_start": false, "state": false,
		"query": false, "backend_type": false, "worker_id": false,
	}
	for _, col := range pgStatActivityColumns {
		if _, ok := expectedColumns[col.name]; ok {
			expectedColumns[col.name] = true
		}
	}
	for name, found := range expectedColumns {
		if !found {
			t.Errorf("expected column %q not found in pgStatActivityColumns", name)
		}
	}
}

func TestProjectReaderPgStatActivityCTECannotSeeOtherConnections(t *testing.T) {