| package btts_testutils |
| |
| import ( |
| "context" |
| |
| "cloud.google.com/go/bigtable" |
| "github.com/stretchr/testify/require" |
| "go.skia.org/infra/go/sktest" |
| "golang.org/x/oauth2" |
| ) |
| |
| func CreateTestTable(t sktest.TestingT) { |
| ctx := context.Background() |
| client, _ := bigtable.NewAdminClient(ctx, "test", "test") |
| err := client.CreateTableFromConf(ctx, &bigtable.TableConf{ |
| TableID: "test", |
| Families: map[string]bigtable.GCPolicy{ |
| "V": bigtable.MaxVersionsPolicy(1), |
| "S": bigtable.MaxVersionsPolicy(1), |
| "D": bigtable.MaxVersionsPolicy(1), |
| "H": bigtable.MaxVersionsPolicy(1), |
| "I": bigtable.MaxVersionsPolicy(1), |
| }, |
| }) |
| require.NoError(t, err) |
| } |
| |
| func CleanUpTestTable(t sktest.TestingT) { |
| ctx := context.Background() |
| client, _ := bigtable.NewAdminClient(ctx, "test", "test") |
| err := client.DeleteTable(ctx, "test") |
| require.NoError(t, err) |
| } |
| |
| type MockTS struct{} |
| |
| func (t *MockTS) Token() (*oauth2.Token, error) { |
| return nil, nil |
| } |