def test_onObservationAborted_calls_cancel_trigger_with_correct_trigger(
self):
handler = TriggerCancellationHandler(self.momqueryrpc_mock)
handler.onObservationAborted(self.obs_sas_id, None)
self.momqueryrpc_mock.cancel_trigger.assert_called()
def test_onObservationConflict_calls_cancel_trigger_when_its_a_trigger(
self):
handler = TriggerCancellationHandler(self.momqueryrpc_mock)
handler.onObservationConflict(self.obs_sas_id, None)
self.momqueryrpc_mock.cancel_trigger.assert_called()
def test_onObservationAborted_does_not_call_cancel_trigger_when_its_not_a_trigger(
self):
self.momqueryrpc_mock.get_trigger_id.return_value = {
'trigger_id': None,
'status': "Error"
}
handler = TriggerCancellationHandler(self.momqueryrpc_mock)
handler.onObservationAborted(self.obs_sas_id, None)
self.momqueryrpc_mock.cancel_trigger.assert_not_called()
def test_onObservationAborted_uses_correct_triggerid_and_sets_correct_cancellation_reason(
self):
handler = TriggerCancellationHandler(self.momqueryrpc_mock)
handler.onObservationAborted(self.obs_sas_id, None)
# correct otdb id is used to obtain trigger
call_otdb_id = self.momqueryrpc_mock.getMoMIdsForOTDBIds.call_args[0][
0]
self.assertEqual(self.obs_sas_id, call_otdb_id)
# correct trigger id and reason are used to abort
call_id, call_reason = self.momqueryrpc_mock.cancel_trigger.call_args[
0]
self.assertEqual(self.trigger_id, call_id)
self.assertIn('aborted', call_reason)
def test_stop_listening_closes_momquery_rpc(self):
handler = TriggerCancellationHandler(self.momqueryrpc_mock)
handler.stop_handling()
self.momqueryrpc_mock.close.assert_called()
def test_start_listening_opens_momquery_rpc(self):
handler = TriggerCancellationHandler(self.momqueryrpc_mock)
handler.start_handling()
self.momqueryrpc_mock.open.assert_called()