MCPcopy Create free account
hub / github.com/bytebase/bytebase / TestQueryDatabase_EmptyResult

Function TestQueryDatabase_EmptyResult

backend/api/mcp/tool_query_test.go:494–518  ·  view source on GitHub ↗
(t *testing.T)

Source from the content-addressed store, hash-verified

492}
493
494func TestQueryDatabase_EmptyResult(t *testing.T) {
495 databases := []map[string]any{
496 makeDatabase("instances/prod-pg/databases/employee_db", "instances/prod-pg", "projects/hr-system", "POSTGRES", "ds-admin-1"),
497 }
498 qr := makeQueryResponse(
499 []string{"id", "name"},
500 []string{"int4", "varchar"},
501 [][]any{},
502 "0.005s",
503 )
504 s := newTestServerWithMock(t, mockQueryServer(databases, qr))
505
506 result, structured, err := s.handleQueryDatabase(testContext(), nil, QueryInput{
507 Database: "employee_db",
508 Statement: "SELECT * FROM users WHERE id = -1",
509 })
510 require.NoError(t, err)
511 require.False(t, result.IsError)
512
513 output, ok := structured.(*QueryOutput)
514 require.True(t, ok)
515 require.Empty(t, output.Rows)
516 require.Equal(t, 0, output.RowCount)
517 require.False(t, output.Truncated)
518}
519
520func TestQueryDatabase_Truncation(t *testing.T) {
521 databases := []map[string]any{

Callers

nothing calls this directly

Calls 7

makeQueryResponseFunction · 0.85
newTestServerWithMockFunction · 0.85
mockQueryServerFunction · 0.85
testContextFunction · 0.85
handleQueryDatabaseMethod · 0.80
makeDatabaseFunction · 0.70
EqualMethod · 0.65

Tested by

no test coverage detected