feat: add raw_get to vm::testing::Contract

Finschia · Apr 6, 2022 · b196b94 · b196b94
1 parent abc91a4
commit b196b94
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/packages/vm/src/testing/contract.rs b/packages/vm/src/testing/contract.rs
@@ -1,6 +1,6 @@
 use wasmer::Module;
 
-use crate::backend::Backend;
+use crate::backend::{Backend, Storage};
 use crate::compatibility::check_wasm;
 use crate::instance::Instance;
 use crate::wasm_backend::compile;
@@ -81,6 +81,14 @@ impl Contract {
         self.storage = backend.storage;
         Ok(())
     }
+
+    /// get value from storage
+    pub fn raw_get(
+        &self,
+        key: &[u8],
+    ) -> Option<Vec<u8>> {
+        self.storage.get(key).0.unwrap()
+    }
 }
 
 #[cfg(test)]