def test_used_media_get_usage(self): media = models.Media.objects.get(id=1) page = EventPage.objects.get(id=3) event_page_related_link = EventPageRelatedMedia() event_page_related_link.page = page event_page_related_link.link_media = media event_page_related_link.save() self.assertTrue(issubclass(Page, type(media.get_usage()[0])))
def test_used_media_usage_count(self): media = models.Media.objects.get(id=1) page = EventPage.objects.get(id=3) event_page_related_link = EventPageRelatedMedia() event_page_related_link.page = page event_page_related_link.link_media = media event_page_related_link.save() self.assertEqual(media.get_usage().count(), 1)
def test_usage_page(self): media = models.Media.objects.get(id=1) page = EventPage.objects.get(id=3) event_page_related_link = EventPageRelatedMedia() event_page_related_link.page = page event_page_related_link.link_media = media event_page_related_link.save() response = self.client.get(reverse('wagtailmedia:media_usage', args=(1,))) self.assertContains(response, 'Christmas')
def test_usage_count_appears(self): media = models.Media.objects.get(id=1) page = EventPage.objects.get(id=3) event_page_related_link = EventPageRelatedMedia() event_page_related_link.page = page event_page_related_link.link_media = media event_page_related_link.save() response = self.client.get(reverse('wagtailmedia:edit', args=(1,))) self.assertContains(response, 'Used 1 time')
def test_usage_count_appears(self): media = models.Media.objects.get(id=1) page = EventPage.objects.get(id=3) event_page_related_link = EventPageRelatedMedia() event_page_related_link.page = page event_page_related_link.link_media = media event_page_related_link.save() response = self.client.get(reverse('wagtailmedia:edit', args=(1, ))) self.assertContains(response, 'Used 1 time')
def test_usage_page(self): media = models.Media.objects.get(id=1) page = EventPage.objects.get(id=3) event_page_related_link = EventPageRelatedMedia() event_page_related_link.page = page event_page_related_link.link_media = media event_page_related_link.save() response = self.client.get( reverse('wagtailmedia:media_usage', args=(1, ))) self.assertContains(response, 'Christmas')