def test_put_relative_path(self): conn_params = {"hostname": "dummy.host.org", "username": "******"} client = ParamikoSSHClient(**conn_params) mock_client = Mock() mock_sftp_client = Mock() mock_transport = Mock() mock_client.get_transport.return_value = mock_transport mock_sftp_client.getcwd.return_value = "/mock/cwd" client.client = mock_client client.sftp_client = mock_sftp_client result = client.put(path="path2.txt", contents="foo bar 2", chmod=466, mode="a") self.assertEqual(result, "/mock/cwd/path2.txt") calls = [call(".")] mock_sftp_client.chdir.assert_has_calls(calls, any_order=False) calls = [ call("path2.txt", mode="a"), call().write("foo bar 2"), call().chmod(466), call().close(), ] mock_sftp_client.file.assert_has_calls(calls, any_order=False)
def test_put_relative_path(self): conn_params = {'hostname': 'dummy.host.org', 'username': '******'} client = ParamikoSSHClient(**conn_params) mock_client = Mock() mock_sftp_client = Mock() mock_transport = Mock() mock_client.get_transport.return_value = mock_transport mock_sftp_client.getcwd.return_value = '/mock/cwd' client.client = mock_client client.sftp_client = mock_sftp_client result = client.put(path='path2.txt', contents='foo bar 2', chmod=466, mode='a') self.assertEqual(result, '/mock/cwd/path2.txt') calls = [call('.')] mock_sftp_client.chdir.assert_has_calls(calls, any_order=False) calls = [ call('path2.txt', mode='a'), call().write('foo bar 2'), call().chmod(466), call().close() ] mock_sftp_client.file.assert_has_calls(calls, any_order=False)
def test_put_absolute_path_windows(self): conn_params = {"hostname": "dummy.host.org", "username": "******"} client = ParamikoSSHClient(**conn_params) mock_client = Mock() mock_sftp_client = Mock() mock_transport = Mock() mock_client.get_transport.return_value = mock_transport mock_sftp_client.getcwd.return_value = "C:\\Administrator" client.client = mock_client client.sftp_client = mock_sftp_client result = client.put(path="C:\\users\\user1\\1.txt", contents="foo bar", chmod=455, mode="w") self.assertEqual(result, "C:\\users\\user1\\1.txt") result = client.put(path="\\users\\user1\\1.txt", contents="foo bar", chmod=455, mode="w") self.assertEqual(result, "\\users\\user1\\1.txt") result = client.put(path="1.txt", contents="foo bar", chmod=455, mode="w") self.assertEqual(result, "C:\\Administrator\\1.txt") mock_client.get_transport.return_value = mock_transport mock_sftp_client.getcwd.return_value = "/C:\\User1" client.client = mock_client client.sftp_client = mock_sftp_client result = client.put(path="1.txt", contents="foo bar", chmod=455, mode="w") self.assertEqual(result, "C:\\User1\\1.txt")
def test_put_absolute_path_windows(self): conn_params = {'hostname': 'dummy.host.org', 'username': '******'} client = ParamikoSSHClient(**conn_params) mock_client = Mock() mock_sftp_client = Mock() mock_transport = Mock() mock_client.get_transport.return_value = mock_transport mock_sftp_client.getcwd.return_value = 'C:\\Administrator' client.client = mock_client client.sftp_client = mock_sftp_client result = client.put(path='C:\\users\\user1\\1.txt', contents='foo bar', chmod=455, mode='w') self.assertEqual(result, 'C:\\users\\user1\\1.txt') result = client.put(path='\\users\\user1\\1.txt', contents='foo bar', chmod=455, mode='w') self.assertEqual(result, '\\users\\user1\\1.txt') result = client.put(path='1.txt', contents='foo bar', chmod=455, mode='w') self.assertEqual(result, 'C:\\Administrator\\1.txt') mock_client.get_transport.return_value = mock_transport mock_sftp_client.getcwd.return_value = '/C:\\User1' client.client = mock_client client.sftp_client = mock_sftp_client result = client.put(path='1.txt', contents='foo bar', chmod=455, mode='w') self.assertEqual(result, 'C:\\User1\\1.txt')