Преглед изворни кода

Stresstests: envvar C_SLEEP can now be used to add tracebacks to sleep calls

Ask Solem пре 11 година
родитељ
комит
2b4a3a7ff6
1 измењених фајлова са 15 додато и 0 уклоњено
  1. 15 0
      funtests/stress/stress/__init__.py

+ 15 - 0
funtests/stress/stress/__init__.py

@@ -1,4 +1,19 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import
 
+import os
+import time
+
+if os.environ.get('C_SLEEP'):
+
+    _orig_sleep = time.sleep
+
+    def _sleep(n):
+        print('WARNING: Time sleep for {0}s'.format(n))
+        import traceback
+        traceback.print_stack()
+        _orig_sleep(n)
+    time.sleep = _sleep
+
+
 from .app import app  # noqa