Force state version to 0 for host api test runtime

lib/runtime/wasmer/config.go

@@ -8,17 +8,23 @@ import (
 	"github.com/ChainSafe/gossamer/lib/common"
 	"github.com/ChainSafe/gossamer/lib/keystore"
 	"github.com/ChainSafe/gossamer/lib/runtime"
+	"github.com/ChainSafe/gossamer/lib/trie"
 )
 
 // Config is the configuration used to create a
 // Wasmer runtime instance.
 type Config struct {
-	Storage     runtime.Storage
-	Keystore    *keystore.GlobalKeystore
-	LogLvl      log.Level
-	Role        common.Roles
-	NodeStorage runtime.NodeStorage
-	Network     runtime.BasicNetwork
-	Transaction runtime.TransactionState
-	CodeHash    common.Hash
+	Storage      runtime.Storage
+	Keystore     *keystore.GlobalKeystore
+	LogLvl       log.Level
+	Role         common.Roles
+	NodeStorage  runtime.NodeStorage
+	Network      runtime.BasicNetwork
+	Transaction  runtime.TransactionState
+	CodeHash     common.Hash
+	testVersions *testVersions
+}
+
+type testVersions struct {
+	stateVersion trie.Version
 }

lib/runtime/wasmer/instance.go

@@ -109,19 +109,33 @@ func NewInstance(code []byte, cfg Config) (instance *Instance, err error) {
 		codeHash: cfg.CodeHash,
 	}
 
-	// Find runtime instance state version and cache it in its
-	// instance context.
-	version, err := instance.Version()
+	stateVersion, err := getStateVersion(instance, cfg.testVersions)
 	if err != nil {
 		instance.close()
-		return nil, fmt.Errorf("getting instance version: %w", err)
+		return nil, fmt.Errorf("getting state version: %w", err)
 	}
-	instance.ctx.StateVersion = trie.Version(version.StateVersion)
+
+	instance.ctx.StateVersion = stateVersion
 	wasmInstance.SetContextData(instance.ctx)
 
 	return instance, nil
 }
 
+func getStateVersion(instance *Instance, testVersions *testVersions) (
+	stateVersion trie.Version, err error) {
+	if testVersions != nil {
+		return testVersions.stateVersion, nil
+	}
+
+	// Find runtime state version from running instance
+	version, err := instance.Version()
+	if err != nil {
+		return stateVersion, fmt.Errorf("getting instance version: %w", err)
+	}
+
+	return trie.Version(version.StateVersion), nil
+}
+
 // decompressWasm decompresses a Wasm blob that may or may not be compressed with zstd
 // ref: https://github.com/paritytech/substrate/blob/master/primitives/maybe-compressed-blob/src/lib.rs
 func decompressWasm(code []byte) ([]byte, error) {

lib/runtime/wasmer/test_helpers.go

@@ -31,7 +31,7 @@ func NewTestInstance(t *testing.T, targetRuntime string) *Instance {
 func NewTestInstanceWithTrie(t *testing.T, targetRuntime string, tt *trie.Trie) *Instance {
 	t.Helper()
 
-	cfg := setupConfig(t, tt, DefaultTestLogLvl, common.NoNetworkRole)
+	cfg := setupConfig(t, tt, DefaultTestLogLvl, common.NoNetworkRole, targetRuntime)
 	runtimeFilepath, err := runtime.GetRuntime(context.Background(), targetRuntime)
 	require.NoError(t, err)
 
@@ -40,7 +40,8 @@ func NewTestInstanceWithTrie(t *testing.T, targetRuntime string, tt *trie.Trie)
 	return r
 }
 
-func setupConfig(t *testing.T, tt *trie.Trie, lvl log.Level, role common.Roles) Config {
+func setupConfig(t *testing.T, tt *trie.Trie, lvl log.Level,
+	role common.Roles, targetRuntime string) Config {
 	t.Helper()
 
 	s := storage.NewTrieState(tt)
@@ -51,14 +52,25 @@ func setupConfig(t *testing.T, tt *trie.Trie, lvl log.Level, role common.Roles)
 		BaseDB:            runtime.NewInMemoryDB(t), // we're using a local storage here since this is a test runtime
 	}
 
+	var versions *testVersions
+	if targetRuntime == runtime.HOST_API_TEST_RUNTIME {
+		// Force state version to 0 since the host api test runtime
+		// does not implement the Core_version call so we cannot get the
+		// state version from it.
+		versions = &testVersions{
+			stateVersion: 0,
+		}
+	}
+
 	return Config{
-		Storage:     s,
-		Keystore:    keystore.NewGlobalKeystore(),
-		LogLvl:      lvl,
-		NodeStorage: ns,
-		Network:     new(runtime.TestRuntimeNetwork),
-		Transaction: newTransactionStateMock(),
-		Role:        role,
+		Storage:      s,
+		Keystore:     keystore.NewGlobalKeystore(),
+		LogLvl:       lvl,
+		NodeStorage:  ns,
+		Network:      new(runtime.TestRuntimeNetwork),
+		Transaction:  newTransactionStateMock(),
+		Role:         role,
+		testVersions: versions,
 	}
 }