Fix after rebase

Sovereign-Labs · Aug 10, 2023 · f0c4126 · f0c4126
1 parent c9715bf
commit f0c4126
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 17 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/module-system/module-implementations/integration-tests/src/chain_state/helpers.rs b/module-system/module-implementations/integration-tests/src/chain_state/helpers.rs
@@ -1,4 +1,5 @@
 use sov_chain_state::{ChainState, ChainStateConfig};
+use sov_modules_api::capabilities::{BlobRefOrOwned, BlobSelector};
 use sov_modules_api::default_context::DefaultContext;
 use sov_modules_api::hooks::{ApplyBlobHooks, SlotHooks, TxHooks};
 use sov_modules_api::transaction::Transaction;
@@ -73,6 +74,22 @@ impl<C: Context, Cond: ValidityCondition> SlotHooks<Cond> for TestRuntime<C, Con
     fn end_slot_hook(&self, _working_set: &mut WorkingSet<<Self::Context as Spec>::Storage>) {}
 }
 
+impl<C: Context, Cond: ValidityCondition> BlobSelector for TestRuntime<C, Cond> {
+    type Context = C;
+
+    fn get_blobs_for_this_slot<'a, I, B>(
+        &self,
+        current_blobs: I,
+        _working_set: &mut WorkingSet<<Self::Context as Spec>::Storage>,
+    ) -> anyhow::Result<Vec<BlobRefOrOwned<'a, B>>>
+    where
+        B: BlobReaderTrait,
+        I: IntoIterator<Item = &'a mut B>,
+    {
+        Ok(current_blobs.into_iter().map(Into::into).collect())
+    }
+}
+
 impl<C: Context, Cond: ValidityCondition> Runtime<C, Cond> for TestRuntime<C, Cond> {}
 
 pub(crate) fn create_demo_genesis_config<C: Context, Cond: ValidityCondition>(

diff --git a/module-system/module-implementations/sov-blob-storage/tests/capability_tests.rs b/module-system/module-implementations/sov-blob-storage/tests/capability_tests.rs
@@ -52,10 +52,7 @@ fn priority_sequencer_flow() {
     let regular_sequencer_da = MockAddress::from([30u8; 32]);
     let regular_sequencer_rollup = generate_address(REGULAR_SEQUENCER_KEY);
 
-    let bank_config = get_bank_config(
-        preferred_sequencer_rollup.clone(),
-        regular_sequencer_rollup.clone(),
-    );
+    let bank_config = get_bank_config(preferred_sequencer_rollup, regular_sequencer_rollup);
 
     let token_address = sov_bank::get_genesis_token_address::<C>(
         &bank_config.tokens[0].token_name,
@@ -155,10 +152,7 @@ fn test_blobs_from_non_registered_sequencers_are_not_saved() {
     let regular_sequencer_rollup = generate_address(REGULAR_SEQUENCER_KEY);
     let some_sequencer = MockAddress::from([40u8; 32]);
 
-    let bank_config = get_bank_config(
-        preferred_sequencer_rollup.clone(),
-        regular_sequencer_rollup.clone(),
-    );
+    let bank_config = get_bank_config(preferred_sequencer_rollup, regular_sequencer_rollup);
 
     let token_address = sov_bank::get_genesis_token_address::<C>(
         &bank_config.tokens[0].token_name,
@@ -226,10 +220,7 @@ fn test_blobs_no_deferred_without_preferred_sequencer() {
     let regular_sequencer_da = MockAddress::from([30u8; 32]);
     let regular_sequencer_rollup = generate_address(REGULAR_SEQUENCER_KEY);
 
-    let bank_config = get_bank_config(
-        preferred_sequencer_rollup.clone(),
-        regular_sequencer_rollup.clone(),
-    );
+    let bank_config = get_bank_config(preferred_sequencer_rollup, regular_sequencer_rollup);
 
     let token_address = sov_bank::get_genesis_token_address::<C>(
         &bank_config.tokens[0].token_name,
@@ -298,18 +289,15 @@ fn deferred_blobs_are_first_after_preferred_sequencer_exit() {
     let regular_sequencer_da = MockAddress::from([30u8; 32]);
     let regular_sequencer_rollup = generate_address(REGULAR_SEQUENCER_KEY);
 
-    let bank_config = get_bank_config(
-        preferred_sequencer_rollup.clone(),
-        regular_sequencer_rollup.clone(),
-    );
+    let bank_config = get_bank_config(preferred_sequencer_rollup, regular_sequencer_rollup);
 
     let token_address = sov_bank::get_genesis_token_address::<C>(
         &bank_config.tokens[0].token_name,
         bank_config.tokens[0].salt,
     );
 
     let sequencer_registry_config = SequencerConfig {
-        seq_rollup_address: preferred_sequencer_rollup.clone(),
+        seq_rollup_address: preferred_sequencer_rollup,
         seq_da_address: preferred_sequencer_da.as_ref().to_vec(),
         coins_to_lock: sov_bank::Coins {
             amount: LOCKED_AMOUNT,