def test_rename(self): from_path = '/path/from' to_path = '/path/to' with fh.FTPHook() as ftp_hook: ftp_hook.rename(from_path, to_path) self.conn_mock.rename.assert_called_once_with(from_path, to_path) self.conn_mock.quit.assert_called_once_with()
def test_mod_time_micro(self): self.conn_mock.sendcmd.return_value = '213 20170428010138.003' path = '/path/file' with fh.FTPHook() as ftp_hook: ftp_hook.get_mod_time(path) self.conn_mock.sendcmd.assert_called_once_with('MDTM ' + path)
def test_get_size(self): self.conn_mock.size.return_value = 1942 path = '/path/file' with fh.FTPHook() as ftp_hook: ftp_hook.get_size(path) self.conn_mock.size.assert_called_once_with(path)
def test_delete_file(self): with fh.FTPHook() as ftp_hook: ftp_hook.delete_file(self.path) self.conn_mock.delete.assert_called_once_with(self.path)
def test_delete_directory(self): with fh.FTPHook() as ftp_hook: ftp_hook.delete_directory(self.path) self.conn_mock.rmd.assert_called_once_with(self.path)
def test_list_directory(self): with fh.FTPHook() as ftp_hook: ftp_hook.list_directory(self.path) self.conn_mock.cwd.assert_called_once_with(self.path) self.conn_mock.nlst.assert_called_once_with()
def test_close_conn(self): ftp_hook = fh.FTPHook() ftp_hook.get_conn() ftp_hook.close_conn() self.conn_mock.quit.assert_called_once_with()
def test_retrieve_file_with_callback(self): func = mock.Mock() _buffer = io.StringIO('buffer') with fh.FTPHook() as ftp_hook: ftp_hook.retrieve_file(self.path, _buffer, callback=func) self.conn_mock.retrbinary.assert_called_once_with('RETR path', func)
def test_retrieve_file(self): _buffer = io.StringIO('buffer') with fh.FTPHook() as ftp_hook: ftp_hook.retrieve_file(self.path, _buffer) self.conn_mock.retrbinary.assert_called_once_with( 'RETR path', _buffer.write)