Add more unit tests for wsman.py module

Signed-off-by: Mihaela Balutoiu <[email protected]>
cloudbase · Mar 26, 2024 · 7296019 · 7296019
1 parent 75f6c39
commit 7296019
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/coriolis/tests/test_wsman.py b/coriolis/tests/test_wsman.py
@@ -122,6 +122,14 @@ def test__exec_command_exception(self):
             mock.ANY, mock.ANY)
         self.conn._protocol.close_shell.assert_called_once_with(mock.ANY)
 
+    def test__exec_command_invalid_credentials(self):
+        self.conn._protocol.open_shell.side_effect = (
+            wsman.winrm_exceptions.InvalidCredentialsError)
+
+        self.assertRaises(exception.NotAuthorized, self.conn._exec_command,
+                          self.cmd, self.args)
+        self.conn._protocol.close_shell.assert_not_called()
+
     def test_exec_command(self):
         self.conn._protocol.get_command_output.return_value = (
             "std_out", "std_err", 0)