def test_get_new_session(self, ssh_client_cls): # Arrange host = "localhost" ssh = SSHEntity(host) ssh._ssh_client = ssh_client_cls() # Act session = ssh.get_new_session() # Assert self.assertEquals(ssh._ssh_client.get_transport.call_count, 1)