def test_basic(self): instance = BadgeInstance.objects.all()[0] old_clicks = (instance.clickstats_set .aggregate(Sum('clicks'))['clicks__sum']) response = handle_affiliate_link(instance) new_clicks = (instance.clickstats_set .aggregate(Sum('clicks'))['clicks__sum']) eq_(new_clicks, old_clicks + 1) eq_(response['Location'], instance.badge.href)
def link(request, user_id, banner_id, banner_img_id): try: instance, created = (BannerInstance.objects.select_related() .get_or_create(user_id=user_id, badge_id=banner_id, image_id=banner_img_id)) except (IntegrityError): return HttpResponseRedirect(settings.DEFAULT_AFFILIATE_LINK) if created: banner_img = BannerImage.objects.select_related().get(pk=banner_img_id) instance.badge = banner_img.banner instance.save() return handle_affiliate_link(instance)