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_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_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]))
