Add ftp add_stream unit test

The patch add ftp add_stream unti test

Change-Id: I94b22c50d6af742182e151d122ecca6e61b8b5dc
This commit is contained in:
gecong1973 2018-12-23 18:53:37 -08:00
parent 286b4c5a44
commit 619b471cef
1 changed files with 35 additions and 0 deletions

View File

@ -253,6 +253,41 @@ class FtpStorageTestCase(unittest.TestCase):
self.assertTrue(mock_create_dirs.called)
self.assertTrue(mock_put_file.called)
@unittest.skipIf(sys.version_info.major == 3,
'Not supported on python v 3.x')
@patch("freezer.storage.ftp.BaseFtpStorage.create_dirs")
@patch("freezer.storage.ftp.BaseFtpStorage.put_file")
@patch("__builtin__.open")
@patch('ftplib.FTP')
def test_add_stream_FtpStorage(self, mock_ftp_constructor,
mock_open,
mock_put_file,
mock_create_dirs):
ftpobj = self.create_ftpstorage_obj()
rich_queue = mock.MagicMock()
rich_queue.get_messages = mock.MagicMock()
rich_queue.get_messages.return_value = ['1']
package_name = 'fakedir/fakename'
stream = ['fakestream']
backup = mock.MagicMock()
backup.copy = mock.MagicMock()
backup.copy.return_value = backup
b_file = mock.MagicMock()
b_file.write = mock.MagicMock()
mock_open = mock.MagicMock()
mock_open.return_value = b_file
ftpobj.add_stream(stream=stream,
package_name=package_name)
self.assertTrue(mock_create_dirs.called)
self.assertTrue(mock_put_file.called)
class FtpsStorageTestCase(unittest.TestCase):