Test collecting of departments.

talks/api/tests.py

@@ -146,6 +146,16 @@ def setUp(self):
         )
         ti.save()
         future_event.save()
+
+        collectedItemCt = ContentType.objects.get_for_model(models.CollectedDepartment)
+        collectedDep = factories.CollectedDepartmentFactory.create(
+            department=self.department1
+        )
+        collectionItem = factories.CollectionItemFactory.create(
+            collection=collection1,
+            content_type=collectedItemCt,
+            object_id=collectedDep.id
+        )
         self.req_factory = APIRequestFactory()
         self.client = APIClient()
 
@@ -183,6 +193,9 @@ def test_retrieve_collection_happy(self, requests_get):
         self.assertContains(response, "IT talks collection")
         self.assertContains(response, "_links")
         self.assertContains(response, "_embedded")
+        # should return a talk organised by department 1
+        self.assertContains(response, "A future event")
+        self.assertNotContains(response, "A past event")
 
     def test_retrieve_collection_invalid(self):
         response = self.client.get('/api/collections/id/foo')

talks/events/factories.py

@@ -3,7 +3,7 @@
 import factory
 
 from . import models
-from talks.users.models import Collection
+from talks.users.models import Collection, CollectionItem, CollectedDepartment
 
 
 class EventGroupFactory(factory.django.DjangoModelFactory):
@@ -47,3 +47,11 @@ class Meta:
 class TopicItemFactory_noSubFactory(factory.django.DjangoModelFactory):
     class Meta:
         model = models.TopicItem
+
+class CollectedDepartmentFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = CollectedDepartment
+
+class CollectionItemFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = CollectionItem