Function TestGetSchema_IncludeColumns

backend/api/mcp/tool_schema_test.go:236–278

(t *testing.T)

}

func TestGetSchema_IncludeColumns(t *testing.T) {
	usersTable := makeTable("users", 10432,
		[]map[string]any{
			makeColumn("id", "int4", false),
			makeColumn("email", "varchar(255)", false),
			makeColumn("created_at", "timestamptz", true),
		},
		[]map[string]any{makePrimaryKeyIndex("users_pkey", []string{"id"})},
	)
	resp := makeMetadataResponse(makeSchema("public", usersTable))

	s, mock := mockMetadataServer(t, employeeDB(), resp)

	result, structured, err := s.handleGetSchema(testContext(), nil, SchemaInput{
		Database: "employee_db",
		Include:  "columns",
	})
	require.NoError(t, err)
	require.False(t, result.IsError)

	output, ok := structured.(*SchemaOutput)
	require.True(t, ok)
	users := output.Schemas[0].Tables[0]
	require.Len(t, users.Columns, 3)
	// Proto order preserved (id, email, created_at) — NOT alphabetized.
	require.Equal(t, "id", users.Columns[0].Name)
	require.Equal(t, "int4", users.Columns[0].Type)
	require.False(t, users.Columns[0].Nullable)
	require.True(t, users.Columns[0].PrimaryKey)
	require.Equal(t, "email", users.Columns[1].Name)
	require.False(t, users.Columns[1].PrimaryKey)
	require.Equal(t, "created_at", users.Columns[2].Name)
	require.True(t, users.Columns[2].Nullable)
	// columns mode has no defaults/comments populated.
	require.Empty(t, users.Columns[0].Default)
	require.Empty(t, users.Columns[0].Comment)
	// No indexes or FKs in columns mode.
	require.Empty(t, users.Indexes)
	require.Empty(t, users.ForeignKeys)

	// Request includes limit=201.
	require.Equal(t, int32(201), mock.lastCall().Limit)
}

func TestGetSchema_IncludeDetails(t *testing.T) {
	ordersTable := map[string]any{