MCPcopy Create free account
hub / github.com/PostHog/duckgres / TestInitConnsMap

Function TestInitConnsMap

server/conn_test.go:3929–3943  ·  view source on GitHub ↗
(t *testing.T)

Source from the content-addressed store, hash-verified

3927 otherProject := &clientConn{username: "project-reader-b", orgID: "org-a", pid: 102}
3928 otherProject.currentQuery.Store("same-org-hidden-marker")
3929 otherOrg := &clientConn{username: "project-reader-a", orgID: "org-b", pid: 103}
3930 otherOrg.currentQuery.Store("other-org-hidden-marker")
3931
3932 for _, conn := range []*clientConn{reader, sameProject, otherProject, otherOrg} {
3933 srv.registerConn(conn)
3934 }
3935
3936 query := "WITH pg_stat_activity AS (SELECT * FROM team_1.events) SELECT * FROM pg_stat_activity"
3937 if err := reader.handleQuery([]byte(query + "\x00")); err != nil {
3938 t.Fatalf("handleQuery: %v", err)
3939 }
3940
3941 if !bytes.Contains(output.Bytes(), []byte("same-project-visible-marker")) {
3942 t.Fatal("project reader should see sessions for its own project credential")
3943 }
3944 for _, hidden := range []string{"same-org-hidden-marker", "other-org-hidden-marker"} {
3945 if bytes.Contains(output.Bytes(), []byte(hidden)) {
3946 t.Fatalf("project reader saw another project's pg_stat_activity row %q", hidden)

Callers

nothing calls this directly

Calls 2

initConnsMapMethod · 0.95
registerConnMethod · 0.95

Tested by

no test coverage detected