Skip to content

Commit

Permalink
mypy fixed
Browse files Browse the repository at this point in the history
  • Loading branch information
nforsg committed Aug 8, 2023
1 parent dfa00da commit 13e7af2
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -62,13 +62,13 @@ def flask_app(self):
return create_app(static_folder="../../../../../management-system/csle-mgmt-webapp/build")

@pytest.fixture
def list_sim(self, mocker: pytest_mock.MockFixture) -> List[SimulationEnvConfig]:
def list_sim(self, mocker: pytest_mock.MockFixture):
"""
Pytst dixture for mocking the list_simulations method
:param mocker: the pytest mocker object
"""
def list_simulations():
def list_simulations() -> List[SimulationEnvConfig]:
list_obj = TestResourcesSimulationsSuite.get_ex_sim_env()
return [list_obj]
list_simulations_mocker = mocker.MagicMock(side_effect=list_simulations)
Expand Down Expand Up @@ -108,7 +108,7 @@ def list_sim_id(self, mocker):
:param mocker: the pytest mocker object
:return: the mocked function
"""
def list_simulation_ids() -> List[Tuple]:
def list_simulation_ids() -> List[Tuple[str, int]]:
return [("null", 1)]
list_simulation_ids_mocker = mocker.MagicMock(side_effect=list_simulation_ids)
return list_simulation_ids_mocker
Expand Down

0 comments on commit 13e7af2

Please sign in to comment.