Change: Remove console output from ssh connection tests

Cleanup the console output while running some ssh connection tests.
greenbone · Apr 3, 2023 · d93a3ab · d93a3ab
1 parent 878d4c3
commit d93a3ab
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/connections/test_ssh_connection.py b/tests/connections/test_ssh_connection.py
@@ -140,8 +140,9 @@ def test_connect_no_known_hosts_file(self):
         ):
             ssh_connection.connect()
 
+    @patch("builtins.print")
     @patch("builtins.input")
-    def test_connect_adding_and_save_hostkey(self, input_mock):
+    def test_connect_adding_and_save_hostkey(self, input_mock, _print_mock):
         key_io = StringIO(
             """-----BEGIN OPENSSH PRIVATE KEY-----
 b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW
@@ -189,8 +190,11 @@ def test_connect_adding_and_save_hostkey(self, input_mock):
                 f"0.0.0.0 {key.get_name()} {key.get_base64()}\n",
             )
 
+    @patch("builtins.print")
     @patch("builtins.input")
-    def test_connect_adding_and_dont_save_hostkey(self, input_mock):
+    def test_connect_adding_and_dont_save_hostkey(
+        self, input_mock, _print_mock
+    ):
         key_io = StringIO(
             """-----BEGIN OPENSSH PRIVATE KEY-----
 b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW