Stage 1 of caching layer. Framework (#6693)

1 files changed, 7 insertions, 7 deletions

diff --git a/store/sql_upgrade_test.go b/store/sql_upgrade_test.go
index 7a65e5b81..c2d28c90b 100644
--- a/store/sql_upgrade_test.go
+++ b/store/sql_upgrade_test.go
@@ -12,17 +12,17 @@ import (
 func TestStoreUpgrade(t *testing.T) {
 	Setup()
 
-	saveSchemaVersion(store.(*SqlStore), VERSION_3_0_0)
-	UpgradeDatabase(store.(*SqlStore))
+	saveSchemaVersion(store.(*LayeredStore).DatabaseLayer, VERSION_3_0_0)
+	UpgradeDatabase(store.(*LayeredStore).DatabaseLayer)
 
-	store.(*SqlStore).SchemaVersion = ""
-	UpgradeDatabase(store.(*SqlStore))
+	saveSchemaVersion(store.(*LayeredStore).DatabaseLayer, "")
+	UpgradeDatabase(store.(*LayeredStore).DatabaseLayer)
 }
 
 func TestSaveSchemaVersion(t *testing.T) {
 	Setup()
 
-	saveSchemaVersion(store.(*SqlStore), VERSION_3_0_0)
+	saveSchemaVersion(store.(*LayeredStore).DatabaseLayer, VERSION_3_0_0)
 	if result := <-store.System().Get(); result.Err != nil {
 		t.Fatal(result.Err)
 	} else {
@@ -32,9 +32,9 @@ func TestSaveSchemaVersion(t *testing.T) {
 		}
 	}
 
-	if store.(*SqlStore).SchemaVersion != VERSION_3_0_0 {
+	if store.(*LayeredStore).DatabaseLayer.GetCurrentSchemaVersion() != VERSION_3_0_0 {
 		t.Fatal("version not updated")
 	}
 
-	saveSchemaVersion(store.(*SqlStore), model.CurrentVersion)
+	saveSchemaVersion(store.(*LayeredStore).DatabaseLayer, model.CurrentVersion)
 }