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)
+	})
 }