[no ci] Custom codec fix

Sovereign-Labs · Dec 6, 2023 · 546765a · 546765a
1 parent 7b7a426
commit 546765a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/module-system/sov-modules-macros/tests/custom_codec_must_be_used.rs b/module-system/sov-modules-macros/tests/custom_codec_must_be_used.rs
@@ -1,9 +1,10 @@
 use std::panic::catch_unwind;
 
-use sov_modules_api::default_context::DefaultContext;
-use sov_modules_api::{prelude::*, Context, ModuleInfo, StateValue, WorkingSet};
+use sov_modules_api::default_context::ZkDefaultContext;
+use sov_modules_api::prelude::*;
+use sov_modules_api::{Context, ModuleInfo, StateValue, WorkingSet};
 use sov_modules_core::{StateCodec, StateKeyCodec, StateValueCodec};
-use sov_state::{DefaultStorageSpec, ProverStorage};
+use sov_state::{DefaultStorageSpec, ZkStorage};
 
 #[derive(ModuleInfo)]
 struct TestModule<C>
@@ -56,12 +57,11 @@ impl<V> StateValueCodec<V> for CustomCodec {
 }
 
 fn main() {
-    let tempdir = tempfile::tempdir().unwrap();
-    let storage: ProverStorage<DefaultStorageSpec> = ProverStorage::with_path(&tempdir).unwrap();
-    let module: TestModule<DefaultContext> = TestModule::default();
+    let storage: ZkStorage<DefaultStorageSpec> = ZkStorage::new();
+    let module: TestModule<ZkDefaultContext> = TestModule::default();
 
     catch_unwind(|| {
-        let mut working_set: WorkingSet<DefaultContext> = WorkingSet::new(storage);
+        let mut working_set: WorkingSet<ZkDefaultContext> = WorkingSet::new(storage);
         module.state_value.set(&0u32, &mut working_set);
     })
     .unwrap_err();

diff --git a/module-system/sov-state/src/prover_storage.rs b/module-system/sov-state/src/prover_storage.rs
@@ -1,5 +1,4 @@
 use std::marker::PhantomData;
-use std::path::Path;
 use std::sync::Arc;
 
 use jmt::storage::{NodeBatch, TreeWriter};