def test_get_objects_waiting_for_approval_for_user(self):
ct = ExternalContentType.objects.get(pk=1501)
field = Field.objects.get(pk=3501)
middleware.get_user = MagicMock(return_value=User.objects.get(pk=2101))
on_approval_objects = ObjectService.get_objects_waiting_for_approval(ct.pk, field.pk, 1)
self.assertEqual(0, on_approval_objects.count())
ObjectService.register_object(ct.pk, 5001, field.pk)
ObjectService.register_object(ct.pk, 5002, field.pk)
on_approval_objects = ObjectService.get_objects_waiting_for_approval(ct.pk, field.pk, 1)
self.assertEqual(2, on_approval_objects.count())
self.assertEqual(5001, on_approval_objects[0].object_id)
on_approval_objects = ObjectService.get_objects_waiting_for_approval(ct.pk, field.pk, 2)
self.assertEqual(0, on_approval_objects.count())
on_approval_objects = ObjectService.get_objects_waiting_for_approval(ct.pk, field.pk, 3)
self.assertEqual(0, on_approval_objects.count())
on_approval_objects = ObjectService.get_objects_waiting_for_approval(ct.pk, field.pk, 4)
self.assertEqual(0, on_approval_objects.count())
def get_objects_waiting_for_approval(request, content_type_id, field_id, user_id):
return ObjectService.get_objects_waiting_for_approval(content_type_id, field_id, user_id)
