apollo_integration_tests: add regression tests for matching the proof generation

Author: einat-starkware

Cargo.lock

@@ -64,6 +64,7 @@ blockifier = { workspace = true, features = ["testing"] }
 blockifier_test_utils.workspace = true
 cairo-lang-starknet-classes.workspace = true
 clap = { workspace = true, features = ["derive"] }
+expect-test.workspace = true
 futures.workspace = true
 hex.workspace = true
 http.workspace = true

crates/apollo_integration_tests/Cargo.toml

@@ -22,6 +22,7 @@ use blockifier::context::ChainInfo;
 use blockifier_test_utils::cairo_versions::{CairoVersion, RunnableCairo1};
 use blockifier_test_utils::contracts::FeatureContract;
 use cairo_lang_starknet_classes::casm_contract_class::CasmContractClass;
+use expect_test::{expect, Expect};
 use indexmap::IndexMap;
 use mempool_test_utils::starknet_api_test_utils::{AccountTransactionGenerator, Contract};
 use starknet_api::abi::abi_utils::get_fee_token_var_address;
@@ -376,8 +377,14 @@ fn initialize_papyrus_test_state(
 }
 
 /// Global root that matches the proof flow fixtures.
-fn integration_test_genesis_global_root() -> GlobalRoot {
-    GlobalRoot(felt!("0x68fed6a062d385db8d1dd9096060e822196feb311fc3bb4f0018635461ca85e"))
+/// Any change to this value requires regenerating the proof fixtures by running
+/// `cargo +nightly-2025-07-14 test -p starknet_os_flow_tests --features
+/// starknet_transaction_prover/stwo_proving --release generate_proof_fixtures -- --ignored`.
+pub const EXPECTED_PROOF_FLOW_GENESIS_GLOBAL_ROOT: Expect =
+    expect!["0x68fed6a062d385db8d1dd9096060e822196feb311fc3bb4f0018635461ca85e"];
+
+pub fn integration_test_genesis_global_root() -> GlobalRoot {
+    GlobalRoot(Felt::from_hex_unchecked(EXPECTED_PROOF_FLOW_GENESIS_GLOBAL_ROOT.data()))
 }
 
 fn prepare_state_diff(
@@ -664,9 +671,18 @@ impl<'a> ThinStateDiffBuilder<'a> {
     }
 }
 
+/// STRK fee token address that matches the proof flow fixtures.
+/// Any change to this value requires regenerating the proof fixtures by running
+/// `cargo +nightly-2025-07-14 test -p starknet_os_flow_tests --features
+/// starknet_transaction_prover/stwo_proving --release generate_proof_fixtures -- --ignored`.
+pub const EXPECTED_PROOF_FLOW_STRK_FEE_TOKEN_ADDRESS: Expect =
+    expect!["0x4ff17bf76a1c6cebb82601a43bcab4f9650aea543c44f28e8863f8b624e4b58"];
+
 pub fn proof_flow_chain_info() -> ChainInfo {
     let mut chain_info = ChainInfo::create_for_testing();
-    chain_info.fee_token_addresses.strk_fee_token_address =
-        contract_address!("0x4ff17bf76a1c6cebb82601a43bcab4f9650aea543c44f28e8863f8b624e4b58");
+    chain_info.fee_token_addresses.strk_fee_token_address = ContractAddress::try_from(
+        Felt::from_hex_unchecked(EXPECTED_PROOF_FLOW_STRK_FEE_TOKEN_ADDRESS.data()),
+    )
+    .unwrap();
     chain_info
 }

crates/apollo_integration_tests/src/state_reader.rs

@@ -1,5 +1,9 @@
 use std::path::Path;
 
+use apollo_integration_tests::state_reader::{
+    EXPECTED_PROOF_FLOW_GENESIS_GLOBAL_ROOT,
+    EXPECTED_PROOF_FLOW_STRK_FEE_TOKEN_ADDRESS,
+};
 use blockifier::execution::contract_class::TrackedResource;
 use blockifier::test_utils::dict_state_reader::DictStateReader;
 use blockifier::test_utils::get_valid_virtual_os_program_hash;
@@ -11,6 +15,7 @@ use rstest::rstest;
 use starknet_api::abi::abi_utils::selector_from_name;
 use starknet_api::block::{BlockInfo, BlockNumber, BlockTimestamp};
 use starknet_api::core::EthAddress;
+use starknet_api::hash::StateRoots;
 use starknet_api::test_utils::{CURRENT_BLOCK_TIMESTAMP, TEST_SEQUENCER_ADDRESS};
 use starknet_api::transaction::fields::{ProofFacts, TransactionSignature};
 use starknet_api::transaction::{
@@ -22,7 +27,9 @@ use starknet_api::transaction::{
 use starknet_api::{calldata, contract_address, invoke_tx_args};
 use starknet_types_core::felt::Felt;
 
+use crate::initial_state::create_default_initial_state_data;
 use crate::test_manager::{
+    block_context_for_flow_tests,
     EventPredicateExpectation,
     TestBuilder,
     TestBuilderConfig,
@@ -308,3 +315,36 @@ async fn generate_proof_fixtures() {
     .expect("write proof_facts.json");
     std::fs::write(&proof_path, output.proof.0.as_slice()).expect("write proof.bin");
 }
+
+/// Guards against drift between the STRK fee token address used by the proof-flow integration
+/// test and the one produced by the virtual OS test pipeline. Run with `UPDATE_EXPECT=1` to refresh
+/// the constant in `apollo_integration_tests::state_reader`, then regenerate the proof fixtures by
+/// running `cargo +nightly-2025-07-14 test -p starknet_os_flow_tests --features
+/// starknet_transaction_prover/stwo_proving --release generate_proof_fixtures -- --ignored`.
+#[test]
+fn proof_flow_chain_info_matches_virtual_os_test() {
+    let virtual_os_strk_fee_token_address = block_context_for_flow_tests(BlockNumber(0), false)
+        .chain_info()
+        .fee_token_addresses
+        .strk_fee_token_address;
+    EXPECTED_PROOF_FLOW_STRK_FEE_TOKEN_ADDRESS
+        .assert_eq(&virtual_os_strk_fee_token_address.to_hex_string());
+}
+
+/// Guards against drift between the genesis global root the proof-flow integration test seeds into
+/// storage and the initial global root produced by the virtual OS test pipeline. Run with
+/// `UPDATE_EXPECT=1` to refresh the constant in `apollo_integration_tests::state_reader`, then
+/// regenerate the proof fixtures by running `cargo +nightly-2025-07-14 test -p
+/// starknet_os_flow_tests --features starknet_transaction_prover/stwo_proving --release
+/// generate_proof_fixtures -- --ignored`.
+#[tokio::test(flavor = "multi_thread")]
+async fn proof_flow_global_root_matches_virtual_os_test() {
+    let (initial_state_data, []) =
+        create_default_initial_state_data::<DictStateReader, 0>([]).await;
+    let virtual_os_global_root = StateRoots {
+        contracts_trie_root_hash: initial_state_data.initial_state.contracts_trie_root_hash,
+        classes_trie_root_hash: initial_state_data.initial_state.classes_trie_root_hash,
+    }
+    .global_root();
+    EXPECTED_PROOF_FLOW_GENESIS_GLOBAL_ROOT.assert_eq(&virtual_os_global_root.0.to_hex_string());
+}