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')