diff options
Diffstat (limited to 'store/sqlstore/upgrade_test.go')
-rw-r--r-- | store/sqlstore/upgrade_test.go | 40 |
1 files changed, 20 insertions, 20 deletions
diff --git a/store/sqlstore/upgrade_test.go b/store/sqlstore/upgrade_test.go index d8d6a9e9d..7c3b2f124 100644 --- a/store/sqlstore/upgrade_test.go +++ b/store/sqlstore/upgrade_test.go @@ -11,31 +11,31 @@ import ( ) func TestStoreUpgrade(t *testing.T) { - ss := Setup() + StoreTest(t, func(t *testing.T, ss store.Store) { + saveSchemaVersion(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore), VERSION_3_0_0) + UpgradeDatabase(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore)) - saveSchemaVersion(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore), VERSION_3_0_0) - UpgradeDatabase(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore)) - - saveSchemaVersion(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore), "") - UpgradeDatabase(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore)) + saveSchemaVersion(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore), "") + UpgradeDatabase(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore)) + }) } func TestSaveSchemaVersion(t *testing.T) { - ss := Setup() - - saveSchemaVersion(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore), VERSION_3_0_0) - if result := <-ss.System().Get(); result.Err != nil { - t.Fatal(result.Err) - } else { - props := result.Data.(model.StringMap) - if props["Version"] != VERSION_3_0_0 { - t.Fatal("version not updated") + StoreTest(t, func(t *testing.T, ss store.Store) { + saveSchemaVersion(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore), VERSION_3_0_0) + if result := <-ss.System().Get(); result.Err != nil { + t.Fatal(result.Err) + } else { + props := result.Data.(model.StringMap) + if props["Version"] != VERSION_3_0_0 { + t.Fatal("version not updated") + } } - } - if ss.(*store.LayeredStore).DatabaseLayer.(SqlStore).GetCurrentSchemaVersion() != VERSION_3_0_0 { - t.Fatal("version not updated") - } + if ss.(*store.LayeredStore).DatabaseLayer.(SqlStore).GetCurrentSchemaVersion() != VERSION_3_0_0 { + t.Fatal("version not updated") + } - saveSchemaVersion(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore), model.CurrentVersion) + saveSchemaVersion(ss.(*store.LayeredStore).DatabaseLayer.(SqlStore), model.CurrentVersion) + }) } |