Execution of contract initialization code

package/version

@@ -1 +1 @@
-0.1.13
+0.1.14

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "kontrol"
-version = "0.1.13"
+version = "0.1.14"
 description = "Foundry integration for KEVM"
 authors = [
     "Runtime Verification, Inc. <contact@runtimeverification.com>",

src/kontrol/__main__.py

@@ -160,6 +160,7 @@ def exec_prove(
     counterexample_info: bool = False,
     trace_rewrites: bool = False,
     auto_abstract_gas: bool = False,
+    run_constructor: bool = False,
     **kwargs: Any,
 ) -> None:
     _ignore_arg(kwargs, 'main_module', f'--main-module: {kwargs["main_module"]}')
@@ -196,6 +197,7 @@ def exec_prove(
         smt_retry_limit=smt_retry_limit,
         trace_rewrites=trace_rewrites,
         auto_abstract_gas=auto_abstract_gas,
+        run_constructor=run_constructor,
     )
     failed = 0
     for pid, r in results.items():
@@ -524,6 +526,13 @@ def _parse_test_version_tuple(value: str) -> tuple[str, int | None]:
         action='store_true',
         help='Use the booster RPC server instead of kore-rpc.',
     )
+    prove_args.add_argument(
+        '--run-constructor',
+        dest='run_constructor',
+        default=False,
+        action='store_true',
+        help='Include the contract constructor in the test execution.',
+    )
 
     show_args = command_parser.add_parser(
         'show',

src/kontrol/foundry.py

@@ -241,7 +241,7 @@ def all_non_tests(self) -> list[str]:
             for contract in self.contracts.values()
             for method in contract.methods
             if f'{contract.name}.{method.signature}' not in self.all_tests
-        ]
+        ] + [f'{contract.name}.init' for contract in self.contracts.values()]
 
     @staticmethod
     def _escape_brackets(regs: list[str]) -> list[str]:
@@ -387,8 +387,14 @@ def get_optional_proof(self, test_id: str) -> Proof | None:
             return Proof.read_proof_data(self.proofs_dir, test_id)
         return None
 
-    def get_method(self, test: str) -> Contract.Method:
+    def get_method(self, test: str) -> Contract.Method | Contract.Constructor:
         contract_name, method_name = test.split('.')
+
+        if method_name == 'init':
+            constructor = self.contracts[contract_name].constructor
+            assert constructor is not None
+            return constructor
+
         contract = self.contracts[contract_name]
         return contract.method_by_sig[method_name]
 
@@ -426,7 +432,7 @@ def resolve_proof_version(
             _LOGGER.info(
                 f'Using the the latest version {latest_version} of test {test} because it is up to date and no version was specified.'
             )
-            if not method.contract_up_to_date(self.digest_file):
+            if type(method) is Contract.Method and not method.contract_up_to_date(self.digest_file):
                 _LOGGER.warning(
                     f'Test {test} was not reinitialized because it is up to date, but the contract it is a part of has changed.'
                 )