Add ftp check unit test
The patch add ftp check unit test Change-Id: Ieb8281db6ac3164e2a6a63b4feb00ca836c067e1
This commit is contained in:
parent
189bf302f6
commit
e380a3f135
|
@ -82,7 +82,11 @@ class FtpStorageTestCase(unittest.TestCase):
|
|||
self.ftp_opt.ftp_remote_ip = '0.0.0.0'
|
||||
self.ftp_opt.ftp_port = 2121
|
||||
|
||||
def test_init_fail_FtpStorage(self):
|
||||
@patch('ftplib.FTP')
|
||||
def test_init_fail_raise_FtpStorage(self, mock_ftp_constructor):
|
||||
mock_ftp = mock_ftp_constructor.return_value
|
||||
seffect = mock.Mock(side_effect=Exception('create ftp failed error'))
|
||||
mock_ftp.raiseError.side_effect = seffect
|
||||
with self.assertRaises(Exception) as cm: # noqa
|
||||
ftp.FtpStorage(
|
||||
storage_path=self.ftp_opt.ftp_storage_path,
|
||||
|
@ -115,6 +119,21 @@ class FtpStorageTestCase(unittest.TestCase):
|
|||
mock_ftp.login.assert_called_with(self.ftp_opt.ftp_remote_username,
|
||||
self.ftp_opt.ftp_remote_pwd)
|
||||
|
||||
@patch('tempfile.mkdtemp')
|
||||
@patch('ftplib.FTP')
|
||||
def test_create_tempdir_FtpStorage(self, mock_ftp_constructor,
|
||||
mock_tempfile_constructor):
|
||||
mock_tempfile = mock_tempfile_constructor.return_value
|
||||
ftp_obj = ftp.FtpStorage(
|
||||
storage_path=self.ftp_opt.ftp_storage_path,
|
||||
remote_pwd=self.ftp_opt.ftp_remote_pwd,
|
||||
remote_username=self.ftp_opt.ftp_remote_username,
|
||||
remote_ip=self.ftp_opt.ftp_remote_ip,
|
||||
port=self.ftp_opt.ftp_port,
|
||||
max_segment_size=self.ftp_opt.ftp_max_segment_size)
|
||||
tmp = ftp_obj._create_tempdir()
|
||||
self.assertEqual(tmp, mock_tempfile)
|
||||
|
||||
|
||||
class FtpsStorageTestCase(unittest.TestCase):
|
||||
|
||||
|
|
Loading…
Reference in New Issue