focalboard/server/services/store/sqlstore/sqlstore_test.go

package sqlstore

import (
	"os"
	"testing"

	"github.com/mattermost/mattermost-octo-tasks/server/model"
	"github.com/stretchr/testify/require"
)

func SetupTests(t *testing.T) (*SQLStore, func()) {
	dbType := os.Getenv("OT_STORE_TEST_DB_TYPE")
	if dbType == "" {
		dbType = "sqlite3"
	}

	connectionString := os.Getenv("OT_STORE_TEST_CONN_STRING")
	if connectionString == "" {
		connectionString = ":memory:"
	}

	store, err := New(dbType, connectionString)
	require.Nil(t, err)

	tearDown := func() {
		err = store.Shutdown()
		require.Nil(t, err)
	}

	return store, tearDown
}

func InsertBlocks(t *testing.T, s *SQLStore, blocks []model.Block) {
	for _, block := range blocks {
		err := s.InsertBlock(block)
		require.NoError(t, err)
	}
}

func DeleteBlocks(t *testing.T, s *SQLStore, blocks []model.Block) {
	for _, block := range blocks {
		err := s.DeleteBlock(block.ID)
		require.NoError(t, err)
	}
}

func ContainsBlockWithID(blocks []model.Block, blockID string) bool {
	for _, block := range blocks {
		if block.ID == blockID {
			return true
		}
	}

	return false
}