p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 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())
def test_save_simulation(simulation_repository):
simulation = Simulation(IslandModeEnum.RANSOMWARE)
old_simulation = simulation_repository.get_simulation()
@pytest.mark.parametrize("mode", list(IslandModeEnum))
def test_save_simulation(simulation_repository, mode):
simulation = Simulation(mode)
simulation_repository.save_simulation(simulation)
new_simulation = simulation_repository.get_simulation()
assert old_simulation != simulation
assert new_simulation == simulation
assert simulation_repository.get_simulation() == simulation
def test_get_default_simulation(simulation_repository):