From b196b945207928d17b92b632343ce536f47a9546 Mon Sep 17 00:00:00 2001
From: loloicci <loloicci@linecorp.com>
Date: Wed, 6 Apr 2022 22:26:23 +0900
Subject: [PATCH] feat: add raw_get to vm::testing::Contract

---
 packages/vm/src/testing/contract.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/packages/vm/src/testing/contract.rs b/packages/vm/src/testing/contract.rs
index b80a8d0c5..fcd59b7be 100644
--- 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)]