diff --git a/tools/db_crashtest.py b/tools/db_crashtest.py index b2a04736d..6b352248d 100644 --- a/tools/db_crashtest.py +++ b/tools/db_crashtest.py @@ -67,7 +67,11 @@ def main(argv): + str(ops_per_thread) + "\nwrite_buffer_size=" + str(write_buf_size) + "\n") - dbname = tempfile.mkdtemp(prefix='rocksdb_crashtest_') + test_tmpdir = os.environ.get("TEST_TMPDIR") + if test_tmpdir is None or test_tmpdir == "": + dbname = tempfile.mkdtemp(prefix='rocksdb_crashtest_') + else: + dbname = test_tmpdir + "/rocksdb_crashtest" while time.time() < exit_time: run_had_errors = False diff --git a/tools/db_crashtest2.py b/tools/db_crashtest2.py index 90d202011..9abd6f170 100644 --- a/tools/db_crashtest2.py +++ b/tools/db_crashtest2.py @@ -70,7 +70,12 @@ def main(argv): total_check_mode = 4 check_mode = 0 - dbname = tempfile.mkdtemp(prefix='rocksdb_crashtest_') + test_tmpdir = os.environ.get("TEST_TMPDIR") + if test_tmpdir is None or test_tmpdir == "": + dbname = tempfile.mkdtemp(prefix='rocksdb_crashtest2_') + else: + dbname = test_tmpdir + "/rocksdb_crashtest2" + while time.time() < exit_time: killoption = "" if check_mode == 0: