Merge pull request #214 from Syncano/LIB-757

ilu2112 · web-flow · commit a0997f40d4a7 · 2016-07-06T07:21:49.000+02:00
[LIB-757] - Add possibility to restore instance from backup via Python lib
diff --git a/syncano/models/backups.py b/syncano/models/backups.py
@@ -45,5 +45,34 @@ class Meta:
             'list': {
                 'methods': ['post', 'get'],
                 'path': '/backups/full/',
+            },
+        }
+
+    def schedule_restore(self):
+        restore = Restore(backup=self.id).save()
+        return restore
+
+
+class Restore(Model):
+
+    author = fields.ModelField('Admin')
+    status = fields.StringField(read_only=True)
+    status_info = fields.StringField(read_only=True)
+    updated_at = fields.DateTimeField(read_only=True, required=False)
+    created_at = fields.DateTimeField(read_only=True, required=False)
+    links = fields.LinksField()
+    backup = fields.StringField()
+    archive = fields.StringField(read_only=True)
+
+    class Meta:
+        parent = Instance
+        endpoints = {
+            'list': {
+                'methods': ['get', 'post'],
+                'path': '/restores/',
+            },
+            'detail': {
+                'methods': ['get'],
+                'path': '/restores/{id}/',
             }
         }
diff --git a/tests/integration_test_backups.py b/tests/integration_test_backups.py
@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+import time
+
 from syncano.models import Backup
 from tests.integration_test import InstanceMixin, IntegrationTest
 
@@ -27,6 +29,20 @@ def _test_backup_list(self):
         backups = [backup for backup in Backup.please.list()]
         self.assertTrue(len(backups))  # at least one backup here;
 
+    def _test_backup_schedule_restore(self, backup_id):
+        backup = Backup.please.get(id=backup_id)
+
+        # wait for backup to be saved
+        seconds_waited = 0
+        while backup.status in ['scheduled', 'running']:
+            seconds_waited += 1
+            self.assertTrue(seconds_waited < 20, 'Waiting for backup to be saved takes too long.')
+            time.sleep(1)
+            backup.reload()
+
+        restore = backup.schedule_restore()
+        self.assertIn(restore.status, ['success', 'scheduled'])
+
     def _test_backup_delete(self, backup_id):
         backup = Backup.please.get(id=backup_id)
         backup.delete()
@@ -38,4 +54,5 @@ def test_backup(self):
         backup_id = self._test_backup_create()
         self._test_backup_list()
         self._test_backup_detail(backup_id=backup_id)
+        self._test_backup_schedule_restore(backup_id=backup_id)
         self._test_backup_delete(backup_id=backup_id)
