ProtostarFixture

protostar/__init__.py

@@ -1,7 +1,7 @@
 import sys
 
-from starkware.crypto.signature.fast_pedersen_hash import pedersen_hash
 from crypto_cpp_py.cpp_bindings import cpp_hash
+from starkware.crypto.signature.fast_pedersen_hash import pedersen_hash
 
 
 def patched_pedersen_hash(left: int, right: int) -> int:

protostar/commands/build/build_command.py

@@ -1,4 +1,5 @@
 from logging import Logger
+from pathlib import Path
 from typing import List, Optional
 
 from protostar.cli import ActivityIndicator, Command
@@ -48,15 +49,27 @@ def arguments(self) -> List[Command.Argument]:
         ]
 
     async def run(self, args):
+        await self.build(
+            output_dir=args.output,
+            disable_hint_validation=args.disable_hint_validation,
+            relative_cairo_path=args.cairo_path,
+        )
+
+    async def build(
+        self,
+        output_dir: Path,
+        disable_hint_validation=False,
+        relative_cairo_path: Optional[List[Path]] = None,
+    ):
         with ActivityIndicator(
             log_color_provider.colorize("GRAY", "Building projects' contracts")
         ):
             try:
                 self._project_compiler.compile_project(
-                    output_dir=args.output,
+                    output_dir=output_dir,
                     config=ProjectCompilerConfig(
-                        hint_validation_disabled=args.disable_hint_validation,
-                        relative_cairo_path=args.cairo_path,
+                        hint_validation_disabled=disable_hint_validation,
+                        relative_cairo_path=relative_cairo_path or [],
                     ),
                 )
             except BaseException as exc:

protostar/commands/init/init_command.py

@@ -1,4 +1,5 @@
-import glob
+from glob import glob
+from pathlib import Path
 from typing import List, Optional
 
 from protostar.cli import Command
@@ -17,11 +18,13 @@ def __init__(
         requester: InputRequester,
         new_project_creator: NewProjectCreator,
         adapted_project_creator: AdaptedProjectCreator,
+        cwd: Path,
     ) -> None:
         super().__init__()
         self._adapted_project_creator = adapted_project_creator
         self._new_project_creator = new_project_creator
         self._requester = requester
+        self._cwd = cwd
 
     @property
     def name(self) -> str:
@@ -65,9 +68,13 @@ def init(self, force_adapting_existing_project: bool):
         else:
             self._new_project_creator.run()
 
-    @staticmethod
-    def _can_be_protostar_project() -> bool:
-        files_depth_3 = glob.glob("*") + glob.glob("*/*") + glob.glob("*/*/*")
+    def _can_be_protostar_project(self) -> bool:
+
+        files_depth_3 = (
+            glob(str(self._cwd / "*"))
+            + glob(str(self._cwd / "*" / "*"))
+            + glob(str(self._cwd / "*" / "*" / "*"))
+        )
         return any(
             map(lambda f: f.endswith(".cairo") or f == ".git", files_depth_3)
-        ) and "protostar.toml" not in glob.glob("*")
+        ) and "protostar.toml" not in glob(str(self._cwd / "*"))

protostar/commands/init/project_creator/new_project_creator.py

@@ -1,5 +1,6 @@
 from dataclasses import dataclass
 from pathlib import Path
+from typing import Optional
 
 from git import InvalidGitRepositoryError
 from git.repo import Repo
@@ -16,17 +17,20 @@ class UserInput:
         project_dirname: str
         lib_dirname: str
 
+    # pylint: disable=too-many-arguments
     def __init__(
         self,
         script_root: Path,
         requester: InputRequester,
         protostar_toml_writer: ProtostarTOMLWriter,
         version_manager: VersionManager,
+        output_dir_path: Optional[Path] = None,
     ):
         super().__init__(script_root, protostar_toml_writer, version_manager)
         self._protostar_toml_writer = protostar_toml_writer
         self._version_manager = version_manager
         self._requester = requester
+        self._output_dir_path = output_dir_path or Path()
 
     def run(self):
         self._create_project(self._gather_input())
@@ -46,7 +50,8 @@ def _gather_input(self) -> "NewProjectCreator.UserInput":
         return NewProjectCreator.UserInput(project_dirname, lib_dirname)
 
     def _create_project(self, user_input: "NewProjectCreator.UserInput") -> None:
-        project_root_path = Path() / user_input.project_dirname
+        output_dir_path = self._output_dir_path
+        project_root_path = output_dir_path / user_input.project_dirname
         self.copy_template("default", project_root_path)
 
         libs_path = Path(project_root_path, user_input.lib_dirname)

protostar/commands/migrate/migrate_command.py

@@ -3,12 +3,12 @@
 from typing import List, Optional
 
 from protostar.cli import Command
+from protostar.commands.deploy import DeployCommand
 from protostar.commands.test.test_environment_exceptions import CheatcodeException
 from protostar.migrator import Migrator
 from protostar.protostar_exception import ProtostarException
 from protostar.utils.input_requester import InputRequester
 from protostar.utils.log_color_provider import LogColorProvider
-from protostar.commands.deploy import DeployCommand
 
 
 class MigrateCommand(Command):
@@ -69,8 +69,8 @@ def arguments(self) -> List[Command.Argument]:
             DeployCommand.network_arg,
         ]
 
-    async def run(self, args):
-        await self.migrate(
+    async def run(self, args) -> Optional[Migrator.History]:
+        return await self.migrate(
             migration_file_path=args.path,
             rollback=args.rollback,
             network=args.network or args.gateway_url,
@@ -114,8 +114,10 @@ async def migrate(
                 migrator.save_history(
                     migrator_history,
                     migration_file_basename=Path(migration_file_path).stem,
-                    output_dir_path=output_dir_path,
+                    output_dir_relative_path=output_dir_path,
                 )
             self._logger.info("Migration completed")
+
+            return migrator_history
         except CheatcodeException as ex:
             raise ProtostarException(str(ex)) from ex

protostar/composition_root.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from logging import getLogger
 from pathlib import Path
-from typing import List
+from typing import List, Optional
 
 from protostar.cli import Command
 from protostar.commands import (
@@ -55,13 +55,12 @@ class DIContainer:
 
 
 # pylint: disable=too-many-locals
-def build_di_container(script_root: Path):
+def build_di_container(script_root: Path, cwd: Optional[Path] = None):
+    cwd = cwd or Path()
     logger = getLogger()
-    protostar_toml_path = search_upwards_protostar_toml_path(
-        start_path=Path().resolve()
-    )
+    protostar_toml_path = search_upwards_protostar_toml_path(start_path=cwd.resolve())
     project_root_path = (
-        protostar_toml_path.parent if protostar_toml_path is not None else Path()
+        protostar_toml_path.parent if protostar_toml_path is not None else cwd
     )
     protostar_toml_path = protostar_toml_path or project_root_path / "protostar.toml"
     protostar_directory = ProtostarDirectory(script_root)
@@ -112,6 +111,7 @@ def build_di_container(script_root: Path):
                 protostar_toml_writer,
                 version_manager,
             ),
+            cwd=cwd,
         ),
         BuildCommand(project_compiler, logger),
         InstallCommand(
@@ -152,6 +152,7 @@ def build_di_container(script_root: Path):
             migrator_builder=Migrator.Builder(
                 migrator_execution_environment_builder=MigratorExecutionEnvironment.Builder(),
                 gateway_facade_builder=GatewayFacade.Builder(project_root_path),
+                cwd=project_root_path,
             ),
             requester=requester,
             logger=logger,

protostar/migrator/cheatcodes/migrator_call_cheatcode.py

@@ -0,0 +1,60 @@
+import asyncio
+from typing import Any, Optional
+
+from typing_extensions import Protocol
+
+from protostar.commands.test.test_environment_exceptions import CheatcodeException
+from protostar.starknet.cheatcode import Cheatcode
+from protostar.starknet_gateway import (
+    ContractNotFoundException,
+    GatewayFacade,
+    UnknownFunctionException,
+)
+from protostar.utils.data_transformer import CairoOrPythonData
+
+
+class CallCheatcodeProtocol(Protocol):
+    def __call__(
+        self,
+        contract_address: int,
+        function_name: str,
+        inputs: Optional[CairoOrPythonData] = None,
+    ) -> Any:
+        ...
+
+
+class MigratorCallCheatcode(Cheatcode):
+    def __init__(
+        self,
+        syscall_dependencies: Cheatcode.SyscallDependencies,
+        gateway_facade: GatewayFacade,
+    ):
+        super().__init__(syscall_dependencies)
+        self._gateway_facade = gateway_facade
+
+    @property
+    def name(self) -> str:
+        return "call"
+
+    def build(self) -> CallCheatcodeProtocol:
+        return self.call
+
+    def call(
+        self,
+        contract_address: int,
+        function_name: str,
+        inputs: Optional[CairoOrPythonData] = None,
+    ):
+        try:
+            return asyncio.run(
+                self._gateway_facade.call(
+                    address=contract_address,
+                    function_name=function_name,
+                    inputs=inputs,
+                )
+            )
+        except (UnknownFunctionException, ContractNotFoundException) as err:
+            raise CheatcodeException(
+                self.name,
+                message=err.message,
+            ) from err

protostar/migrator/migrator.py

@@ -9,7 +9,6 @@
 from protostar.migrator.migrator_execution_environment import (
     MigratorExecutionEnvironment,
 )
-
 from protostar.starknet_gateway.gateway_facade import GatewayFacade
 from protostar.starknet_gateway.starknet_request import StarknetRequest
 from protostar.utils.log_color_provider import LogColorProvider
@@ -29,6 +28,7 @@ def __init__(
             self,
             migrator_execution_environment_builder: MigratorExecutionEnvironment.Builder,
             gateway_facade_builder: GatewayFacade.Builder,
+            cwd: Optional[Path] = None,
         ) -> None:
             self._migrator_execution_environment_builder = (
                 migrator_execution_environment_builder
@@ -39,6 +39,7 @@ def __init__(
             self._migrator_execution_environment_config = (
                 MigratorExecutionEnvironment.Config()
             )
+            self._cwd = cwd
 
         def set_logger(
             self, logger: Logger, log_color_provider: LogColorProvider
@@ -72,12 +73,16 @@ async def build(self, migration_file_path: Path):
                 )
             )
 
-            return Migrator(migrator_execution_environment=migrator_execution_env)
+            return Migrator(
+                migrator_execution_environment=migrator_execution_env, cwd=self._cwd
+            )
 
     def __init__(
         self,
         migrator_execution_environment: MigratorExecutionEnvironment,
+        cwd: Optional[Path] = None,
     ) -> None:
+        self._cwd = cwd or Path()
         self._migrator_execution_environment = migrator_execution_environment
 
     async def run(self, rollback=False) -> History:
@@ -90,16 +95,19 @@ async def run(self, rollback=False) -> History:
             starknet_requests=self._migrator_execution_environment.cheatcode_factory.gateway_facade.get_starknet_requests()
         )
 
-    @staticmethod
     def save_history(
+        self,
         history: History,
         migration_file_basename: str,
-        output_dir_path: Path,
+        output_dir_relative_path: Path,
     ):
+        output_dir_path = self._cwd / output_dir_relative_path
         prefix = datetime.strftime(datetime.now(), "%Y%m%d%H%M%S")
         output_file_path = output_dir_path / f"{prefix}_{migration_file_basename}.json"
 
         if not output_dir_path.exists():
-            output_dir_path.mkdir(parents=True)
+            output_dir_path.mkdir(
+                parents=True,
+            )
 
         history.save_as_json(output_file_path)

protostar/migrator/migrator_cheatcodes_factory.py

@@ -3,6 +3,7 @@
 
 from starkware.starknet.business_logic.execution.objects import CallInfo
 
+from protostar.migrator.cheatcodes.migrator_call_cheatcode import MigratorCallCheatcode
 from protostar.migrator.cheatcodes.migrator_declare_cheatcode import (
     MigratorDeclareCheatcode,
 )
@@ -51,4 +52,5 @@ def build(
                 self.gateway_facade,
                 config=MigratorDeployContractCheatcode.Config(token=self._config.token),
             ),
+            MigratorCallCheatcode(syscall_dependencies, self.gateway_facade),
         ]

protostar/migrator/migrator_test.py

@@ -2,14 +2,19 @@
 from pathlib import Path
 
 from freezegun import freeze_time
+from pytest_mock import MockerFixture
 
 from protostar.migrator.migrator import Migrator
 from protostar.starknet_gateway.starknet_request import StarknetRequest
 
 
 @freeze_time("2022-04-02 21:37:42")
-def test_migrator_saves_result_successfully_with_proper_name(tmp_path: Path):
-    Migrator.save_history(
+def test_migrator_saves_result_successfully_with_proper_name(
+    mocker: MockerFixture, tmp_path: Path
+):
+    migrator = Migrator(migrator_execution_environment=mocker.MagicMock())
+
+    migrator.save_history(
         history=Migrator.History(
             starknet_requests=[
                 StarknetRequest(
@@ -18,7 +23,7 @@ def test_migrator_saves_result_successfully_with_proper_name(tmp_path: Path):
             ]
         ),
         migration_file_basename="01_init",
-        output_dir_path=tmp_path,
+        output_dir_relative_path=tmp_path,
     )
 
     assert "20220402213742_01_init.json" in listdir(tmp_path)

protostar/protostar_toml/io/protostar_toml_reader.py

@@ -73,7 +73,7 @@ def _read_if_cache_miss(self) -> Dict[str, Any]:
 
         if not self.path.is_file():
             raise NoProtostarProjectFoundException(
-                "No protostar.toml found in the working directory"
+                "No protostar.toml found in the working directory\n" f"{str(self.path)}"
             )
 
         with open(self.path, "rb") as protostar_toml_file:

protostar/starknet_gateway/__init__.py

@@ -1,4 +1,8 @@
-from protostar.starknet_gateway.gateway_facade import GatewayFacade
+from protostar.starknet_gateway.gateway_facade import (
+    ContractNotFoundException,
+    GatewayFacade,
+    UnknownFunctionException,
+)
 from protostar.starknet_gateway.gateway_response import SuccessfulDeclareResponse
 from protostar.starknet_gateway.network_config import (
     InvalidNetworkConfigurationException,