cloudshell-cli 4.0.5__zip → 4.1.1__zip

{cloudshell-cli-4.0.5/cloudshell_cli.egg-info → cloudshell-cli-4.1.1}/PKG-INFO

@@ -1,14 +1,11 @@
 Metadata-Version: 2.1
 Name: cloudshell-cli
-Version: 4.0.5
+Version: 4.1.1
 Summary: QualiSystems CloudShell CLI support package
 Home-page: http://www.qualisystems.com/
 Author: QualiSystems
 Author-email: info@qualisystems.com
-License: UNKNOWN
-Platform: UNKNOWN
 Requires-Python: >=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*, !=3.5.*, !=3.6.*, <4
 License-File: LICENSE
 
-UNKNOWN
-
+QualiSystems CloudShell CLI support package

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/configurator.py

@@ -2,7 +2,13 @@
 # -*- coding: utf-8 -*-
 import sys
 from abc import ABCMeta, abstractmethod
+from collections import defaultdict
 
+from cloudshell.cli.factory.session_factory import (
+    CloudInfoAccessKeySessionFactory,
+    GenericSessionFactory,
+    SessionFactory,
+)
 from cloudshell.cli.service.cli import CLI
 from cloudshell.cli.session.ssh_session import SSHSession
 from cloudshell.cli.session.telnet_session import TelnetSession
@@ -16,39 +22,33 @@ else:
 
 
 class CLIServiceConfigurator(object):
-    REGISTERED_SESSIONS = (SSHSession, TelnetSession)
-
-    def __init__(self, resource_config, logger, cli=None, registered_sessions=None):
+    REGISTERED_SESSIONS = (CloudInfoAccessKeySessionFactory(SSHSession), TelnetSession)
+    """Using factories instead of """
+
+    def __init__(
+        self,
+        resource_config,
+        logger,
+        cli=None,
+        registered_sessions=None,
+        access_key=None,
+        access_key_passphrase=None,
+    ):
         """Initialize CLI service configurator.
 
         :param cloudshell.shell.standards.resource_config_generic_models.GenericCLIConfig resource_config:  # noqa: E501
         :param logging.Logger logger:
         :param cloudshell.cli.service.cli.CLI cli:
         :param registered_sessions: Session types and order
+        :param access_key: access key for the resource
+        :param access_key_passphrase: access key passphrase for the resource
         """
         self._cli = cli or CLI()
         self._resource_config = resource_config
         self._logger = logger
         self._registered_sessions = registered_sessions or self.REGISTERED_SESSIONS
-
-    @property
-    def _username(self):
-        return self._resource_config.user
-
-    @property
-    @lru_cache()
-    def _password(self):
-        return self._resource_config.password
-
-    @property
-    def _resource_address(self):
-        """Resource IP."""
-        return self._resource_config.address
-
-    @property
-    def _port(self):
-        """Connection port property, to open socket on."""
-        return self._resource_config.cli_tcp_port
+        self._access_key = access_key
+        self._access_key_passphrase = access_key_passphrase
 
     @property
     def _cli_type(self):
@@ -58,29 +58,28 @@ class CLIServiceConfigurator(object):
     @property
     @lru_cache()
     def _session_dict(self):
-        return {sess.SESSION_TYPE.lower(): [sess] for sess in self._registered_sessions}
+        session_dict = defaultdict(list)
+        for sess in self._registered_sessions:
+            session_dict[sess.SESSION_TYPE.lower()].append(sess)
+        return session_dict
 
     def _on_session_start(self, session, logger):
-        """Perform some default commands when session just opened.
-
-        Like 'no logging console'
-        """
         pass
 
-    @property
-    @lru_cache()
-    def _session_kwargs(self):
-        return {
-            "host": self._resource_address,
-            "username": self._username,
-            "password": self._password,
-            "port": self._port,
-            "on_session_start": self._on_session_start,
-        }
+    def initialize_session(self, session):
+        if not isinstance(session, SessionFactory):
+            session = GenericSessionFactory(session)
+        return session.init_session(
+            self._resource_config,
+            self._logger,
+            on_session_start=self._on_session_start,
+            access_key=self._access_key,
+            access_key_passphrase=self._access_key_passphrase,
+        )
 
     def _defined_sessions(self):
         return [
-            sess(**self._session_kwargs)
+            self.initialize_session(sess)
             for sess in self._session_dict.get(
                 self._cli_type.lower(), self._registered_sessions
             )

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/examples/cli_demo/connect_to_switch/SwitchClihandler.py

@@ -96,7 +96,6 @@ class SwitchCliHandler(CliHandlerImpl):
         session.hardware_expect("exit", EnableCommandMode.PROMPT, logger)
 
     def enter_enable_mode(self, session, logger):
-
         enable_password = get_attribute_by_name(
             attribute_name="Enable Password", context=self._context
         )

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/examples/cli_demo/connect_to_switch/runner.py

@@ -40,6 +40,6 @@ if __name__ == "__main__":
         out = session.send_command("echo checking switch")
         with session.enter_mode(cli_handler.config_mode) as config_session:
             out = config_session.send_command("echo checking switch")
-            print(out)  # noqa: T001
+            print(out)  # noqa: T201
             out = config_session.send_command("echo checking switch")
-            print(out)  # noqa: T001
+            print(out)  # noqa: T201

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/examples/cli_demo/connect_to_ubuntu/runnur.py

@@ -13,7 +13,6 @@ class CliCommandMode(CommandMode):
     EXIT_COMMAND = "exit"
 
     def __init__(self, context):
-
         CommandMode.__init__(
             self,
             CliCommandMode.PROMPT,
@@ -45,6 +44,6 @@ if __name__ == "__main__":
 
     with cli.get_session(session, default_mode) as default_session:
         out = default_session.send_command("echo Cli Demo connected to Ubuntu Machine")
-        print(out)  # noqa: T001
+        print(out)  # noqa: T201
         fd = open("<src_file_path>")
         default_session.session.upload_scp(fd, "<dst_file_path>", "<file_size>", "0666")

cloudshell-cli-4.1.1/cloudshell/cli/factory/session_factory.py

@@ -0,0 +1,96 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+from abc import ABCMeta, abstractmethod
+
+ABC = ABCMeta("ABC", (object,), {"__slots__": ()})
+
+
+class SessionFactory(ABC):
+    """Session factory.
+
+    Help to initialize session for specified session class.
+    """
+
+    def __init__(self, session_class):
+        """:param session_class: Session class."""
+        self.session_class = session_class
+
+    @abstractmethod
+    def init_session(
+        self,
+        resource_config,
+        logger,
+        on_session_start=None,
+        access_key=None,
+        access_key_passphrase=None,
+    ):
+        """Initialize session instance.
+
+        Encapsulate the logic of the session instance creation.
+        :param resource_config:
+        :param logging.Logger logger:
+        :param on_session_start: function that will be called on session start
+        :param access_key: access key for the resource
+        :param access_key_passphrase: access key passphrase for the resource
+        """
+        raise NotImplementedError
+
+
+class GenericSessionFactory(SessionFactory):
+    def init_session(
+        self,
+        resource_config,
+        logger,
+        on_session_start=None,
+        access_key=None,
+        access_key_passphrase=None,
+    ):
+        return self.session_class(
+            **self._session_kwargs(
+                resource_config,
+                logger,
+                on_session_start,
+                access_key,
+                access_key_passphrase,
+            )
+        )
+
+    @property
+    def SESSION_TYPE(self):
+        return self.session_class.SESSION_TYPE
+
+    def _session_kwargs(
+        self,
+        resource_config,
+        logger,
+        on_session_start,
+        access_key,
+        access_key_passphrase,
+    ):
+        return {
+            "host": resource_config.address,
+            "username": resource_config.user,
+            "password": resource_config.password,
+            "port": resource_config.cli_tcp_port,
+            "on_session_start": on_session_start,
+        }
+
+
+class CloudInfoAccessKeySessionFactory(GenericSessionFactory):
+    def _session_kwargs(
+        self,
+        resource_config,
+        logger,
+        on_session_start,
+        access_key,
+        access_key_passphrase,
+    ):
+        return {
+            "host": resource_config.address,
+            "username": resource_config.user,
+            "password": resource_config.password,
+            "port": resource_config.cli_tcp_port,
+            "pkey": access_key,
+            "pkey_passphrase": access_key_passphrase,
+            "on_session_start": on_session_start,
+        }

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/service/cli_service_impl.py

@@ -158,7 +158,8 @@ class CliServiceImpl(CliService):
             steps = CommandModeHelper.calculate_route_steps(
                 self.command_mode, requested_command_mode
             )
-            list(map(lambda x: x(self, self._logger), steps))
+            for s in steps:
+                s(self, self._logger)
 
     def reconnect(self, timeout=None):
         """Reconnect session, keep current command mode.

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/service/command_mode_helper.py

@@ -35,7 +35,7 @@ class CommandModeHelper(NodeOperations):
 
         :rtype: OrderedDict
         """
-        # noqa
+
         def _get_child_nodes(command_node):
             return reduce(
                 lambda x, y: x + _get_child_nodes(y),
@@ -60,7 +60,7 @@ class CommandModeHelper(NodeOperations):
 
         :rtype: dict
         """
-        # noqa
+
         def _create_child_modes(instance, child_dict):
             instance_dict = {}
             for mode_type in child_dict:

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/session/expect_session.py

@@ -1,4 +1,5 @@
 import re
+import socket
 import time
 from abc import ABCMeta, abstractmethod
 from collections import OrderedDict
@@ -158,6 +159,51 @@ class ExpectSession(Session, ABC):
                         self.__class__.__name__, "Socket closed by timeout"
                     )
 
+    def _receive(self, timeout, logger):
+        """Read session's buffer.
+
+        :type timeout: float
+        :type logger: logging.Logger
+        :rtype: str
+        """
+        timeout = timeout or self._timeout
+        self._set_timeout(timeout)
+        try:
+            data = self._read_str_data()
+        except socket.timeout:
+            raise SessionReadTimeout
+        if not data:
+            raise SessionReadEmptyData
+        return data
+
+    @abstractmethod
+    def _set_timeout(self, timeout):
+        pass
+
+    def _read_str_data(self):
+        byte_data = b""
+        for _ in range(5):
+            new_data = self._read_byte_data()
+
+            if not new_data:
+                str_data = byte_data.decode()
+                break
+
+            byte_data += new_data
+            try:
+                str_data = byte_data.decode()
+            except UnicodeDecodeError:
+                continue  # byte data can end with a wrong utf8 symbol, read more
+            else:
+                break
+        else:
+            str_data = byte_data.decode()
+        return str_data
+
+    @abstractmethod
+    def _read_byte_data(self):
+        pass
+
     def _generate_command_pattern(self, command):
         """Generate command_pattern.
 
@@ -262,7 +308,6 @@ class ExpectSession(Session, ABC):
             self._loop_detector_max_combination_length,
         )
         while retries == 0 or retries_count < retries:
-
             read_buffer = self._receive_all(timeout, logger)
 
             if read_buffer:

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/session/scpi_session.py

@@ -38,7 +38,6 @@ class SCPISession(TCPSession):
         remove_command_from_output=True,
         **optional_args
     ):
-
         if ";:system:error?" not in command.lower():
             command += ";:system:error?"
 

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/session/ssh_session.py

@@ -1,15 +1,11 @@
-import socket
+from io import StringIO
 
 import paramiko
 from scp import SCPClient
 
 from cloudshell.cli.session.connection_params import ConnectionParams
 from cloudshell.cli.session.expect_session import ExpectSession
-from cloudshell.cli.session.session_exceptions import (
-    SessionException,
-    SessionReadEmptyData,
-    SessionReadTimeout,
-)
+from cloudshell.cli.session.session_exceptions import SessionException
 
 
 class SSHSessionException(SessionException):
@@ -28,6 +24,7 @@ class SSHSession(ExpectSession, ConnectionParams):
         port=None,
         on_session_start=None,
         pkey=None,
+        pkey_passphrase=None,
         *args,
         **kwargs
     ):
@@ -42,6 +39,7 @@ class SSHSession(ExpectSession, ConnectionParams):
         self.username = username
         self.password = password
         self.pkey = pkey
+        self.pkey_passphrase = pkey_passphrase
 
         self._handler = None
         self._current_channel = None
@@ -58,6 +56,7 @@ class SSHSession(ExpectSession, ConnectionParams):
                 self.username == other.username,
                 self.password == other.password,
                 self.pkey == other.pkey,
+                self.pkey_passphrase == other.pkey_passphrase,
             ]
         )
 
@@ -86,7 +85,7 @@ class SSHSession(ExpectSession, ConnectionParams):
                 banner_timeout=30,
                 allow_agent=False,
                 look_for_keys=False,
-                pkey=self.pkey,
+                pkey=self._get_pkey_object(self.pkey, self.pkey_passphrase, logger),
             )
         except Exception as e:
             logger.exception("Failed to initialize session:")
@@ -122,26 +121,11 @@ class SSHSession(ExpectSession, ConnectionParams):
         """
         self._current_channel.send(command)
 
-    def _receive(self, timeout, logger):
-        """Read session buffer.
-
-        :param int timeout: time between retries
-        :param logging.Logger logger:
-        """
-        # Set the channel timeout
-        timeout = timeout if timeout else self._timeout
+    def _set_timeout(self, timeout):
         self._current_channel.settimeout(timeout)
 
-        try:
-            byte_data = self._current_channel.recv(self._buffer_size)
-            data = byte_data.decode()
-        except socket.timeout:
-            raise SessionReadTimeout()
-
-        if not data:
-            raise SessionReadEmptyData()
-
-        return data
+    def _read_byte_data(self):
+        return self._current_channel.recv(self._buffer_size)
 
     def upload_scp(
         self, file_stream, dest_pathname, file_size=None, dest_permissions="0601"
@@ -182,3 +166,18 @@ class SSHSession(ExpectSession, ConnectionParams):
         sftp.putfo(file_stream, dest_pathname)
         sftp.chmod(dest_pathname, int(dest_permissions, base=8))
         sftp.close()
+
+    @staticmethod
+    def _get_pkey_object(key_material, passphrase, logger):
+        """Try to detect private key type and return paramiko.PKey object."""
+        if not key_material:
+            return None
+
+        for cls in [paramiko.RSAKey, paramiko.DSSKey, paramiko.ECDSAKey]:
+            try:
+                key = cls.from_private_key(StringIO(key_material), password=passphrase)
+            except paramiko.ssh_exception.SSHException as e:
+                # Invalid key, try other key type
+                logger.warning(e)
+            else:
+                return key

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/session/tcp_session.py

@@ -2,14 +2,9 @@ import socket
 
 from cloudshell.cli.session.connection_params import ConnectionParams
 from cloudshell.cli.session.expect_session import ExpectSession
-from cloudshell.cli.session.session_exceptions import (
-    SessionReadEmptyData,
-    SessionReadTimeout,
-)
 
 
 class TCPSession(ExpectSession, ConnectionParams):
-
     SESSION_TYPE = "TCP"
     BUFFER_SIZE = 1024
 
@@ -49,17 +44,11 @@ class TCPSession(ExpectSession, ConnectionParams):
         """
         self._handler.sendall(command)
 
-    def _receive(self, timeout, logger):
-        """Read session buffer."""
-        timeout = timeout if timeout else self._timeout
+    def _set_timeout(self, timeout):
         self._handler.settimeout(timeout)
 
-        try:
-            data = self._handler.recv(self._buffer_size)
-        except socket.timeout:
-            raise SessionReadTimeout()
-
-        if not data:
-            raise SessionReadEmptyData()
+    def _read_byte_data(self):
+        pass
 
-        return data
+    def _read_str_data(self):
+        return self._handler.recv(self._buffer_size)

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell/cli/session/telnet_session.py

@@ -1,14 +1,9 @@
-import socket
 import telnetlib
 from collections import OrderedDict
 
 from cloudshell.cli.session.connection_params import ConnectionParams
 from cloudshell.cli.session.expect_session import ExpectSession
-from cloudshell.cli.session.session_exceptions import (
-    SessionException,
-    SessionReadEmptyData,
-    SessionReadTimeout,
-)
+from cloudshell.cli.session.session_exceptions import SessionException
 
 
 class TelnetSessionException(SessionException):
@@ -100,18 +95,8 @@ class TelnetSession(ExpectSession, ConnectionParams):
         byte_command = command.encode()
         self._handler.write(byte_command)
 
-    def _receive(self, timeout, logger):
-        """Read session buffer."""
-        timeout = timeout if timeout else self._timeout
+    def _set_timeout(self, timeout):
         self._handler.get_socket().settimeout(timeout)
 
-        try:
-            byte_data = self._handler.read_some()
-        except socket.timeout:
-            raise SessionReadTimeout()
-
-        if not byte_data:
-            raise SessionReadEmptyData()
-
-        data = byte_data.decode()
-        return data
+    def _read_byte_data(self):
+        self._handler.read_some()

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1/cloudshell_cli.egg-info}/PKG-INFO

@@ -1,14 +1,11 @@
 Metadata-Version: 2.1
 Name: cloudshell-cli
-Version: 4.0.5
+Version: 4.1.1
 Summary: QualiSystems CloudShell CLI support package
 Home-page: http://www.qualisystems.com/
 Author: QualiSystems
 Author-email: info@qualisystems.com
-License: UNKNOWN
-Platform: UNKNOWN
 Requires-Python: >=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*, !=3.5.*, !=3.6.*, <4
 License-File: LICENSE
 
-UNKNOWN
-
+QualiSystems CloudShell CLI support package

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell_cli.egg-info/SOURCES.txt

@@ -26,6 +26,8 @@ cloudshell/cli/examples/cli_demo/connect_to_switch/runner.py
 cloudshell/cli/examples/cli_demo/connect_to_ubuntu/__init__.py
 cloudshell/cli/examples/cli_demo/connect_to_ubuntu/readfrom.txt
 cloudshell/cli/examples/cli_demo/connect_to_ubuntu/runnur.py
+cloudshell/cli/factory/__init__.py
+cloudshell/cli/factory/session_factory.py
 cloudshell/cli/service/__init__.py
 cloudshell/cli/service/cli.py
 cloudshell/cli/service/cli_exception.py
@@ -57,6 +59,7 @@ cloudshell_cli.egg-info/dependency_links.txt
 cloudshell_cli.egg-info/requires.txt
 cloudshell_cli.egg-info/top_level.txt
 tests/__init__.py
+tests/conftest.py
 tests/cli/__init__.py
 tests/cli/test_cli.py
 tests/cli/test_cli_service_impl.py
@@ -71,5 +74,6 @@ tests/cli/command_template/test_command_template.py
 tests/cli/session/__init__.py
 tests/cli/session/test_connection_params.py
 tests/cli/session/test_expect_session.py
+tests/cli/session/test_expect_session_py3.py
 tests/cli/session/test_ssh_session.py
 tests/cli/session/test_telnet_session.py

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/cloudshell_cli.egg-info/requires.txt

@@ -13,3 +13,4 @@ PyNaCl<1.5,>=1.0.1
 bcrypt~=3.2.0
 cryptography<37,>=3.3
 PyNaCl>=1.5
+six

cloudshell-cli-4.1.1/dev_requirements.txt

@@ -0,0 +1,4 @@
+pre-commit
+tox
+-r test_requirements.txt
+-r requirements.txt

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/requirements.txt

@@ -7,3 +7,4 @@ PyNaCl>=1.5 ; python_version > "3.7"
 paramiko>=2.6,<2.7
 scp>=0.13,<1
 functools32; python_version <= '2.7'
+six; python_version > "3.7"

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/setup.py

@@ -26,5 +26,6 @@ setup(
     version=version_from_file,
     package_data={"": ["*.txt"]},
     description="QualiSystems CloudShell CLI support package",
+    long_description="QualiSystems CloudShell CLI support package",
     include_package_data=True,
 )

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/tests/cli/session/test_expect_session.py

@@ -15,7 +15,7 @@ except ImportError:
     from mock import MagicMock, Mock, call, patch
 
 
-class TestExpectSessionException(Exception):
+class ExpectSessionException(Exception):
     pass
 
 
@@ -32,10 +32,13 @@ class ExpectSessionImpl(ExpectSession):
     def disconnect(self):
         pass
 
-    def _receive(self, timeout, logger):
+    def _send(self, command, logger):
         pass
 
-    def _send(self, command, logger):
+    def _read_byte_data(self):
+        pass
+
+    def _set_timeout(self, timeout):
         pass
 
 
@@ -98,7 +101,7 @@ class TestExpectSession(TestCase):
             self._instance._clear_buffer(timeout, self._logger)
 
     def test_clear_buffer_exit_with_no_data(self, normalize_buffer, loops_detected):
-        self._receive.side_effect = ["", TestExpectSessionException("Breaking loop")]
+        self._receive.side_effect = ["", ExpectSessionException("Breaking loop")]
         timeout = Mock()
         self._instance._clear_buffer(timeout, self._logger)
         self.assertTrue(True)
@@ -109,7 +112,7 @@ class TestExpectSession(TestCase):
         self._receive.side_effect = [
             "test",
             "",
-            TestExpectSessionException("Breaking loop"),
+            ExpectSessionException("Breaking loop"),
         ]
         timeout = Mock()
         self._instance._clear_buffer(timeout, self._logger)

cloudshell-cli-4.1.1/tests/cli/session/test_expect_session_py3.py

@@ -0,0 +1,13 @@
+from tests.cli.session.test_expect_session import ExpectSessionImpl
+
+
+def test_expect_session_read_wrong_utf8_symbol(logger):
+    class TestSession(ExpectSessionImpl):
+        data = [b"\xe2\x80\x99hi\xe2", b"\x80\x99"]  # ’hi’
+
+        def _read_byte_data(self):
+            return self.data.pop(0)
+
+    session = TestSession()
+
+    assert session._receive(1, logger) == "’hi’"

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/tests/cli/session/test_ssh_session.py

@@ -46,7 +46,7 @@ y9/ziFJPPMN703N1dJ0e1oRWpfSFUzlGn1OgF1EY9qQTwQK4PL9bhyJi1/MswZHB
 /XmDlKr7An89Iy+B/Q9ncja2FfK49AP23AMYKhdWBrEq+enXwjkr5Uz8ffD3eagD
 VZxBlCGUp4bmlS5s7sttMZRvJq0CXfxy+q8Qe0sz/CyOAN/J8iZu8AJyp04DQk4d
 -----END RSA PRIVATE KEY-----
-"""
+"""  # noqa
 KEY_PASSPHRASE = "quali1234"
 
 
@@ -443,9 +443,6 @@ class TestSshSession(TestCase):
             )
         )
 
-        pkey = paramiko.RSAKey.from_private_key(
-            StringIO(KEY_WITH_PASSPHRASE), password=KEY_PASSPHRASE
-        )
         self.assertFalse(
             self._instance.__eq__(
                 SSHSession(
@@ -454,23 +451,22 @@ class TestSshSession(TestCase):
                     "",
                     port=self._port,
                     on_session_start=self._on_session_start,
-                    pkey=pkey,
+                    pkey=KEY_WITH_PASSPHRASE,
+                    pkey_passphrase=KEY_PASSPHRASE,
                 )
             )
         )
 
     @patch("cloudshell.cli.session.ssh_session.ExpectSession")
     def test_eq_rsa(self, expect_session):
-        pkey = paramiko.RSAKey.from_private_key(
-            StringIO(KEY_WITH_PASSPHRASE), password=KEY_PASSPHRASE
-        )
         self._instance = SSHSession(
             self._hostname,
             self._username,
             self._password,
             port=self._port,
             on_session_start=self._on_session_start,
-            pkey=pkey,
+            pkey=KEY_WITH_PASSPHRASE,
+            pkey_passphrase=KEY_PASSPHRASE,
         )
 
         self.assertTrue(
@@ -481,7 +477,8 @@ class TestSshSession(TestCase):
                     self._password,
                     port=self._port,
                     on_session_start=self._on_session_start,
-                    pkey=pkey,
+                    pkey=KEY_WITH_PASSPHRASE,
+                    pkey_passphrase=KEY_PASSPHRASE,
                 )
             )
         )
@@ -640,16 +637,13 @@ class TestSshSession(TestCase):
             "",
             port=server.port,
             on_session_start=self._on_session_start,
-            pkey=pkey,
+            pkey=KEY_WITH_PASSPHRASE,
+            pkey_passphrase=KEY_PASSPHRASE,
         )
         self._instance.connect(">", logger=Mock())
         self._instance.hardware_expect("dummy command", ">", Mock())
 
     def test_rsa_failure(self):
-        pkey = paramiko.RSAKey.from_private_key(
-            StringIO(KEY_WITH_PASSPHRASE), password=KEY_PASSPHRASE
-        )
-
         server = SSHServer(user2key={})
 
         with self.assertRaises(SSHSessionException):
@@ -659,7 +653,8 @@ class TestSshSession(TestCase):
                 "",
                 port=server.port,
                 on_session_start=self._on_session_start,
-                pkey=pkey,
+                pkey=KEY_WITH_PASSPHRASE,
+                pkey_passphrase=KEY_PASSPHRASE,
             )
             self._instance.connect(">", logger=Mock())
             self._instance.hardware_expect("dummy command", ">", Mock())

cloudshell-cli-4.1.1/tests/conftest.py

@@ -0,0 +1,15 @@
+import logging
+import sys
+
+import pytest
+
+
+@pytest.fixture()
+def logger():
+    return logging.getLogger(__file__)
+
+
+def pytest_ignore_collect(path):
+    if sys.version_info.major == 2:
+        if str(path).endswith("_py3.py"):
+            return True

{cloudshell-cli-4.0.5 → cloudshell-cli-4.1.1}/tox.ini

@@ -19,7 +19,7 @@ commands =
     pytest --cov=cloudshell.cli tests --cov-report=xml
 
 [testenv:pre-commit]
-basepython = python3
+basepython = python3.9
 skip_install = true
 deps = pre-commit
 commands = pre-commit run --all-files --show-diff-on-failure

cloudshell-cli-4.1.1/version.txt

@@ -0,0 +1 @@
+4.1.1

cloudshell-cli-4.0.5/dev_requirements.txt

@@ -1,5 +0,0 @@
-pre-commit
-tox
-tox-factor
--r test_requirements.txt
---extra-index-url https://test.pypi.org/simple -r requirements.txt

cloudshell-cli-4.0.5/version.txt

@@ -1 +0,0 @@
-4.0.5