dashboard: test destroy VmOperationView

circle/dashboard/tests/test_mockedviews.py

@@ -3,10 +3,11 @@ from factory import Factory, Sequence
 from mock import patch, MagicMock
 from mock import patch, MagicMock
 
 
 from django.contrib.auth.models import User
 from django.contrib.auth.models import User
-# from django.core.exceptions import PermissionDenied
+from django.core.exceptions import PermissionDenied
 from django.http import HttpRequest, Http404
 from django.http import HttpRequest, Http404
 
 
-from dashboard.views import InstanceActivityDetail, InstanceActivity
+from ..views import InstanceActivityDetail, InstanceActivity
+from ..views import vm_ops, Instance
 
 
 
 
 class ViewUserTestCase(unittest.TestCase):
 class ViewUserTestCase(unittest.TestCase):
@@ -36,6 +37,33 @@ class ViewUserTestCase(unittest.TestCase):
             self.assertEquals(view(request, pk=1234).render().status_code, 200)
             self.assertEquals(view(request, pk=1234).render().status_code, 200)
 
 
 
 
+class VmOperationViewTestCase(unittest.TestCase):
+    def test_available(self):
+        request = FakeRequestFactory(superuser=True)
+        view = vm_ops['destroy']
+
+        with patch.object(view, 'get_object') as go:
+            inst = MagicMock(spec=Instance)
+            inst._meta.object_name = "Instance"
+            inst.destroy = Instance._ops['destroy'](inst)
+            go.return_value = inst
+            self.assertEquals(
+                view.as_view()(request, pk=1234).render().status_code, 200)
+
+    def test_unpermitted(self):
+        request = FakeRequestFactory()
+        view = vm_ops['destroy']
+
+        with patch.object(view, 'get_object') as go:
+            inst = MagicMock(spec=Instance)
+            inst._meta.object_name = "Instance"
+            inst.destroy = Instance._ops['destroy'](inst)
+            inst.has_level.return_value = False
+            go.return_value = inst
+            with self.assertRaises(PermissionDenied):
+                view.as_view()(request, pk=1234).render()
+
+
 def FakeRequestFactory(*args, **kwargs):
 def FakeRequestFactory(*args, **kwargs):
     ''' FakeRequestFactory, FakeMessages and FakeRequestContext are good for
     ''' FakeRequestFactory, FakeMessages and FakeRequestContext are good for
     mocking out django views; they are MUCH faster than the Django test client.
     mocking out django views; they are MUCH faster than the Django test client.