def test_announcement_no_experimental(self, dispatch):
runner = announcementrunner.get_runner()
runner.action = mock.Mock(ref='some.thing')
runner.runner_parameters = {
'route': 'general'
}
runner.liveaction = mock.Mock(context={})
expected_msg = 'Experimental flag is missing for action some.thing'
self.assertRaisesRegexp(Exception, expected_msg, runner.pre_run)
def test_announcement(self, dispatch):
runner = announcementrunner.get_runner()
runner.runner_parameters = {
'experimental': True,
'route': 'general'
}
runner.liveaction = mock.Mock(context={})
runner.pre_run()
(status, result, _) = runner.run({'test': 'passed'})
self.assertEqual(status, LIVEACTION_STATUS_SUCCEEDED)
self.assertTrue(result is not None)
self.assertEqual(result['test'], 'passed')
dispatch.assert_called_once_with('general', payload={'test': 'passed'},
trace_context=None)
def test_announcement_with_trace(self, context, dispatch):
runner = announcementrunner.get_runner()
runner.runner_parameters = {
'experimental': True,
'route': 'general'
}
runner.liveaction = mock.Mock(context={
'trace_context': {
'id_': 'a',
'trace_tag': 'b'
}
})
runner.pre_run()
(status, result, _) = runner.run({'test': 'passed'})
self.assertEqual(status, LIVEACTION_STATUS_SUCCEEDED)
self.assertTrue(result is not None)
self.assertEqual(result['test'], 'passed')
context.assert_called_once_with(TraceContext,
**runner.liveaction.context['trace_context'])
dispatch.assert_called_once_with('general', payload={'test': 'passed'},
trace_context=context.return_value)
def test_runner_creation(self, dispatch):
runner = announcementrunner.get_runner()
self.assertTrue(runner is not None, 'Creation failed. No instance.')
self.assertEqual(type(runner), announcementrunner.AnnouncementRunner,
'Creation failed. No instance.')
self.assertEqual(runner._dispatcher.dispatch, dispatch)