refactor(blockifier_reexecution): add chain ID to SerializableOffline…

…ReexecutionData
starkware-libs · Nov 12, 2024 · a4d664b · a4d664b
1 parent c4b1621
commit a4d664b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/crates/blockifier_reexecution/src/main.rs b/crates/blockifier_reexecution/src/main.rs
@@ -134,6 +134,7 @@ fn main() {
             SerializableOfflineReexecutionData {
                 serializable_data_prev_block,
                 serializable_data_next_block,
+                chain_id: ChainId::Mainnet,
                 old_block_hash,
             }
             .write_to_file(&full_file_path)

diff --git a/crates/blockifier_reexecution/src/state_reader/test_state_reader.rs b/crates/blockifier_reexecution/src/state_reader/test_state_reader.rs
@@ -83,6 +83,7 @@ pub struct SerializableDataPrevBlock {
 pub struct SerializableOfflineReexecutionData {
     pub serializable_data_prev_block: SerializableDataPrevBlock,
     pub serializable_data_next_block: SerializableDataNextBlock,
+    pub chain_id: ChainId,
     pub old_block_hash: BlockHash,
 }
 
@@ -116,6 +117,7 @@ impl From<SerializableOfflineReexecutionData> for OfflineReexecutionData {
                     transactions_next_block,
                     state_diff_next_block,
                 },
+            chain_id,
             old_block_hash,
         } = value;
 
@@ -131,7 +133,7 @@ impl From<SerializableOfflineReexecutionData> for OfflineReexecutionData {
             offline_state_reader_prev_block,
             block_context_next_block: BlockContext::new(
                 block_info_next_block,
-                get_chain_info(&ChainId::Mainnet),
+                get_chain_info(&chain_id),
                 VersionedConstants::get(&starknet_version).unwrap().clone(),
                 BouncerConfig::max(),
             ),