Cyberdefense/VulnWhisperer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updates tests

Browse Source
This commit is contained in:
pemontto
2019-04-08 17:43:15 +10:00
parent ec5d6cd388
commit 973c69dffb
4 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
.gitmodules vendored
View File

@ -1,3 +1,3 @@
[submodule "test"] [submodule "tests/data"]
path = test path = tests/data
url = https://github.com/HASecuritySolutions/VulnWhisperer-tests url = https://github.com/HASecuritySolutions/VulnWhisperer-tests

11
bin/vuln_whisperer
View File

@ -37,10 +37,13 @@ def main():
help='The NESSUS username', type=lambda x: x.strip()) help='The NESSUS username', type=lambda x: x.strip())
parser.add_argument('-p', '--password', dest='password', required=False, default=None, parser.add_argument('-p', '--password', dest='password', required=False, default=None,
help='The NESSUS password', type=lambda x: x.strip()) help='The NESSUS password', type=lambda x: x.strip())
parser.add_argument('-F', '--fancy', action='store_true', help='Enable colourful logging output') parser.add_argument('-F', '--fancy', action='store_true',
parser.add_argument('-d', '--debug', action='store_true', help='Enable debugging messages') help='Enable colourful logging output')
parser.add_argument('--mock', action='store_true', help='Enable mocked API responses') parser.add_argument('-d', '--debug', action='store_true',
parser.add_argument('--mock_dir', dest='mock_dir', required=False, default='test', help='Enable debugging messages')
parser.add_argument('--mock', action='store_true',
help='Enable mocked API responses')
parser.add_argument('--mock_dir', dest='mock_dir', required=False, default=None,
help='Path of test directory') help='Path of test directory')
args = parser.parse_args() args = parser.parse_args()

1
test

Submodule test deleted from 606b8bcbe3

4
vulnwhisp/test/mock.py
View File

@ -6,12 +6,12 @@ import httpretty
class mockAPI(object): class mockAPI(object):
def __init__(self, mock_dir=None, debug=False): def __init__(self, mock_dir=None, debug=False):
self.mock_dir = mock_dir self.mock_dir = mock_dir
if not self.mock_dir: if not self.mock_dir:
# Try to guess the mock_dir if python setup.py develop was used # Try to guess the mock_dir if python setup.py develop was used
self.mock_dir = '/'.join(__file__.split('/')[:-3]) + '/test' self.mock_dir = '/'.join(__file__.split('/')[:-3]) + '/tests/data'
self.logger = logging.getLogger('mockAPI') self.logger = logging.getLogger('mockAPI')
if debug: if debug:
self.logger.setLevel(logging.DEBUG) self.logger.setLevel(logging.DEBUG)