commit d5b1e1a6213d3a8e83072474735387713f8715a3
Author: gecong1973 <ge.cong@zte.com.cn>
Date:   Sun Oct 11 23:42:09 2020 -0700

    add test_get_backups_exception  unittest for restore
    
    Change-Id: I4cdf72f405786c826fee4635c2b53fb165b205e9

diff --git a/freezer/tests/unit/openstack/test_restore.py b/freezer/tests/unit/openstack/test_restore.py
index 9215027..767c865 100644
--- a/freezer/tests/unit/openstack/test_restore.py
+++ b/freezer/tests/unit/openstack/test_restore.py
@@ -16,11 +16,17 @@
 """
 Freezer restore.py related tests
 """
+from unittest import mock
 
 from freezer.openstack import restore
 from freezer.tests import commons
 
 
+class Image(object):
+    def __init__(self):
+        self.id = 'test'
+
+
 class TestRestore(commons.FreezerBaseTestCase):
     def setUp(self):
         super(TestRestore, self).setUp()
@@ -56,3 +62,26 @@ class TestRestore(commons.FreezerBaseTestCase):
         backup_opt = commons.BackupOpt1()
         restore.RestoreOs(backup_opt.client_manager, backup_opt.container,
                           'local')
+
+    def test_get_backups_exception(self):
+        storage = mock.MagicMock()
+        storage.type = 'ss3'
+        client_manager = mock.MagicMock()
+        restore_os = restore.RestoreOs(client_manager, '/root/test/', storage)
+        try:
+            restore_os._get_backups('/root', 12347)
+        except BaseException as e:
+            self.assertEqual(str(e), "ss3 storage type is not supported at the"
+                                     " moment. Try local, SWIFT, SSH(SFTP),"
+                                     " FTP or FTPS ")
+
+        storage = mock.MagicMock()
+        storage.type = 's3'
+        storage.list_all_objects.return_value = [{'Key': '/12345'},
+                                                 {'Key': '/12346'}]
+        restore_os = restore.RestoreOs(client_manager, '/root/test/', storage)
+        try:
+            restore_os._get_backups('/root', 12347)
+        except BaseException as e:
+            self.assertEqual(str(e), "Cannot find backups for"
+                                     " path: root/test///root")