55 lines
1.1 KiB
Go
55 lines
1.1 KiB
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
|
|
}
|