pbp_com/apps/adzone/tests.py

from django.test import TestCase
from django.test.client import Client
from django.core.handlers.wsgi import WSGIRequest
from django.core.urlresolvers import reverse

from adzone.models import *
from django.contrib.auth.models import User


def datenow():
    from datetime import datetime

    return datetime.now()


class RequestFactory(Client):
    """
    Class that lets you create mock Request objects for use in testing.
    
    Usage:
    
    rf = RequestFactory()
    get_request = rf.get('/hello/')
    post_request = rf.post('/submit/', {'foo': 'bar'})
    
    This class re-uses the django.test.client.Client interface, docs here:
    http://www.djangoproject.com/documentation/testing/#the-test-client
    
    Once you have a request object you can pass it to any view function, 
    just as if that view had been hooked up using a URLconf.
    
    """

    def request(self, **request):
        """
        Similar to parent class, but returns the request object as soon as it
        has created it.
        """
        environ = {
            "HTTP_COOKIE": self.cookies,
            "PATH_INFO": "/",
            "QUERY_STRING": "",
            "REQUEST_METHOD": "GET",
            "SCRIPT_NAME": "",
            "SERVER_NAME": "testserver",
            "SERVER_PORT": 80,
            "SERVER_PROTOCOL": "HTTP/1.1",
            "REMOTE_ADDR": "127.0.0.1",
        }
        environ.update(self.defaults)
        environ.update(request)
        return WSGIRequest(environ)


class AdvertisingTestCase(TestCase):
    def setUp(self):
        self.request = RequestFactory().request()

        self.user = User.objects.create_user("test", "test@example.com", "testpass")

        self.advertiser = Advertiser.objects.create(
            company_name="teh_node Web Development",
            website="http://andre.smoenux.webfactional.com/",
            user=self.user,
        )

        # Categories setup
        self.category = AdCategory.objects.create(
            title="Internet Services",
            slug="internet-services",
            description="All internet based services",
        )

        self.category2 = AdCategory.objects.create(
            title="Category Two", slug="categorytwo", description="A Second Category"
        )

        # Zones setup
        self.adzone = AdZone.objects.create(
            title="Sidebar", slug="sidebar", description="Side Bar Ads"
        )

        self.adzone2 = AdZone.objects.create(
            title="Content Banner", slug="contentbanner", description="Content Adverts"
        )

        # Ad setup
        self.ad = TextAd.objects.create(
            title="First Ad",
            content="For all your web design and development needs, at competitive rates.",
            url="http://www.teh-node.co.za/",
            enabled=True,
            advertiser=self.advertiser,
            category=self.category,
            zone=self.adzone,
        )

        self.ad2 = TextAd.objects.create(
            title="Second Ad",
            content="A second advert.",
            url="http://www.teh-node.co.za/",
            enabled=True,
            advertiser=self.advertiser,
            category=self.category2,
            zone=self.adzone2,
        )

        self.ad3 = TextAd.objects.create(
            title="A Third Ad",
            content="A third advert.",
            url="http://www.teh-node.co.za/",
            enabled=True,
            advertiser=self.advertiser,
            category=self.category2,
            zone=self.adzone2,
        )

        # AdImpression Setup
        self.impression1 = AdImpression.objects.create(
            impression_date=datenow(), source_ip="127.0.0.2", ad=self.ad
        )
        self.impression2 = AdImpression.objects.create(
            impression_date=datenow(), source_ip="127.0.0.3", ad=self.ad2
        )

        # Clicks Setup
        self.adclick1 = AdClick.objects.create(
            click_date=datenow(), source_ip="127.0.0.1", ad=self.ad
        )


class AdvertiserTestCase(AdvertisingTestCase):
    def testAdvertiser(self):
        self.assertEquals(
            self.advertiser.get_website_url(), "http://andre.smoenux.webfactional.com/"
        )


class CategoryTestCase(AdvertisingTestCase):
    def testAdCategory(self):
        self.assertEquals(self.category.__unicode__(), "Internet Services")


class ZoneTestCase(AdvertisingTestCase):
    def testAdZone(self):
        self.assertEquals(self.adzone.__unicode__(), "Sidebar")

    def testAdinZone(self):
        ads = TextAd.objects.filter(zone__slug="sidebar")
        self.assertEquals(len(ads), 1)


class AdvertTestCase(AdvertisingTestCase):
    def testAd(self):
        self.assertEquals(reverse("adzone_ad_view", args=["1"])[-8:], "/view/1/")
        adimpressions = AdImpression.objects.filter(ad=self.ad)
        self.assertEquals(len(adimpressions), 1)
        self.assertEquals(adimpressions[0].source_ip, "127.0.0.2")

    def testAdAdvertiser(self):
        self.assertEquals(self.ad.advertiser.__unicode__(), "teh_node Web Development")
        self.assertEquals(self.ad.advertiser.company_name, "teh_node Web Development")

    def testAddsInCategory(self):
        ads = TextAd.objects.filter(category__slug="internet-services")
        self.assertEquals(len(ads), 1)
        self.assertEquals(ads[0].title, "First Ad")

    def testRandomAd(self):
        ad = AdBase.objects.get_random_ad(
            ad_category="internet-services", ad_zone="sidebar"
        )
        self.assertEquals(ad.title, "First Ad")


class ImpressionTestCase(AdvertisingTestCase):
    def testImpression(self):
        impressions = AdImpression.objects.all()
        self.assertEquals(len(impressions), 2)


class ClickTestCase(AdvertisingTestCase):
    def testClicks(self):
        clicks = AdClick.objects.all()
        self.assertEquals(len(clicks), 1)

    def testClickOnAds(self):
        c = Client(REMOTE_ADDR="127.0.0.1")
        response = c.get(reverse("adzone_ad_view", args=["1"]))
        self.assertEquals(len(AdClick.objects.all()), 2)
        click = AdClick.objects.all()[1]
        self.assertEquals(click.source_ip, "127.0.0.1")

    def testInvalidAdURL(self):
        c = Client(REMOTE_ADDR="127.0.0.1")
        response = c.get("/te/10")
        self.assertEquals(len(AdClick.objects.all()), 1)