Function TestGetSchema_LimitForBulkModes

backend/api/mcp/tool_schema_test.go:559–587

(t *testing.T)

}

func TestGetSchema_LimitForBulkModes(t *testing.T) {
	resp := makeMetadataResponse(makeSchema("public",
		makeTable("users", 10, []map[string]any{makeColumn("id", "int4", false)}, nil),
	))

	// columns → limit=201
	s, mock := mockMetadataServer(t, employeeDB(), resp)
	_, _, err := s.handleGetSchema(testContext(), nil, SchemaInput{Database: "employee_db", Include: "columns"})
	require.NoError(t, err)
	require.Equal(t, int32(201), mock.lastCall().Limit)

	// details → limit=201
	s2, mock2 := mockMetadataServer(t, employeeDB(), resp)
	_, _, err = s2.handleGetSchema(testContext(), nil, SchemaInput{Database: "employee_db", Include: "details"})
	require.NoError(t, err)
	require.Equal(t, int32(201), mock2.lastCall().Limit)

	// summary → no limit
	s3, mock3 := mockMetadataServer(t, employeeDB(), resp)
	_, _, err = s3.handleGetSchema(testContext(), nil, SchemaInput{Database: "employee_db"})
	require.NoError(t, err)
	require.Equal(t, int32(0), mock3.lastCall().Limit)

	// table= → no limit
	s4, mock4 := mockMetadataServer(t, employeeDB(), resp)
	_, _, err = s4.handleGetSchema(testContext(), nil, SchemaInput{Database: "employee_db", Table: "users"})
	require.NoError(t, err)
	require.Equal(t, int32(0), mock4.lastCall().Limit)
}

func TestGetSchema_NotTruncatedAtExactly200(t *testing.T) {
	tables := make([]map[string]any, schemaTableLimit)