def do_test_response_time(self, sent_at, replied_at):
sender = MagicMock()
type(sender.schedule).time_now = PropertyMock(side_effect = [sent_at, replied_at] )
request = Query(Task(sender), "foo_operation", 1, lambda s: None)
recipient = MagicMock()
request.send_to(recipient)
request.reply_success()
self.assertEqual(replied_at - sent_at, request.response_time)
def do_test_response_time(self, sent_at, replied_at):
sender = MagicMock()
type(sender.schedule).time_now = PropertyMock(side_effect=[sent_at, replied_at])
request = Query(Task(sender), "foo_operation", 1, lambda s: None)
recipient = MagicMock()
request.send_to(recipient)
request.reply_success()
self.assertEqual(replied_at - sent_at, request.response_time)
def test_response_time_on_error(self):
sender = MagicMock()
sender.schedule.time = MagicMock(return_value = (5, 10, 15))
request = Query(Task(sender), "foo_operation", 1, lambda s: None)
recipient = MagicMock()
request.send_to(recipient)
request.reply_error()
with self.assertRaises(AssertionError):
request.response_time
def test_response_time_on_error(self):
sender = MagicMock()
sender.schedule.time = MagicMock(return_value=(5, 10, 15))
request = Query(Task(sender), "foo_operation", 1, lambda s: None)
recipient = MagicMock()
request.send_to(recipient)
request.reply_error()
with self.assertRaises(AssertionError):
request.response_time
