fix(decompression): move concrete fixtures to test-fixtures crate

Author: rymnc

fuel-zkvm-primitives-prover/Cargo.toml

@@ -19,3 +19,6 @@ thiserror = { version = "2.0.3" }
 
 [dev-dependencies]
 rand = { version = "0.9.0" }
+
+[features]
+test-helpers = []

fuel-zkvm-primitives-prover/src/games.rs

@@ -1,3 +1,3 @@
 pub mod block_execution_game;
 
-pub mod decompression_gzip_game;
+pub mod decompression_game;

fuel-zkvm-primitives-prover/src/games/decompression_gzip_game.rs renamed to fuel-zkvm-primitives-prover/src/games/decompression_game.rs

@@ -46,8 +46,8 @@ impl Blob {
         self._inner
     }
 
-    #[cfg(test)]
-    fn from_inner(inner: [u8; 131072]) -> Self {
+    #[cfg(feature = "test-helpers")]
+    pub fn from_inner(inner: [u8; 131072]) -> Self {
         Self {
             _inner: Box::new(inner),
         }
@@ -83,7 +83,7 @@ pub struct Input {
     // a set of blobs make up a compressed bundle
     // a compressed bundle is made of several bundles
     // each bundle is made of several da compressed block
-    raw_da_blobs: Vec<Blob>,
+    pub raw_da_blobs: Vec<Blob>,
 }
 
 sol! {
@@ -289,29 +289,4 @@ mod tests {
         assert_eq!(result.first_block_height, U256::from(first_height));
         assert_eq!(result.last_block_height, U256::from(last_height));
     }
-
-    #[test]
-    fn prove_succeeds__with_real_data() {
-        let blobs = include_bytes!("decompression_gzip_game/mainnet_blobs.bin");
-        let blobs: Vec<Vec<u8>> = bincode::deserialize(blobs).unwrap();
-
-        let blobs = blobs
-            .iter()
-            .map(|blob| {
-                let mut blob_array = [0; 131072];
-                blob_array.copy_from_slice(&blob[8..]);
-                Blob::from_inner(blob_array)
-            })
-            .collect::<Vec<_>>();
-
-        let input = Input {
-            raw_da_blobs: blobs,
-        };
-
-        let input_bytes = bincode::serialize(&input).unwrap();
-
-        let result = prove(&input_bytes).unwrap();
-
-        assert!(result.first_block_height < result.last_block_height);
-    }
 }

fuel-zkvm-primitives-test-fixtures/Cargo.toml

@@ -35,7 +35,7 @@ tokio = { workspace = true }
 
 [dev-dependencies]
 bincode = { workspace = true }
-fuel-zkvm-primitives-prover = { workspace = true }
+fuel-zkvm-primitives-prover = { workspace = true, features = ["test-helpers"] }
 rayon = "1.10.0"
 
 [features]

fuel-zkvm-primitives-test-fixtures/src/decompression_fixtures.rs

@@ -0,0 +1,86 @@
+//! Decompression proving game test fixtures
+
+use clap::builder::PossibleValue;
+use std::sync::OnceLock;
+
+/// Random blob sets from mainnet DA
+#[cfg_attr(
+    feature = "enhanced_enums",
+    derive(enum_iterator::Sequence, clap::ValueEnum)
+)]
+#[cfg_attr(feature = "enhanced_enums", clap(rename_all = "snake_case"))]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
+#[allow(non_camel_case_types)]
+#[derive(Debug, Copy, Clone)]
+pub enum BlobSet {
+    /// Random blob set 0
+    Blob_14133451_14136885,
+}
+
+/// Get prover input for a particular blobset
+pub fn get_blobset_input(blobset: BlobSet) -> Vec<u8> {
+    match blobset {
+        BlobSet::Blob_14133451_14136885 => {
+            include_bytes!("decompression_fixtures/fixtures/Blob_14133451_14136885.bin").to_vec()
+        }
+    }
+}
+
+static FIXTURE_VARIANTS: OnceLock<Vec<Fixture>> = OnceLock::new();
+
+/// Get all fixtures
+pub fn all_fixtures() -> &'static Vec<Fixture> {
+    FIXTURE_VARIANTS.get_or_init(|| enum_iterator::all::<Fixture>().collect())
+}
+
+/// Fixtures for the prover
+#[derive(Debug, Clone, enum_iterator::Sequence)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
+pub enum Fixture {
+    /// Blob set
+    BlobSet(BlobSet),
+}
+
+impl Fixture {
+    /// Get the prover input for the fixture
+    pub fn get_input_for_fixture(&self) -> Vec<u8> {
+        match self {
+            Fixture::BlobSet(blobset) => get_blobset_input(*blobset),
+        }
+    }
+}
+
+impl clap::ValueEnum for Fixture {
+    fn value_variants<'a>() -> &'a [Self] {
+        all_fixtures().as_slice()
+    }
+
+    fn to_possible_value(&self) -> Option<PossibleValue> {
+        match self {
+            Fixture::BlobSet(blobset) => blobset.to_possible_value(),
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use fuel_zkvm_primitives_prover::games::decompression_game::prove;
+
+    #[test]
+    fn test_all_fixtures() -> Result<(), String> {
+        for fixture in all_fixtures() {
+            let prover_input = fixture.get_input_for_fixture();
+
+            if prover_input.is_empty() {
+                return Err(format!("Fixture '{:?}' has empty prover input", fixture));
+            }
+
+            let result = prove(&prover_input).unwrap();
+
+            assert!(result.first_block_height < result.last_block_height);
+        }
+
+        Ok(())
+    }
+}

fuel-zkvm-primitives-prover/src/games/decompression_gzip_game/mainnet_blobs.bin renamed to fuel-zkvm-primitives-test-fixtures/src/decompression_fixtures/fixtures/Blob_14133451_14136885.bin

@@ -9,3 +9,4 @@
 use tai64 as _;
 
 pub mod block_execution_fixtures;
+pub mod decompression_fixtures;