def test_links_filters(self): category = CategoryFactory.create() user = UserFactory.create() text_banner_variation = TextBannerVariationFactory.create(banner__category=category) upgrade_banner_variation = FirefoxUpgradeBannerVariationFactory.create( banner__category=category) text_link = LinkFactory.create(banner_variation=text_banner_variation, user=user) LinkFactory.create(banner_variation=upgrade_banner_variation) eq_(list(category.links(user=user)), [text_link])
def test_generate_banner_code(self): # This is less a test of initial correctness and more to alert # us of unexpected changes to banner code generation. variation = FirefoxUpgradeBannerVariationFactory.create( banner__destination='https://www.mozilla.org', image='uploads/firefox_upgrade_banners/test.png', upgrade_image='uploads/firefox_upgrade_banners/test_upgrade.png') banner = variation.banner with self.settings(MEDIA_URL='/media/', SITE_URL='https://example.com'): self.assertHTMLEqual(banner.generate_banner_code(variation), """ <a href="{{href}}"> <img src="https://example.com/media/uploads/upgrade/{pk}"> </a> """.format(pk=variation.pk))
def test_links(self): """ links should return a queryset of links related to any type of banner variation within the category. """ category = CategoryFactory.create() # Test against several variations, and multiple variations. image_banner_variation1 = ImageBannerVariationFactory.create(banner__category=category) image_banner_variation2 = ImageBannerVariationFactory.create(banner__category=category) text_banner_variation = TextBannerVariationFactory.create(banner__category=category) upgrade_banner_variation = FirefoxUpgradeBannerVariationFactory.create(banner__category=category) # Create links from the variations. image_link1 = LinkFactory.create(banner_variation=image_banner_variation1) image_link2 = LinkFactory.create(banner_variation=image_banner_variation1) image_link3 = LinkFactory.create(banner_variation=image_banner_variation2) text_link = LinkFactory.create(banner_variation=text_banner_variation) upgrade_link = LinkFactory.create(banner_variation=upgrade_banner_variation) eq_(set([image_link1, image_link2, image_link3, text_link, upgrade_link]), set(category.links))
def test_links(self): """ links should return a queryset of links related to any type of banner variation within the category. """ category = CategoryFactory.create() # Test against several variations, and multiple variations. image_banner_variation1 = ImageBannerVariationFactory.create(banner__category=category) image_banner_variation2 = ImageBannerVariationFactory.create(banner__category=category) text_banner_variation = TextBannerVariationFactory.create(banner__category=category) upgrade_banner_variation = FirefoxUpgradeBannerVariationFactory.create( banner__category=category) # Create links from the variations. image_link1 = LinkFactory.create(banner_variation=image_banner_variation1) image_link2 = LinkFactory.create(banner_variation=image_banner_variation1) image_link3 = LinkFactory.create(banner_variation=image_banner_variation2) text_link = LinkFactory.create(banner_variation=text_banner_variation) upgrade_link = LinkFactory.create(banner_variation=upgrade_banner_variation) eq_(set(category.links()), set([image_link1, image_link2, image_link3, text_link, upgrade_link]))