task_scheduler/go/isolate_cache/testutils.go

package isolate_cache

import (
	"github.com/stretchr/testify/require"
	"go.skia.org/infra/go/bt"
	bt_testutil "go.skia.org/infra/go/bt/testutil"
	"go.skia.org/infra/go/sktest"
)

// SetupBigTable performs setup in BigTable. Returns the BigTable project and
// instance name which should be used to instantiate Cache, and a cleanup
// function which should be deferred.
func SetupBigTable(t sktest.TestingT) (string, string, func()) {
	return bt_testutil.SetupBigTable(t, BT_TABLE, BT_COLUMN_FAMILY)
}

// SetupSharedBigTable performs setup in BigTable, using the given project and
// BigTable instance. This is useful when a given test has multiple entities
// which are backed by BigTable and should use the same instance. Returns a
// cleanup function which should be deferred.
func SetupSharedBigTable(t sktest.TestingT, project, instance string) func() {
	require.NoError(t, bt.InitBigtable(project, instance, BT_TABLE, []string{BT_COLUMN_FAMILY}))
	return func() {
		require.NoError(t, bt.DeleteTables(project, instance, BT_TABLE))
	}
}