Updates tests

2019-04-08 17:43:15 +10:00
parent ec5d6cd388
commit 973c69dffb
4 changed files with 11 additions and 9 deletions
--- a/vulnwhisp/test/mock.py
+++ b/vulnwhisp/test/mock.py
@ -6,12 +6,12 @@ import httpretty
 class mockAPI(object):
    def __init__(self, mock_dir=None, debug=False):
        self.mock_dir = mock_dir
+
        if not self.mock_dir:
            # 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')
-
        if debug:
            self.logger.setLevel(logging.DEBUG)