Function TestGetSchema_IncludeDetails

backend/api/mcp/tool_schema_test.go:280–330

(t *testing.T)

}

func TestGetSchema_IncludeDetails(t *testing.T) {
	ordersTable := map[string]any{
		"name":     "orders",
		"rowCount": "182394",
		"comment":  "Customer orders",
		"columns": []map[string]any{
			makeColumnWithDefault("id", "int8", "", "Order ID"),
			makeColumnWithDefault("user_id", "int4", "", "Customer ref"),
			makeColumnWithDefault("status", "varchar(32)", "'pending'", "Status flag"),
		},
		"indexes": []map[string]any{
			// Intentionally out of order.
			makeIndex("orders_user_id_idx", false, []string{"user_id"}),
			makePrimaryKeyIndex("orders_pkey", []string{"id"}),
		},
		"foreignKeys": []map[string]any{
			{
				"name":              "orders_user_id_fkey",
				"columns":           []string{"user_id"},
				"referencedTable":   "users",
				"referencedColumns": []string{"id"},
			},
		},
	}
	resp := makeMetadataResponse(makeSchema("public", ordersTable))

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

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

	output, ok := structured.(*SchemaOutput)
	require.True(t, ok)
	orders := output.Schemas[0].Tables[0]
	require.Equal(t, "Customer orders", orders.Comment)
	// Column defaults and comments populated.
	require.Equal(t, "'pending'", orders.Columns[2].Default)
	require.Equal(t, "Status flag", orders.Columns[2].Comment)
	// Indexes sorted alphabetically.
	require.Len(t, orders.Indexes, 2)
	require.Equal(t, "orders_pkey", orders.Indexes[0].Name)
	require.Equal(t, "orders_user_id_idx", orders.Indexes[1].Name)
	// FKs present.
	require.Len(t, orders.ForeignKeys, 1)
	require.Equal(t, "orders_user_id_fkey", orders.ForeignKeys[0].Name)
	require.Equal(t, "users", orders.ForeignKeys[0].ReferencedTable)
}

func TestGetSchema_TableFilter_ServerSideFilter(t *testing.T) {
	ordersTable := makeTable("orders", 182394,