52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"database/sql"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/mattermost/focalboard/server/services/store"
|
|
"github.com/mattermost/focalboard/server/services/store/storetests"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mattermost/mattermost-server/v6/shared/mlog"
|
|
)
|
|
|
|
func SetupTests(t *testing.T) (store.Store, func()) {
|
|
dbType := os.Getenv("FB_STORE_TEST_DB_TYPE")
|
|
if dbType == "" {
|
|
dbType = sqliteDBType
|
|
}
|
|
|
|
connectionString := os.Getenv("FB_STORE_TEST_CONN_STRING")
|
|
if connectionString == "" {
|
|
connectionString = ":memory:"
|
|
}
|
|
|
|
logger := mlog.CreateConsoleTestLogger(false, mlog.LvlDebug)
|
|
|
|
sqlDB, err := sql.Open(dbType, connectionString)
|
|
require.NoError(t, err)
|
|
err = sqlDB.Ping()
|
|
require.NoError(t, err)
|
|
store, err := New(dbType, connectionString, "test_", logger, sqlDB)
|
|
require.Nil(t, err)
|
|
|
|
tearDown := func() {
|
|
defer func() { _ = logger.Shutdown() }()
|
|
err = store.Shutdown()
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
return store, tearDown
|
|
}
|
|
|
|
func TestBlocksStore(t *testing.T) {
|
|
t.Run("BlocksStore", func(t *testing.T) { storetests.StoreTestBlocksStore(t, SetupTests) })
|
|
t.Run("SharingStore", func(t *testing.T) { storetests.StoreTestSharingStore(t, SetupTests) })
|
|
t.Run("SystemStore", func(t *testing.T) { storetests.StoreTestSystemStore(t, SetupTests) })
|
|
t.Run("UserStore", func(t *testing.T) { storetests.StoreTestUserStore(t, SetupTests) })
|
|
t.Run("SessionStore", func(t *testing.T) { storetests.StoreTestSessionStore(t, SetupTests) })
|
|
t.Run("WorkspaceStore", func(t *testing.T) { storetests.StoreTestWorkspaceStore(t, SetupTests) })
|
|
}
|