def test_get_running_trace(self): ready = threading.Event() done = threading.Event() def worker(): inner() def inner(): ready.set() done.wait() t = concurrent.thread(worker, name="Test") t.start() try: if not ready.wait(1): raise RuntimeError("Timeout waiting for worker thread") formatted_traceback = concurrent.format_traceback(t.ident) finally: done.set() t.join() # The functions called from the worker thread should appear in the # traceback. self.assertIn("in worker", formatted_traceback) self.assertIn("in inner", formatted_traceback)
def test_get_running_trace(self): ready = threading.Event() done = threading.Event() def worker(): inner() def inner(): ready.set() done.wait() t = concurrent.thread(worker, name="Test") t.start() try: if not ready.wait(1): raise RuntimeError("Timeout waiting for worker thread") formatted_traceback = concurrent.format_traceback(t.ident) finally: done.set() t.join() # The functions called from the worker thread should appear in the # traceback. self.assertIn("in worker", formatted_traceback) self.assertIn("in inner", formatted_traceback)
def test_get_running_trace(self): def worker(): time.sleep(2) t = concurrent.thread(worker, name="Test") t.start() # yield control to let the Test thread to start time.sleep(1) try: formatted_traceback = concurrent.format_traceback(t.ident) self.assertIn("time.sleep(2)", formatted_traceback) finally: t.join()
def _check_task(self, task_number): with self._lock: if task_number != self._task_counter: return if self._task.discard: if self._discarded: raise AssertionError("Attempt to discard worker twice") self._discarded = True else: self._scheduled_check = self._check_after(self._task.timeout) if self._discarded: # Please make sure the executor call is performed outside the lock # -- there is another lock involved in the executor and we don't # want to fall into a deadlock incidentally. self._executor._worker_discarded(self) self._log.info("Worker discarded: %s", self) else: # we want to avoid to log with the lock held, so we do it here. try: trace = concurrent.format_traceback(self._thread.ident) except KeyError: trace = "(traceback not available)" self._log.warning("Worker blocked: %s, traceback:\n%s", self, trace)
def _check_task(self, task_number): with self._lock: if task_number != self._task_counter: return if self._task.discard: if self._discarded: raise AssertionError("Attempt to discard worker twice") self._discarded = True else: self._scheduled_check = self._check_after(self._task.timeout) if self._discarded: # Please make sure the executor call is performed outside the lock # -- there is another lock involved in the executor and we don't # want to fall into a deadlock incidentally. self._executor._worker_discarded(self) self._log.info("Worker discarded: %s", self) else: # we want to avoid to log with the lock held, so we do it here. try: trace = concurrent.format_traceback(self._thread.ident) except KeyError: trace = "(traceback not available)" self._log.warning("Worker blocked: %s, traceback:\n%s", self, trace)
def test_get_wrong_id_trace(self, ident): with pytest.raises(KeyError): concurrent.format_traceback(ident)