initial contract test framework (#314)

* initial contract test framework2 * clippy * fix
lambdaclass · Oct 10, 2023 · b2a2574 · b2a2574
1 parent c80ab87
commit b2a2574
Show file tree

Hide file tree

Showing 7 changed files with 490 additions and 51 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/utils.rs b/src/utils.rs
@@ -101,6 +101,18 @@ pub fn find_entry_point<'a>(
         .find(|x| x.id.debug_name.as_deref() == Some(entry_point))
 }
 
+/// Given a string representing a function name, searches in the program for the id corresponding to said function, and returns a reference to it.
+pub fn find_function<'a>(
+    program: &'a Program,
+    function_name: &str,
+) -> &'a GenFunction<StatementIdx> {
+    program
+        .funcs
+        .iter()
+        .find(|x| x.id.debug_name.as_deref() == Some(function_name))
+        .unwrap()
+}
+
 /// Given a string representing a function name, searches in the program for the id corresponding to said function, and returns a reference to it.
 pub fn find_function_id<'a>(program: &'a Program, function_name: &str) -> &'a FunctionId {
     &program