--- Task 6 tests: masked values and timeout ---
(t *testing.T)
| 657 | // --- Task 6 tests: masked values and timeout --- |
| 658 | |
| 659 | func TestQueryDatabase_MaskedValues(t *testing.T) { |
| 660 | databases := []map[string]any{ |
| 661 | makeDatabase("instances/prod-pg/databases/employee_db", "instances/prod-pg", "projects/hr-system", "POSTGRES", "ds-admin-1"), |
| 662 | } |
| 663 | qr := makeQueryResponse( |
| 664 | []string{"id", "ssn", "name"}, |
| 665 | []string{"int4", "varchar", "varchar"}, |
| 666 | [][]any{{"1", "******", "**rn**"}}, |
| 667 | "0.010s", |
| 668 | ) |
| 669 | s := newTestServerWithMock(t, mockQueryServer(databases, qr)) |
| 670 | |
| 671 | _, structured, err := s.handleQueryDatabase(testContext(), nil, QueryInput{ |
| 672 | Database: "employee_db", |
| 673 | Statement: "SELECT id, ssn, name FROM users", |
| 674 | }) |
| 675 | require.NoError(t, err) |
| 676 | |
| 677 | output, ok := structured.(*QueryOutput) |
| 678 | require.True(t, ok) |
| 679 | require.Equal(t, "******", output.Rows[0][1]) |
| 680 | require.Equal(t, "**rn**", output.Rows[0][2]) |
| 681 | } |
| 682 | |
| 683 | func TestQueryDatabase_Timeout(t *testing.T) { |
| 684 | databases := []map[string]any{ |
nothing calls this directly
no test coverage detected