diff --git a/server/server_test.go b/server/server_test.go index 210e58caed3f8..f9d22e866458a 100644 --- a/server/server_test.go +++ b/server/server_test.go @@ -779,7 +779,7 @@ func (cli *testServerClient) runTestLoadDataForListColumnPartition2(t *testing.T }) } -func (cli *testServerClient) checkRows(t *testing.T, rows *sql.Rows, expectedRows ...string) { +func (cli *testServerClient) Rows(t *testing.T, rows *sql.Rows) []string { buf := bytes.NewBuffer(nil) result := make([]string, 0, 2) for rows.Next() { @@ -806,7 +806,11 @@ func (cli *testServerClient) checkRows(t *testing.T, rows *sql.Rows, expectedRow } result = append(result, buf.String()) } + return result +} +func (cli *testServerClient) checkRows(t *testing.T, rows *sql.Rows, expectedRows ...string) { + result := cli.Rows(t, rows) require.Equal(t, strings.Join(expectedRows, "\n"), strings.Join(result, "\n")) } diff --git a/server/tidb_test.go b/server/tidb_test.go index 01a19d70df6d1..2536752d3b0fe 100644 --- a/server/tidb_test.go +++ b/server/tidb_test.go @@ -528,6 +528,9 @@ func TestSocketAndIp(t *testing.T) { cli.checkRows(t, rows, "user1@127.0.0.1") rows = dbt.MustQuery("show grants") cli.checkRows(t, rows, "GRANT USAGE ON *.* TO 'user1'@'%'\nGRANT SELECT ON test.* TO 'user1'@'%'") + rows = dbt.MustQuery("select host from information_schema.processlist where user = 'user1'") + records := cli.Rows(t, rows) + require.Contains(t, records[0], ":", "Missing : in is.processlist") }) // Test with unix domain socket file connection with all hosts cli.runTests(t, func(config *mysql.Config) {