diff options
Diffstat (limited to 'plugins/agl_test_base.py')
| -rw-r--r-- | plugins/agl_test_base.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/plugins/agl_test_base.py b/plugins/agl_test_base.py index b30f841..1491f64 100644 --- a/plugins/agl_test_base.py +++ b/plugins/agl_test_base.py @@ -1,6 +1,8 @@ import pytest import os import json +import pathlib +import shutil import plugins.agl_test_utils as utils import plugins.agl_test_conf as conf @@ -111,6 +113,9 @@ class AGLBaseTest: #Write json data to html report.change_json_to_html(self.name) + #Clean temp files and folders + self.clean_self_tmp() + def precheck(self): return True; @@ -133,3 +138,23 @@ class AGLBaseTest: test_suites[self.name] = test_info json.dump(test_suites,f) f.close() + + def prepare_local_log_mode(self): + # Get env 'LOG_MODE' + env_dist=os.environ + local_log_mode = str(env_dist.get("LOG_MODE")) + return local_log_mode + + def clean_self_tmp(self): + path = self.get_temp_logdir() + log_mode = self.prepare_local_log_mode() + if log_mode == "slim" or log_mode == "clear": + # leave <path>/log and report.json as essential + folderObj = pathlib.Path(path) + for element in folderObj.iterdir(): + if element.name != "log" and element.name != "report.json": + # check delete_file_path file type + if element.is_file(): + element.unlink() + elif element.is_dir(): + shutil.rmtree(path) |