def run(self): with session_scope() as db_session: update_last_accessed(db_session, endpoint_name=self._endpoint.name) add_request(db_session, duration=self._duration, endpoint_id=self._endpoint.id, ip=self._ip)
def test_add_request(endpoint, session): num_requests = len(endpoint.requests) add_request( session, duration=200, endpoint_id=endpoint.id, ip='127.0.0.1', group_by=None, status_code=200, ) assert count_requests(session, endpoint.id) == num_requests + 1
def run(self): update_duration_cache(endpoint_name=self._endpoint.name, duration=self._duration) with session_scope() as db_session: add_request( db_session, duration=self._duration, endpoint_id=self._endpoint.id, ip=self._ip, group_by=self._group_by, status_code=self._status_code, )
def test_add_request(self): """ Test whether the function returns the right values. """ from flask_monitoringdashboard.database.request import add_request name2 = 'main2' execution_time = 1234 self.assertNotEqual( NAME, name2, 'Both cannot be equal, otherwise the test will fail') with session_scope() as db_session: endpoint = get_endpoint_by_name(db_session, name2) self.assertEqual(count_requests(db_session, endpoint.id), 0) add_request(db_session, execution_time, endpoint.id, ip=IP) self.assertEqual(count_requests(db_session, endpoint.id), 1)
def _on_thread_stopped(self): with session_scope() as db_session: update_last_accessed(db_session, endpoint_name=self._endpoint.name) request_id = add_request(db_session, duration=self._duration, endpoint_id=self._endpoint.id, ip=self._ip) self._lines_body = order_histogram(self._histogram.items()) self.insert_lines_db(db_session, request_id) if self._outlier_profiler: self._outlier_profiler.add_outlier(request_id)
def _on_thread_stopped(self): update_duration_cache(endpoint_name=self._endpoint.name, duration=self._duration) with session_scope() as db_session: request_id = add_request( db_session, duration=self._duration, endpoint_id=self._endpoint.id, ip=self._ip, status_code=self._status_code, group_by=self._group_by, ) self._lines_body = order_histogram(self._histogram.items()) self.insert_lines_db(db_session, request_id) if self._outlier_profiler: self._outlier_profiler.add_outlier(request_id)
def stop(self, duration, status_code): self._exit.set() update_duration_cache(endpoint_name=self._endpoint.name, duration=duration * 1000) with session_scope() as session: request_id = add_request( session, duration=duration * 1000, endpoint_id=self._endpoint.id, ip=self._ip, group_by=self._group_by, status_code=status_code, ) if self._memory: add_outlier( session, request_id, self._cpu_percent, self._memory, self._stacktrace, self._request, )