Function TestMetadataGraphPopulatesAndLinksCodeSQLRefs

serv/metadata_test.go:16–125

(t *testing.T)

)

func TestMetadataGraphPopulatesAndLinksCodeSQLRefs(t *testing.T) {
	appPath := createMetadataAppDB(t)
	source := t.TempDir()
	writeTestFile(t, filepath.Join(source, "main.go"), `package main

func LookupUser() {
	query := "SELECT users.email FROM users WHERE users.id = ?"
	_ = query
}
`)

	conf := &Config{
		Core: Core{
			DisableAllowList: true,
			Sources: []core.SourceConfig{
				{Name: "app", Kind: "database", Type: "sqlite", Path: appPath, Default: true},
				{Name: "code", Kind: "code", Path: source},
				{Name: defaultMetadataDBName, Kind: "graphjin"},
			},
		},
		Serv: Serv{
			ConfigPath: filepath.Join(t.TempDir(), "config"),
			MCP:        MCPConfig{Disable: true},
		},
	}

	s, err := newGraphJinService(conf, nil)
	if err != nil {
		t.Fatal(err)
	}
	defer closeTestService(s)

	if s.metadataDB != defaultMetadataDBName {
		t.Fatalf("metadata database = %q, want %q", s.metadataDB, defaultMetadataDBName)
	}
	runtime := s.runtimeCore.Databases[defaultMetadataDBName]
	if runtime.Type != "nanodb" || !runtime.ReadOnly {
		t.Fatalf("metadata runtime = %+v, want read-only nanodb", runtime)
	}
	assertGraphJinTable(t, s, defaultMetadataDBName, "gj_catalog")
	assertGraphJinTable(t, s, defaultMetadataDBName, "gj_workflow")
	assertServiceCount(t, s, "code", `SELECT count(*) FROM gj_code WHERE kind = 'db_reference' AND db_object_id = 'app:main.users.email'`, 1)
	res, err := s.gj.GraphQL(sourceModeUserTestContext(), `query {
		gj_catalog(where: { kind: { eq: "column" }, database_name: { eq: "app" }, table_name: { eq: "users" }, column_name: { eq: "email" } }, limit: 1) {
			id
			kind
			database_name
			table_name
			column_name
			code_refs_id
			gj_code {
				kind
				path
				db_object_id
			}
		}
		directives: gj_catalog(where: { kind: { eq: "directive" }, title: { eq: "@running" } }, limit: 1) {
			id