Add rsync process_backup_data and process_restore_data unit test

The patch add rsync process_backup_data and process_restore_data
unit test

Change-Id: Ied53d1ae9c523771aa933f644ef2290c05a80187
This commit is contained in:
gecong1973 2018-12-22 06:53:00 -08:00
parent 0d5327404c
commit 5185860441
1 changed files with 87 additions and 0 deletions

View File

@ -1495,3 +1495,90 @@ class TestRsyncEngine(unittest.TestCase):
flushed=flushed,
current_backup_level=current_backup_level)
self.assertEqual(ret, data_chunk)
def test_process_backup_data(self):
data = 'fakedata'
do_compress = True
fake_rsync = self.mock_rsync
fake_rsync.compressor = self.compressor
fake_rsync.cipher = self.cipher
fake_rsync.encrypt_pass_file = True
fake_rsync.compressor.compress = mock.MagicMock()
fake_rsync.compressor.compress.return_value = data
fake_rsync.cipher.encrypt = mock.MagicMock()
fake_rsync.cipher.encrypt.return_value = data
ret = fake_rsync.process_backup_data(data=data,
do_compress=do_compress)
self.assertEqual(ret, data)
def test_process_backup_data_compress(self):
data = 'fakedata'
do_compress = True
fake_rsync = self.mock_rsync
fake_rsync.compressor = self.compressor
fake_rsync.cipher = self.cipher
fake_rsync.encrypt_pass_file = False
fake_rsync.compressor.compress = mock.MagicMock()
fake_rsync.compressor.compress.return_value = data
ret = fake_rsync.process_backup_data(data=data,
do_compress=do_compress)
self.assertEqual(ret, data)
def test_process_backup_data_encrypt(self):
data = 'fakedata'
do_compress = False
fake_rsync = self.mock_rsync
fake_rsync.compressor = self.compressor
fake_rsync.cipher = self.cipher
fake_rsync.encrypt_pass_file = True
fake_rsync.cipher.encrypt = mock.MagicMock()
fake_rsync.cipher.encrypt.return_value = data
ret = fake_rsync.process_backup_data(data=data,
do_compress=do_compress)
self.assertEqual(ret, data)
def test_process_restore_data(self):
data = 'fakedata'
fake_rsync = self.mock_rsync
fake_rsync.compressor = self.compressor
fake_rsync.cipher = self.cipher
fake_rsync.encrypt_pass_file = False
fake_rsync.compressor.decompress = mock.MagicMock()
fake_rsync.compressor.decompress.return_value = data
ret = fake_rsync.process_restore_data(data=data)
self.assertEqual(ret, data)
def test_process_restore_data_encrypt(self):
data = 'fakedata'
fake_rsync = self.mock_rsync
fake_rsync.compressor = self.compressor
fake_rsync.cipher = self.cipher
fake_rsync.encrypt_pass_file = True
fake_rsync.compressor.decompress = mock.MagicMock()
fake_rsync.compressor.decompress.return_value = data
fake_rsync.cipher.decrypt = mock.MagicMock()
fake_rsync.cipher.decrypt.return_value = data
ret = fake_rsync.process_restore_data(data=data)
self.assertEqual(ret, data)