| 30 | } |
| 31 | |
| 32 | func RequireOLAPTableCount(t testing.TB, rt *runtime.Runtime, id, name string, count int) { |
| 33 | ctx := t.Context() |
| 34 | olap, release, err := rt.OLAP(ctx, id, "") |
| 35 | require.NoError(t, err) |
| 36 | defer release() |
| 37 | |
| 38 | _, err = olap.InformationSchema().Lookup(ctx, "", "", name) |
| 39 | require.NoError(t, err) |
| 40 | |
| 41 | rows, err := olap.Query(ctx, &drivers.Statement{Query: fmt.Sprintf(`SELECT count(*) FROM %s`, olap.Dialect().EscapeIdentifier(name))}) |
| 42 | require.NoError(t, err) |
| 43 | defer rows.Close() |
| 44 | |
| 45 | var n int |
| 46 | for rows.Next() { |
| 47 | err := rows.Scan(&n) |
| 48 | require.NoError(t, err) |
| 49 | } |
| 50 | require.NoError(t, rows.Err()) |
| 51 | require.Equal(t, count, n) |
| 52 | } |
| 53 | |
| 54 | func RequireIsView(t testing.TB, olap drivers.OLAPStore, tableName string, isView bool) { |
| 55 | table, err := olap.InformationSchema().Lookup(t.Context(), "", "", tableName) |