dashboard: test vm migration

circle/dashboard/tests/test_mockedviews.py

@@ -38,6 +38,7 @@ class ViewUserTestCase(unittest.TestCase):
 
 
 
 
 class VmOperationViewTestCase(unittest.TestCase):
 class VmOperationViewTestCase(unittest.TestCase):
+
     def test_available(self):
     def test_available(self):
         request = FakeRequestFactory(superuser=True)
         request = FakeRequestFactory(superuser=True)
         view = vm_ops['destroy']
         view = vm_ops['destroy']
@@ -63,6 +64,53 @@ class VmOperationViewTestCase(unittest.TestCase):
             with self.assertRaises(PermissionDenied):
             with self.assertRaises(PermissionDenied):
                 view.as_view()(request, pk=1234).render()
                 view.as_view()(request, pk=1234).render()
 
 
+    def test_migrate(self):
+        request = FakeRequestFactory(POST={'node': 1})
+        view = vm_ops['migrate']
+
+        with patch.object(view, 'get_object') as go, \
+                patch('dashboard.views.messages') as msg, \
+                patch('dashboard.views.get_object_or_404') as go4:
+            inst = MagicMock(spec=Instance)
+            inst._meta.object_name = "Instance"
+            inst.migrate = Instance._ops['migrate'](inst)
+            inst.migrate.async = MagicMock()
+            inst.has_level.return_value = True
+            go.return_value = inst
+            go4.return_value = MagicMock()
+            assert view.as_view()(request, pk=1234)['location']
+            assert not msg.error.called
+
+    def test_migrate_failed(self):
+        request = FakeRequestFactory(POST={'node': 1})
+        view = vm_ops['migrate']
+
+        with patch.object(view, 'get_object') as go, \
+                patch('dashboard.views.messages') as msg, \
+                patch('dashboard.views.get_object_or_404') as go4:
+            inst = MagicMock(spec=Instance)
+            inst._meta.object_name = "Instance"
+            inst.migrate = Instance._ops['migrate'](inst)
+            inst.migrate.async = MagicMock()
+            inst.migrate.async.side_effect = Exception
+            inst.has_level.return_value = True
+            go.return_value = inst
+            go4.return_value = MagicMock()
+            assert view.as_view()(request, pk=1234)['location']
+            assert msg.error.called
+
+    def test_migrate_template(self):
+        request = FakeRequestFactory()
+        view = vm_ops['migrate']
+
+        with patch.object(view, 'get_object') as go:
+            inst = MagicMock(spec=Instance)
+            inst._meta.object_name = "Instance"
+            inst.migrate = Instance._ops['migrate'](inst)
+            inst.has_level.return_value = True
+            go.return_value = inst
+            self.assertEquals(
+                view.as_view()(request, pk=1234).render().status_code, 200)
 
 
 def FakeRequestFactory(*args, **kwargs):
 def FakeRequestFactory(*args, **kwargs):
     ''' FakeRequestFactory, FakeMessages and FakeRequestContext are good for
     ''' FakeRequestFactory, FakeMessages and FakeRequestContext are good for
@@ -76,12 +124,12 @@ def FakeRequestFactory(*args, **kwargs):
     request = HttpRequest()
     request = HttpRequest()
     request.user = user
     request.user = user
     request.session = kwargs.get('session', {})
     request.session = kwargs.get('session', {})
-    if kwargs.get('POST'):
+    if kwargs.get('POST') is not None:
         request.method = 'POST'
         request.method = 'POST'
         request.POST = kwargs.get('POST')
         request.POST = kwargs.get('POST')
     else:
     else:
         request.method = 'GET'
         request.method = 'GET'
-        request.POST = kwargs.get('GET', {})
+        request.GET = kwargs.get('GET', {})
 
 
     return request
     return request