Add function all() to test QuerySet, model was changed to Node

circle/dashboard/tests/test_mockedviews.py

@@ -28,6 +28,7 @@ from django.http import HttpRequest, Http404, QueryDict
 from django.utils import baseconv
 from django.utils import baseconv
 
 
 from ..models import Profile
 from ..models import Profile
+from vm.models import Node
 from ..views import InstanceActivityDetail, InstanceActivity
 from ..views import InstanceActivityDetail, InstanceActivity
 from ..views import vm_ops, vm_mass_ops, Instance, UnsubscribeFormView
 from ..views import vm_ops, vm_mass_ops, Instance, UnsubscribeFormView
 from ..views import AclUpdateView
 from ..views import AclUpdateView
@@ -35,11 +36,15 @@ from .. import views
 
 
 
 
 class QuerySet(list):
 class QuerySet(list):
-    model = MagicMock()
+    model = Node
 
 
     def get(self, *args, **kwargs):
     def get(self, *args, **kwargs):
         return self.pop()
         return self.pop()
 
 
+    def all(self):
+        return self
+
+
 
 
 class ViewUserTestCase(unittest.TestCase):
 class ViewUserTestCase(unittest.TestCase):