class ResponseTimeMiddlewareTestcase(GoDjangoTestCase):
def setUp(self):
self.vumi_helper = self.add_helper(
DjangoVumiApiHelper(), setup_vumi_api=False)
self.factory = RequestFactory()
self.mw = ResponseTimeMiddleware()
def test_setting_start_time(self):
request = self.factory.get('/accounts/login/')
self.assertFalse(hasattr(request, 'start_time'))
self.mw.process_request(request)
self.assertTrue(request.start_time)
def test_calculating_response_time(self):
response = HttpResponse('ok')
request = self.factory.get('/accounts/login/')
request.start_time = time.time()
self.mw.process_response(request, response)
response_time_header = response['X-Response-Time']
self.assertTrue(response_time_header)
self.assertTrue(float(response_time_header))
[metric] = self.vumi_helper.amqp_connection.get_metrics()
[datapoint] = metric['datapoints']
self.assertEqual(datapoint[0],
'go.django.django.contrib.auth.views.login.get')
self.assertEqual(datapoint[1], ('avg',))
self.assertTrue(datapoint[2])
def test_method_differentiation(self):
response = HttpResponse('ok')
request = self.factory.post('/accounts/login/')
request.start_time = time.time()
self.mw.process_response(request, response)
[metric] = self.vumi_helper.amqp_connection.get_metrics()
[datapoint] = metric['datapoints']
self.assertEqual(datapoint[0],
'go.django.django.contrib.auth.views.login.post')
class ResponseTimeMiddlewareTestcase(GoDjangoTestCase):
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper(),
setup_vumi_api=False)
self.factory = RequestFactory()
self.mw = ResponseTimeMiddleware()
def test_setting_start_time(self):
request = self.factory.get('/accounts/login/')
self.assertFalse(hasattr(request, 'start_time'))
self.mw.process_request(request)
self.assertTrue(request.start_time)
def test_calculating_response_time(self):
response = HttpResponse('ok')
request = self.factory.get('/accounts/login/')
request.start_time = time.time()
self.mw.process_response(request, response)
response_time_header = response['X-Response-Time']
self.assertTrue(response_time_header)
self.assertTrue(float(response_time_header))
[metric] = self.vumi_helper.amqp_connection.get_metrics()
[datapoint] = metric['datapoints']
self.assertEqual(datapoint[0],
'go.django.django.contrib.auth.views.login.get')
self.assertEqual(datapoint[1], ('avg', ))
self.assertTrue(datapoint[2])
def test_method_differentiation(self):
response = HttpResponse('ok')
request = self.factory.post('/accounts/login/')
request.start_time = time.time()
self.mw.process_response(request, response)
[metric] = self.vumi_helper.amqp_connection.get_metrics()
[datapoint] = metric['datapoints']
self.assertEqual(datapoint[0],
'go.django.django.contrib.auth.views.login.post')
def setUp(self):
self.vumi_helper = self.add_helper(
DjangoVumiApiHelper(), setup_vumi_api=False)
self.factory = RequestFactory()
self.mw = ResponseTimeMiddleware()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper(),
setup_vumi_api=False)
self.factory = RequestFactory()
self.mw = ResponseTimeMiddleware()
