p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Parametrize test_save_simulation()

This commit is contained in:
Mike Salvatore 2022-07-01 11:40:21 -04:00
parent 4a91777019
commit 9990cdc6a8
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
monkey/tests/unit_tests/monkey_island/cc/repository/test_file_simulation_repository.py
View File

@ -11,15 +11,12 @@ def simulation_repository():
return FileSimulationRepository(SingleFileRepository()) return FileSimulationRepository(SingleFileRepository())
def test_save_simulation(simulation_repository): @pytest.mark.parametrize("mode", list(IslandModeEnum))
simulation = Simulation(IslandModeEnum.RANSOMWARE) def test_save_simulation(simulation_repository, mode):
simulation = Simulation(mode)
old_simulation = simulation_repository.get_simulation()
simulation_repository.save_simulation(simulation) simulation_repository.save_simulation(simulation)
new_simulation = simulation_repository.get_simulation()
assert old_simulation != simulation assert simulation_repository.get_simulation() == simulation
assert new_simulation == simulation
def test_get_default_simulation(simulation_repository): def test_get_default_simulation(simulation_repository):