fix bugs with request time header

Author: noise

flask_queryinspect.py

@@ -88,6 +88,6 @@ def after_request(self, response, *kw1, **kw2):
             else:
                 response.headers['X-QueryInspect-Num-SQL-Queries'] = qi['reads'] + qi['writes']
                 #response.headers['X-QueryInspect-Duplicate-SQL-Queries'] = qi['dupes']
-                response.headers['X-QueryInspect-Total-SQL-Time'] = qi['q_time']
-                response.headers['X-QueryInspect-Total-Request-Time'] = qi['r_time']
+                response.headers['X-QueryInspect-Total-SQL-Time'] = qi['q_time_ms']
+                response.headers['X-QueryInspect-Total-Request-Time'] = qi['r_time_ms']
         return response

test_queryinspect.py

@@ -68,6 +68,8 @@ def test_rtime(self):
         self.app.config['QUERYINSPECT_HEADERS_COMBINED'] = False
         with self.app.test_client() as c:
             res = c.get('/slow')
+            log.debug(res.headers)
             r_time = float(res.headers['X-QueryInspect-Total-Request-Time'])
-            self.assertTrue(90.0 < r_time > 110.0)
+            log.debug('r_time: %r', r_time)
+            self.assertTrue(90.0 < r_time < 110.0)
         self.app.config['QUERYINSPECT_HEADERS_COMBINED'] = True