Merge branch 'feature-operations-view'

circle/dashboard/tests/test_mockedviews.py

@@ -112,6 +112,7 @@ class VmOperationViewTestCase(unittest.TestCase):
             self.assertEquals(
                 view.as_view()(request, pk=1234).render().status_code, 200)
 
+
 def FakeRequestFactory(*args, **kwargs):
     ''' FakeRequestFactory, FakeMessages and FakeRequestContext are good for
     mocking out django views; they are MUCH faster than the Django test client.

circle/vm/operations.py

 from __future__ import absolute_import, unicode_literals
 from logging import getLogger
+from re import match
 
 from django.core.exceptions import PermissionDenied
 from django.utils import timezone
@@ -202,7 +203,15 @@ class SaveAsTemplateOperation(InstanceOperation):
         """)
     icon = 'save'
 
-    def _operation(self, activity, name, user, system, timeout=300,
+    def _rename(self, name):
+        m = match("^(.*)( v(\d+))?$", name)
+        if m.group(3) is None:
+            v = 2
+        else:
+            v = int(v) + 1
+        return "%s v%d" % (m.group(1), v)
+
+    def _operation(self, activity, user, system, timeout=300,
                    with_shutdown=True, **kwargs):
         if with_shutdown:
             try:
@@ -219,7 +228,7 @@ class SaveAsTemplateOperation(InstanceOperation):
             'description': self.instance.description,
             'lease': self.instance.lease,  # Can be problem in new VM
             'max_ram_size': self.instance.max_ram_size,
-            'name': name,
+            'name': self._rename(self.instance.name),
             'num_cores': self.instance.num_cores,
             'owner': user,
             'parent': self.instance.template,  # Can be problem

circle/vm/tests/test_operations.py

@@ -43,6 +43,13 @@ class SaveAsTemplateOperationTestCase(TestCase):
     def test_operation_registered(self):
         assert SaveAsTemplateOperation.id in getattr(Instance, op_reg_name)
 
+    def test_rename(self):
+        assert SaveAsTemplateOperation._rename("foo") == "foo v2"
+        assert SaveAsTemplateOperation._rename("foo v2") == "foo v3"
+        assert SaveAsTemplateOperation._rename("foo v") == "foo v v2"
+        assert SaveAsTemplateOperation._rename("foo v9") == "foo v10"
+        assert SaveAsTemplateOperation._rename("foo v111") == "foo v112"
+
 
 class ShutdownOperationTestCase(TestCase):
     def test_operation_registered(self):