diff options
Diffstat (limited to 'app/test/fetch/test_fetch.py')
-rw-r--r-- | app/test/fetch/test_fetch.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/app/test/fetch/test_fetch.py b/app/test/fetch/test_fetch.py index e446ee5..55d7d4c 100644 --- a/app/test/fetch/test_fetch.py +++ b/app/test/fetch/test_fetch.py @@ -40,6 +40,9 @@ class TestFetch(unittest.TestCase): self.response.status_code = self.requests.codes.ok self.requests.get.return_value = self.response self.requests.post.return_value = self.response + + self.ssh_patcher = patch("discover.fetchers.cli.cli_access.SshConn") + self.ssh_conn = self.ssh_patcher.start().return_value def configure_environment(self): self.env = ENV_CONFIG @@ -74,4 +77,5 @@ class TestFetch(unittest.TestCase): SshConnection.connect = self._ssh_connect SshConnection.check_definitions = self._ssh_conn_check_defs SshConn.check_definitions = self._ssh_check_defs - self.req_patcher.stop()
\ No newline at end of file + self.req_patcher.stop() + self.ssh_patcher.stop()
\ No newline at end of file |