def _test_notify(self, priority): self.mock_connection = self.mocker.CreateMock(self.orig_connection) self.mock_session = self.mocker.CreateMock(self.orig_session) self.mock_sender = self.mocker.CreateMock(self.orig_sender) self.mock_connection.username = "" self.mock_connection.open() self.mock_connection.session().AndReturn(self.mock_session) for p in ["info", "warn", "error"]: expected_address = ('glance/glance_notifications.%s ; {"node": ' '{"x-declare": {"auto-delete": true, "durable": false}, ' '"type": "topic"}, "create": "always"}' % p) self.mock_session.sender(expected_address).AndReturn( self.mock_sender) self.mock_sender.send('stuff') self.mocker.ReplayAll() conf = utils.TestConfigOpts({"notifier_strategy": "qpid"}) notifier = self.notify_qpid.QpidStrategy(conf) if priority == "info": notifier.info("stuff") elif priority == "warn": notifier.warn("stuff") elif priority == "error": notifier.error("stuff") self.mocker.VerifyAll()
def _test_notify(self, priority): test_msg = {'a': 'b'} self.mock_connection = self.mocker.CreateMock(self.orig_connection) self.mock_session = self.mocker.CreateMock(self.orig_session) self.mock_sender = self.mocker.CreateMock(self.orig_sender) self.mock_connection.username = "" self.mock_connection.open() self.mock_connection.session().AndReturn(self.mock_session) for p in ["info", "warn", "error"]: expected_address = ('glance/glance_notifications.%s ; ' '{"node": {"x-declare": {"auto-delete": true, ' '"durable": false}, "type": "topic"}, ' '"create": "always"}' % p) self.mock_session.sender(expected_address).AndReturn( self.mock_sender) self.mock_sender.send(mox.IgnoreArg()) self.mocker.ReplayAll() self.config(notifier_strategy="qpid") notifier = self.notify_qpid.QpidStrategy() if priority == 'info': notifier.info(test_msg) elif priority == 'warn': notifier.warn(test_msg) elif priority == 'error': notifier.error(test_msg) self.mocker.VerifyAll()
def _test_notify(self, priority, exception=False, exception_send=False): test_msg = {'a': 'b'} self.mock_connection = self.mocker.CreateMock(self.orig_connection) self.mock_session = self.mocker.CreateMock(self.orig_session) self.mock_sender = self.mocker.CreateMock(self.orig_sender) self.mock_connection.username = "" if exception: self.mock_connection.open().AndRaise( Exception('Test open Exception')) else: self.mock_connection.open() self.mock_connection.session().AndReturn(self.mock_session) expected_address = ('glance/notifications.%s ; ' '{"node": {"x-declare": {"auto-delete": true, ' '"durable": false}, "type": "topic"}, ' '"create": "always"}' % priority) self.mock_session.sender(expected_address).AndReturn( self.mock_sender) if exception_send: self.mock_sender.send(mox.IgnoreArg()).AndRaise( Exception('Test send Exception')) # NOTE(afazekas): the opened and close call is expected # in this case, but not expected if the open fails else: self.mock_sender.send(mox.IgnoreArg()) self.mock_connection.opened().AndReturn(True) self.mock_connection.close() self.mocker.ReplayAll() self.config(notifier_strategy="qpid") notifier = self.notify_qpid.QpidStrategy() if priority == 'info': if exception or exception_send: self.assertRaises(Exception, notifier.info, test_msg) else: notifier.info(test_msg) elif priority == 'warn': if exception or exception_send: self.assertRaises(Exception, notifier.warn, test_msg) else: notifier.warn(test_msg) elif priority == 'error': if exception or exception_send: self.assertRaises(Exception, notifier.error, test_msg) else: notifier.error(test_msg) self.mocker.VerifyAll()
def _test_notify(self, priority, exception=False, opened=True): test_msg = {"a": "b"} self.mock_connection = self.mocker.CreateMock(self.orig_connection) self.mock_session = self.mocker.CreateMock(self.orig_session) self.mock_sender = self.mocker.CreateMock(self.orig_sender) self.mock_connection.username = "" if exception: self.mock_connection.open().AndRaise(Exception("Test Exception")) else: self.mock_connection.open() self.mock_connection.session().AndReturn(self.mock_session) expected_address = ( "glance/notifications.%s ; " '{"node": {"x-declare": {"auto-delete": true, ' '"durable": false}, "type": "topic"}, ' '"create": "always"}' % priority ) self.mock_session.sender(expected_address).AndReturn(self.mock_sender) self.mock_sender.send(mox.IgnoreArg()) self.mock_connection.opened().AndReturn(opened) if opened: self.mock_connection.close() self.mocker.ReplayAll() self.config(notifier_strategy="qpid") notifier = self.notify_qpid.QpidStrategy() if priority == "info": if exception: self.assertRaises(Exception, notifier.info, test_msg) else: notifier.info(test_msg) elif priority == "warn": if exception: self.assertRaises(Exception, notifier.warn, test_msg) else: notifier.warn(test_msg) elif priority == "error": if exception: self.assertRaises(Exception, notifier.error, test_msg) else: notifier.error(test_msg) self.mocker.VerifyAll()