go/bt/testutil/testutil.go

package bt_testutil

import (
	"fmt"

	"github.com/google/uuid"
	"github.com/stretchr/testify/require"
	"go.skia.org/infra/go/bt"
	"go.skia.org/infra/go/emulators/gcp_emulator"
	"go.skia.org/infra/go/sktest"
)

// SetupBigTable creates the given BigTable table and column families. Returns
// the project and instance names which can be passed to tests which use
// BigTable, and a cleanup function which should be deferred.
func SetupBigTable(t sktest.TestingT, tableID string, colFamilies ...string) (string, string, func()) {
	gcp_emulator.RequireBigTable(t)
	project := "test-project"
	instance := fmt.Sprintf("test-instance-%s", uuid.New())
	require.NoError(t, bt.InitBigtable(project, instance, tableID, colFamilies))
	return project, instance, func() {
		require.NoError(t, bt.DeleteTables(project, instance, tableID))
	}
}