From e2aa695655cd8b6d31ee3e00437b1f1aba0f626d Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Mon, 3 Dec 2012 17:20:18 +0100 Subject: [PATCH] Minor update --- lib/core/testing.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lib/core/testing.py b/lib/core/testing.py index 8d9046509..3f72eead9 100644 --- a/lib/core/testing.py +++ b/lib/core/testing.py @@ -34,18 +34,18 @@ def smokeTest(): count, length = 0, 0 for root, _, files in os.walk(paths.SQLMAP_ROOT_PATH): - if 'extra' in root: + if any(_ in root for _ in ("thirdparty", "extra")): continue for ifile in files: length += 1 for root, _, files in os.walk(paths.SQLMAP_ROOT_PATH): - if 'extra' in root: + if any(_ in root for _ in ("thirdparty", "extra")): continue for ifile in files: - if os.path.splitext(ifile)[1].lower() == '.py' and ifile != '__init__.py': + if os.path.splitext(ifile)[1].lower() == ".py" and ifile != "__init__.py": path = os.path.join(root, os.path.splitext(ifile)[0]) path = path.replace(paths.SQLMAP_ROOT_PATH, '.') path = path.replace(os.sep, '.').lstrip('.')