From 3abcbc306ca1cb57b615c0bbd516bd817d1022db Mon Sep 17 00:00:00 2001
From: Steven Li <stevenli@taosdata.com>
Date: Thu, 9 Jul 2020 08:02:05 +0000
Subject: [PATCH] Added additional protection for crash_gen tool, to capture
 sys.exit() in lower level libraries

---
 tests/pytest/crash_gen.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/pytest/crash_gen.py b/tests/pytest/crash_gen.py
index 542bbfb05f..53957fa96a 100755
--- a/tests/pytest/crash_gen.py
+++ b/tests/pytest/crash_gen.py
@@ -1239,6 +1239,11 @@ class Task():
             self._err = e 
             self._aborted = True
             traceback.print_exc()
+        except BaseException as e :
+            self.logInfo("Python base exception encountered")
+            self._err = e 
+            self._aborted = True
+            traceback.print_exc()
         except :
             self.logDebug("[=] Unexpected exception, SQL: {}".format(self._lastSql))
             raise
-- 
GitLab