vm: fix tests

circle/dashboard/fixtures/test-vm-fixture.json

@@ -1240,6 +1240,24 @@
   }
   }
 },
 },
 {
 {
+  "pk": 1367,
+  "model": "auth.permission",
+  "fields": {
+    "codename": "create_vm",
+    "name": "Can create a new VM.",
+    "content_type": 28
+  }
+},
+{
+  "pk": 1368,
+  "model": "auth.permission",
+  "fields": {
+    "codename": "access_console",
+    "name": "Can access the graphical console of a VM.",
+    "content_type": 28
+  }
+},
+{
   "pk": 1, 
   "pk": 1, 
   "model": "auth.group", 
   "model": "auth.group", 
   "fields": {
   "fields": {

circle/dashboard/tests/test_mockedviews.py

@@ -159,7 +159,7 @@ class VmOperationViewTestCase(unittest.TestCase):
             assert not msg.error.called
             assert not msg.error.called
 
 
     def test_migrate_failed(self):
     def test_migrate_failed(self):
-        request = FakeRequestFactory(POST={'node': 1})
+        request = FakeRequestFactory(POST={'node': 1}, superuser=True)
         view = vm_ops['migrate']
         view = vm_ops['migrate']
 
 
         with patch.object(view, 'get_object') as go, \
         with patch.object(view, 'get_object') as go, \
@@ -177,7 +177,7 @@ class VmOperationViewTestCase(unittest.TestCase):
             assert msg.error.called
             assert msg.error.called
 
 
     def test_migrate_template(self):
     def test_migrate_template(self):
-        request = FakeRequestFactory()
+        request = FakeRequestFactory(superuser=True)
         view = vm_ops['migrate']
         view = vm_ops['migrate']
 
 
         with patch.object(view, 'get_object') as go:
         with patch.object(view, 'get_object') as go:
@@ -190,7 +190,7 @@ class VmOperationViewTestCase(unittest.TestCase):
                 view.as_view()(request, pk=1234).render().status_code, 200)
                 view.as_view()(request, pk=1234).render().status_code, 200)
 
 
     def test_save_as_wo_name(self):
     def test_save_as_wo_name(self):
-        request = FakeRequestFactory(POST={})
+        request = FakeRequestFactory(POST={}, has_perms_mock=True)
         view = vm_ops['save_as_template']
         view = vm_ops['save_as_template']
 
 
         with patch.object(view, 'get_object') as go, \
         with patch.object(view, 'get_object') as go, \
@@ -224,7 +224,7 @@ class VmOperationViewTestCase(unittest.TestCase):
             assert not msg.error.called
             assert not msg.error.called
 
 
     def test_save_as_template(self):
     def test_save_as_template(self):
-        request = FakeRequestFactory()
+        request = FakeRequestFactory(has_perms_mock=True)
         view = vm_ops['save_as_template']
         view = vm_ops['save_as_template']
 
 
         with patch.object(view, 'get_object') as go:
         with patch.object(view, 'get_object') as go:
@@ -246,6 +246,8 @@ def FakeRequestFactory(*args, **kwargs):
     user = UserFactory()
     user = UserFactory()
     user.is_authenticated = lambda: kwargs.get('authenticated', True)
     user.is_authenticated = lambda: kwargs.get('authenticated', True)
     user.is_superuser = kwargs.get('superuser', False)
     user.is_superuser = kwargs.get('superuser', False)
+    if kwargs.get('has_perms_mock', False):
+        user.has_perms = MagicMock(return_value=True)
 
 
     request = HttpRequest()
     request = HttpRequest()
     request.user = user
     request.user = user

circle/dashboard/tests/test_views.py

@@ -63,6 +63,8 @@ class VmDetailTest(LoginMixin, TestCase):
         self.g1.user_set.add(self.u1)
         self.g1.user_set.add(self.u1)
         self.g1.user_set.add(self.u2)
         self.g1.user_set.add(self.u2)
         self.g1.save()
         self.g1.save()
+        self.u1.user_permissions.add(Permission.objects.get(
+            codename='create_vm'))
         settings["default_vlangroup"] = 'public'
         settings["default_vlangroup"] = 'public'
         VlanGroup.objects.create(name='public')
         VlanGroup.objects.create(name='public')
 
 
@@ -1544,6 +1546,8 @@ class VmDetailVncTest(LoginMixin, TestCase):
         inst.node = Node.objects.all()[0]
         inst.node = Node.objects.all()[0]
         inst.save()
         inst.save()
         inst.set_level(self.u1, 'operator')
         inst.set_level(self.u1, 'operator')
+        self.u1.user_permissions.add(Permission.objects.get(
+            codename='access_console'))
         response = c.get('/dashboard/vm/1/vnctoken/')
         response = c.get('/dashboard/vm/1/vnctoken/')
         self.assertEqual(response.status_code, 200)
         self.assertEqual(response.status_code, 200)
 
 
@@ -1554,6 +1558,8 @@ class VmDetailVncTest(LoginMixin, TestCase):
         inst.node = Node.objects.all()[0]
         inst.node = Node.objects.all()[0]
         inst.save()
         inst.save()
         inst.set_level(self.u1, 'user')
         inst.set_level(self.u1, 'user')
+        self.u1.user_permissions.add(Permission.objects.get(
+            codename='access_console'))
         response = c.get('/dashboard/vm/1/vnctoken/')
         response = c.get('/dashboard/vm/1/vnctoken/')
         self.assertEqual(response.status_code, 403)
         self.assertEqual(response.status_code, 403)