Example #1
0
    def test_mixin_get_queryset(self):
        self.moxx.StubOutWithMock(UserModelManager, 'filter_user_objects')
        UserModelManager.filter_user_objects(self.user).AndReturn('object')
        self.moxx.StubOutWithMock(CreateView, 'get_queryset')
        CreateView.get_queryset().AndReturn('pass')

        self.assertEqual(self.view.queryset, None)
        self.moxx.ReplayAll()
        self.view.get_queryset()
        self.moxx.VerifyAll()
        self.assertEqual(self.view.queryset, 'object')