def test_list_files_remote_dir_not_connected(self):
foo = FtpFileTransfer(None)
self.assertFalse(foo.list_files('/foo'))
self.assertEqual(
foo.get_error_msg(), "Unable to get file list "
"for /foo : 'NoneType' object "
"has no attribute 'list'")
def test_list_files_fail(self):
mockftp = MockFtp()
mockftp.list = Mock(side_effect=IOError('error'))
mockftp.close = Mock(return_value=None)
foo = FtpFileTransfer(None)
foo.set_connection(mockftp)
foo.connect()
filelist = foo.list_files('/foo2')
self.assertTrue(filelist is None)
foo.disconnect()
self.assertEqual(foo.get_error_msg(), 'Unable to get file list '
'for /foo2 : error')
mockftp.list.assert_called_with('/foo2', extra=True)
def test_list_files_success(self):
mockftp = MockFtp()
mockftp.list = Mock(return_value=[{'directory': 'd', 'name': '.'},
{'directory': 'd', 'name': '..'},
{'directory': 'd', 'name': 'foo'},
{'directory': '-', 'name': 'file'}])
mockftp.close = Mock(return_value=None)
foo = FtpFileTransfer(None)
foo.set_connection(mockftp)
foo.connect()
filelist = foo.list_files('/foo')
self.assertTrue(len(filelist) == 1)
self.assertEqual(filelist[0], 'file')
foo.disconnect()
mockftp.list.assert_called_with('/foo', extra=True)
def test_list_files_remote_dir_none(self):
foo = FtpFileTransfer(None)
self.assertFalse(foo.list_files(None))
self.assertEqual(foo.get_error_msg(), 'remote_dir None')
