MCPcopy
hub / github.com/pingcap/tidb / TestGetSpecifiedColumnValueAndClose

Function TestGetSpecifiedColumnValueAndClose

dumpling/export/sql_test.go:1853–1881  ·  view source on GitHub ↗
(t *testing.T)

Source from the content-addressed store, hash-verified

1851}
1852
1853func TestGetSpecifiedColumnValueAndClose(t *testing.T) {
1854 db, mock, err := sqlmock.New()
1855 require.NoError(t, err)
1856 defer func() {
1857 require.NoError(t, db.Close())
1858 }()
1859 ctx := context.Background()
1860 conn, err := db.Conn(ctx)
1861 require.NoError(t, err)
1862
1863 mock.ExpectQuery("SHOW BINARY LOGS").
1864 WillReturnRows(sqlmock.NewRows([]string{"Log_name", "File_size"}).
1865 AddRow("mysql-bin.000001", 52119).
1866 AddRow("mysql-bin.000002", 114))
1867
1868 query := "SHOW BINARY LOGS"
1869 rows, err := conn.QueryContext(ctx, query)
1870 require.NoError(t, err)
1871 defer rows.Close()
1872 var rowsResult []string
1873 rowsResult, err = GetSpecifiedColumnValueAndClose(rows, "Log_name")
1874 require.NoError(t, err)
1875 require.Equal(t, 2, len(rowsResult))
1876 require.Equal(t, "mysql-bin.000001", rowsResult[0])
1877 require.Equal(t, "mysql-bin.000002", rowsResult[1])
1878
1879 err = mock.ExpectationsWereMet()
1880 require.NoError(t, err)
1881}
1882
1883func TestGetSpecifiedColumnValuesAndClose(t *testing.T) {
1884 db, mock, err := sqlmock.New()

Callers

nothing calls this directly

Calls 6

BackgroundFunction · 0.92
NewMethod · 0.65
CloseMethod · 0.65
QueryContextMethod · 0.65
EqualMethod · 0.65

Tested by

no test coverage detected