feat(apollo_starknet_os_program): add test contracts, add dumper in CLI (#5946)

Author: dorimedini-starkware

crates/apollo_starknet_os_program/Cargo.toml

@@ -9,6 +9,7 @@ build = "build/main.rs"
 
 [features]
 dump_source_files = []
+test_programs = []
 
 [lints]
 workspace = true

crates/apollo_starknet_os_program/build/compile_program.rs

@@ -35,6 +35,17 @@ pub async fn compile_and_output_program(
     });
 }
 
+#[cfg(feature = "test_programs")]
+pub async fn compile_test_contracts(out_dir: PathBuf) {
+    let mut task_set = tokio::task::JoinSet::new();
+    task_set.spawn(compile_and_output_program(
+        out_dir,
+        "starkware/starknet/core/os/state/aliases_test.cairo",
+        "aliases_test",
+    ));
+    task_set.join_all().await;
+}
+
 fn cairo_root_path() -> PathBuf {
     PathBuf::from(compile_time_cargo_manifest_dir!()).join("src/cairo")
 }

crates/apollo_starknet_os_program/build/main.rs

@@ -15,6 +15,8 @@ async fn main() {
     dump_source::dump_source_files(&out_dir.join("cairo_files_map.json"));
 
     let mut task_set = tokio::task::JoinSet::new();
+    #[cfg(feature = "test_programs")]
+    task_set.spawn(compile_program::compile_test_contracts(out_dir.clone()));
     task_set.spawn(compile_program::compile_and_output_program(
         out_dir.clone(),
         "starkware/starknet/core/os/os.cairo",

crates/apollo_starknet_os_program/src/lib.rs

@@ -9,6 +9,8 @@ use crate::program_hash::ProgramHashes;
 #[cfg(test)]
 mod constants_test;
 pub mod program_hash;
+#[cfg(feature = "test_programs")]
+pub mod test_programs;
 
 #[cfg(feature = "dump_source_files")]
 pub static CAIRO_FILES_MAP: LazyLock<HashMap<String, String>> = LazyLock::new(|| {

crates/apollo_starknet_os_program/src/test_programs.rs

@@ -0,0 +1,2 @@
+pub const ALIASES_TEST_BYTES: &[u8] =
+    include_bytes!(concat!(env!("OUT_DIR"), "/aliases_test_bytes"));

crates/starknet_committer_and_os_cli/Cargo.toml

@@ -18,8 +18,9 @@ tempfile.workspace = true
 
 # TODO(Amos): Add `testing` feature and move Python test dependencies under it.
 [dependencies]
-# The 'dump_source_files' feature should be moved under `testing` feature, when it exists.
-apollo_starknet_os_program = { workspace = true, features = ["dump_source_files"] }
+# The 'dump_source_files' and 'test_programs' features should be moved under `testing` feature, when
+# it exists.
+apollo_starknet_os_program = { workspace = true, features = ["dump_source_files", "test_programs"] }
 blake2s.workspace = true
 cairo-lang-starknet-classes.workspace = true
 cairo-vm = { workspace = true, features = [

crates/starknet_committer_and_os_cli/src/os_cli/commands.rs

@@ -1,6 +1,7 @@
 use std::fs;
 use std::path::Path;
 
+use apollo_starknet_os_program::test_programs::ALIASES_TEST_BYTES;
 use apollo_starknet_os_program::{
     AGGREGATOR_PROGRAM_BYTES,
     CAIRO_FILES_MAP,
@@ -115,6 +116,7 @@ pub(crate) fn dump_source_files(output_path: String) {
 pub(crate) fn dump_program(output_path: String, program: ProgramToDump) {
     let bytes = match program {
         ProgramToDump::Aggregator => AGGREGATOR_PROGRAM_BYTES,
+        ProgramToDump::AliasesTest => ALIASES_TEST_BYTES,
         ProgramToDump::Os => OS_PROGRAM_BYTES,
     };
     // Dumping the `Program` struct won't work - it is not deserializable via cairo-lang's Program

crates/starknet_committer_and_os_cli/src/os_cli/run_os_cli.rs

@@ -28,6 +28,7 @@ pub struct OsCliCommand {
 #[serde(rename_all = "kebab-case")]
 pub enum ProgramToDump {
     Aggregator,
+    AliasesTest,
     Os,
 }