diff --git a/circle/dashboard/tests/test_mockedviews.py b/circle/dashboard/tests/test_mockedviews.py index d83dbc4..4f3c3ac 100644 --- a/circle/dashboard/tests/test_mockedviews.py +++ b/circle/dashboard/tests/test_mockedviews.py @@ -28,6 +28,7 @@ from django.http import HttpRequest, Http404, QueryDict from django.utils import baseconv from ..models import Profile +from vm.models import Node from ..views import InstanceActivityDetail, InstanceActivity from ..views import vm_ops, vm_mass_ops, Instance, UnsubscribeFormView from ..views import AclUpdateView @@ -35,11 +36,15 @@ from .. import views class QuerySet(list): - model = MagicMock() + model = Node def get(self, *args, **kwargs): return self.pop() + def all(self): + return self + + class ViewUserTestCase(unittest.TestCase):