Function TestGetSchema_Summary

backend/api/mcp/tool_schema_test.go:186–234

--- Tests ---

(t *testing.T)

// --- Tests ---

func TestGetSchema_Summary(t *testing.T) {
	usersTable := makeTable("users", 10432,
		[]map[string]any{
			makeColumn("id", "int4", false),
			makeColumn("email", "varchar(255)", false),
		},
		[]map[string]any{makePrimaryKeyIndex("users_pkey", []string{"id"})},
	)
	ordersTable := makeTable("orders", 182394,
		[]map[string]any{
			makeColumn("id", "int8", false),
			makeColumn("user_id", "int4", false),
		},
		[]map[string]any{makePrimaryKeyIndex("orders_pkey", []string{"id"})},
	)
	// Intentionally out of order to verify sorting.
	resp := makeMetadataResponse(makeSchemaWithViews("public", []string{"active_users", "recent_orders"}, ordersTable, usersTable))

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

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

	output, ok := structured.(*SchemaOutput)
	require.True(t, ok)
	require.Equal(t, "POSTGRES", output.Engine)
	require.Len(t, output.Schemas, 1)
	section := output.Schemas[0]
	require.Equal(t, "public", section.Name)
	require.Len(t, section.Tables, 2)
	// Tables sorted alphabetically.
	require.Equal(t, "orders", section.Tables[0].Name)
	require.Equal(t, "users", section.Tables[1].Name)
	// Summary mode has no column details but includes column count.
	require.Equal(t, 2, section.Tables[0].ColumnCount)
	require.Equal(t, int64(182394), section.Tables[0].RowCount)
	require.Empty(t, section.Tables[0].Columns)
	// Views sorted alphabetically.
	require.Equal(t, []string{"active_users", "recent_orders"}, section.Views)

	// Text header mentions schema and tables.
	text := result.Content[0].(*mcpsdk.TextContent).Text
	require.Contains(t, text, "Database: instances/prod-pg/databases/employee_db (POSTGRES)")
	require.Contains(t, text, "Schemas: 1 (public)")
	require.Contains(t, text, "Tables: 2")
}

func TestGetSchema_IncludeColumns(t *testing.T) {
	usersTable := makeTable("users", 10432,