mirror of
https://github.com/empayre/fleet.git
synced 2024-11-07 17:28:54 +00:00
34 lines
767 B
Go
34 lines
767 B
Go
package datastore
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func testMigrationStatus(t *testing.T, ds fleet.Datastore) {
|
|
if ds.Name() == "inmem" {
|
|
t.Skip("inmem is being deprecated, test skipped")
|
|
}
|
|
|
|
require.Nil(t, ds.Drop())
|
|
|
|
status, err := ds.MigrationStatus()
|
|
require.Nil(t, err)
|
|
assert.EqualValues(t, fleet.NoMigrationsCompleted, status)
|
|
|
|
require.Nil(t, ds.MigrateTables())
|
|
|
|
status, err = ds.MigrationStatus()
|
|
require.Nil(t, err)
|
|
assert.EqualValues(t, fleet.SomeMigrationsCompleted, status)
|
|
|
|
require.Nil(t, ds.MigrateData())
|
|
|
|
status, err = ds.MigrationStatus()
|
|
require.Nil(t, err)
|
|
assert.EqualValues(t, fleet.AllMigrationsCompleted, status)
|
|
}
|