From 5185860441c73b3db5fa389a8d2f618bf7961807 Mon Sep 17 00:00:00 2001
From: gecong1973 <ge.cong@zte.com.cn>
Date: Sat, 22 Dec 2018 06:53:00 -0800
Subject: [PATCH] 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
---
 .../tests/unit/engines/rsync/test_rsync.py    | 87 +++++++++++++++++++
 1 file changed, 87 insertions(+)

diff --git a/freezer/tests/unit/engines/rsync/test_rsync.py b/freezer/tests/unit/engines/rsync/test_rsync.py
index 2255ad89..d16e5ed9 100644
--- a/freezer/tests/unit/engines/rsync/test_rsync.py
+++ b/freezer/tests/unit/engines/rsync/test_rsync.py
@@ -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)