Update current scope when computing closure usage. (#6162)

starkware-libs · Aug 7, 2024 · dd86a0e · dd86a0e
1 parent 209c183
commit dd86a0e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/crates/cairo-lang-semantic/src/usage/mod.rs b/crates/cairo-lang-semantic/src/usage/mod.rs
@@ -290,6 +290,7 @@ impl Usages {
                 usage.introductions.extend(expr.param_ids.iter().map(|id| VarId::Param(*id)));
                 self.handle_expr(arenas, expr.body, &mut usage);
                 usage.finalize_as_scope();
+                current.add_usage_and_changes(&usage);
                 self.usages.insert(expr_id, usage);
             }
             Expr::FunctionCall(expr) => {

diff --git a/crates/cairo-lang-semantic/src/usage/test_data/usage b/crates/cairo-lang-semantic/src/usage/test_data/usage
@@ -125,7 +125,7 @@ test_function_usage
 //! > function
 fn foo(a: u32) {
     |b: u32| {
-        a + b
+         || a + b;
     };
 }
 
@@ -137,6 +137,10 @@ foo
 //! > semantic_diagnostics
 
 //! > usage
+Closure 2:9:
+  Usage: ParamId(test::a), ParamId(test::b),
+  Changes:
+  Snapshot_Usage:
 Closure 1:4:
   Usage: ParamId(test::a),
   Changes: