def test_run_1(self):
scenario_id = "test_id"
action = BreakScenarioAction({"scenario_id": scenario_id})
user = Mock()
scenario_model = MagicMock()
scenario_model.set_break = Mock(return_value=None)
user.scenario_models = {scenario_id: scenario_model}
action.run(user, None)
user.scenario_models[scenario_id].set_break.assert_called_once()
def test_run_2(self):
scenario_id = "test_id"
action = BreakScenarioAction({})
user = PicklableMock()
user.last_scenarios.last_scenario_name = "test_id"
scenario_model = PicklableMagicMock()
scenario_model.set_break = Mock(return_value=None)
user.scenario_models = {scenario_id: scenario_model}
action.run(user, None)
user.scenario_models[scenario_id].set_break.assert_called_once()