Implement PyCairoRunner::add_additional_hash_builtin(). (#144)

* Implement `PyCairoRunner::add_additional_hash_builtin()`.

* Fix test.

* Remove `hash_func` argument.

* Remove unnecessary test.

* Add test.

* Empty commit to trigger CI.

Author: azteca1998

Cargo.toml

@@ -13,7 +13,7 @@ default = ["pyo3/num-bigint", "pyo3/auto-initialize"]
 
 [dependencies]
 pyo3 = { version = "0.16.5" }
-cairo-rs = { git = "https://github.com/lambdaclass/cairo-rs.git", rev = "8c47dda53e874545895b34d675be6254878a9e7b" }
+cairo-rs = { git = "https://github.com/lambdaclass/cairo-rs.git", rev = "c7c3e4fca78d38da968e92c6586ec3a518697745" }
 num-bigint = "0.4"
 lazy_static = "1.4.0"
 

src/cairo_runner.rs

@@ -534,6 +534,12 @@ impl PyCairoRunner {
             .collect::<Vec<_>>()
             .to_object(py))
     }
+
+    /// Add (or replace if already present) a custom hash builtin.
+    pub fn add_additional_hash_builtin(&self) {
+        let mut vm = (*self.pyvm.vm).borrow_mut();
+        self.inner.add_additional_hash_builtin(&mut vm);
+    }
 }
 
 #[pyclass]
@@ -1198,6 +1204,10 @@ mod test {
                 segment_index: 6,
                 offset: 0,
             })],
+            vec![RelocatableValue(PyRelocatable {
+                segment_index: 7,
+                offset: 0,
+            })],
         ];
 
         Python::with_gil(|py| {
@@ -1491,4 +1501,37 @@ mod test {
             );
         });
     }
+
+    /// Test that add_additional_hash_builtin() returns successfully.
+    #[test]
+    fn add_additional_hash_builtin() {
+        Python::with_gil(|_| {
+            let program = fs::read_to_string("cairo_programs/fibonacci.json").unwrap();
+            let runner = PyCairoRunner::new(
+                program,
+                Some("main".to_string()),
+                Some("small".to_string()),
+                false,
+            )
+            .unwrap();
+
+            runner.add_additional_hash_builtin();
+            assert_eq!(
+                (*runner.pyvm.vm)
+                    .borrow()
+                    .get_builtin_runners()
+                    .last()
+                    .map(|(key, _)| key.as_str()),
+                Some("hash_builtin"),
+            );
+
+            let mut vm = (*runner.pyvm.vm).borrow_mut();
+            // Check that the segment exists by writing to it.
+            vm.insert_value(
+                &Relocatable::from((0, 0)),
+                MaybeRelocatable::Int(bigint!(42)),
+            )
+            .expect("memory insert failed");
+        });
+    }
 }