skip reserved function afterInvariant()

https://getfoundry.sh/forge/advanced-testing/invariant-testing/?highlight=afterInvariant%28%29

Author: 0xkarmacoma

src/halmos/__main__.py

@@ -1390,9 +1390,10 @@ def resolve_target_selectors(
                 or fun_sig.startswith("check_")
                 or fun_sig.startswith("prove_")
                 or fun_sig.startswith("invariant_")
-                or fun_sig.startswith("setUp")
+                or fun_sig == "setUp()"
+                or fun_sig == "afterInvariant()"
             ):
-                debug(f"Skipping special function {fun_sig}")
+                debug(f"Skipping {fun_sig} (reserved function)")
                 continue
 
             yield (fun_sig, fun_selector)

tests/expected/all.json

@@ -3338,6 +3338,15 @@
                 "time": null,
                 "num_bounded_loops": null
             },
+            {
+                "name": "invariant_targets_inception_afterInvariant_included()",
+                "exitcode": 1,
+                "num_models": 1,
+                "models": null,
+                "num_paths": null,
+                "time": null,
+                "num_bounded_loops": null
+            },
             {
                 "name": "invariant_targets_inception_check_included()",
                 "exitcode": 1,

tests/regression/test/InvariantTargetThis.t.sol

@@ -10,6 +10,7 @@ contract InvariantTargetBase {
     bool public test_foo_called;
     bool public check_foo_called;
     bool public setUpSymbolic_called;
+    bool public afterInvariant_called;
 
     function setUp() public virtual {
         setUp_called = true;
@@ -27,6 +28,10 @@ contract InvariantTargetBase {
         _invariant_this_called = true;
     }
 
+    function afterInvariant() public {
+        afterInvariant_called = true;
+    }
+
     // should be included
     function foo() public {
         foo_called = true;
@@ -63,6 +68,7 @@ contract InvariantTargetThis is InvariantTargetBase, Test {
         assertEq(setUpSymbolic_called, false);
         assertEq(test_foo_called, false);
         assertEq(_invariant_this_called, false);
+        assertEq(afterInvariant_called, false);
     }
 
     // we expect a counterexample, showing that setUp is called
@@ -85,6 +91,11 @@ contract InvariantTargetThis is InvariantTargetBase, Test {
         assertEq(inception.check_foo_called(), false);
     }
 
+    // we expect a counterexample, showing that afterInvariant is called
+    function invariant_targets_inception_afterInvariant_included() public view {
+        assertEq(inception.afterInvariant_called(), false);
+    }
+
     // we expect a counterexample, showing that foo is indeed called
     function invariant_targets_foo_included() public view {
         assertEq(foo_called, false);