Produits Entr'ouvert » Corbo

import pytest

from uuid import uuid4

import os

import re

import urllib

import mock

from django.core.urlresolvers import reverse

from django.core import mail, signing

from django.utils import timezone

from django.core.files.storage import DefaultStorage

from django.utils.text import slugify

from corbo.models import Category, Announce, Subscription, Broadcast, transform_image_src

pytestmark = pytest.mark.django_db

CATEGORIES = (u'Alerts', u'News')

@pytest.fixture

def categories():

    categories = []

    for category in CATEGORIES:

        c, created = Category.objects.get_or_create(name=category, slug=slugify(category))

        categories.append(c)

    return categories

@pytest.fixture

def announces():

    announces = []

    for category in Category.objects.all():

        a = Announce.objects.create(category=category, title='Announce 1',

                                    publication_time=timezone.now(),

                                    text='<h2>Announce 1</h2>')

        Broadcast.objects.create(announce=a)

        announces.append(a)

        a = Announce.objects.create(category=category, title='Announce 2',

                                    publication_time=timezone.now(),

                                    text='<h2>Announce 2</h2>')

        Broadcast.objects.create(announce=a)

        announces.append(a)

    return announces

def test_emailing_with_no_subscriptions(app, categories, announces, mailoutbox):

    for announce in announces:

        broadcast = Broadcast.objects.get(announce=announce)

        broadcast.send()

        assert not broadcast.result

    assert not len(mailoutbox)

def test_send_email(app, categories, announces, mailoutbox):

    for category in categories:

        uuid = uuid4()

        Subscription.objects.create(category=category,

                                    identifier='%s@example.net' % uuid, uuid=uuid)

    for i, announce in enumerate(announces):

        broadcast = Broadcast.objects.get(announce=announce)

        broadcast.send()

        assert broadcast.result

        assert len(mailoutbox) == i+1

def test_check_inline_css(app, categories, announces, mailoutbox):

    total_sent = 0

    for i, announce in enumerate(announces):

        announce.text = '<style type="text/css">h2 {color: #F00}</style>' + announce.text

        announce.save()

        uuid = uuid4()

        Subscription.objects.create(category=announce.category,

                                    identifier='%s@example.net' % uuid, uuid=uuid)

        broadcast = Broadcast.objects.get(announce=announce)

        broadcast.send()

        assert broadcast.result

        assert len(mailoutbox) == total_sent + broadcast.result

        total_sent += broadcast.result

        assert 'h2 style="color:#F00"' in mailoutbox[i].html

@mock.patch('emails.utils.requests.get')

def test_check_inline_images(mocked_get, app, categories, announces, mailoutbox):

    storage = DefaultStorage()

    media_path = os.path.join(os.path.dirname(__file__), 'media')

    image_name = 'logo.png'

    image_name = storage.save(image_name, file(os.path.join(media_path, image_name)))

    total_sent = 0

    for i, announce in enumerate(announces):

        img_src = "/media/%s" % image_name

        announce.text = announce.text + '<img src="%s" />' % img_src

        announce.save()

        uuid = uuid4()

        Subscription.objects.create(category=announce.category,

                                    identifier='%s@example.net' % uuid, uuid=uuid)

        broadcast = Broadcast.objects.get(announce=announce)

        broadcast.send()

        assert broadcast.result

        assert len(mailoutbox) == total_sent + broadcast.result

        transformed_image_src = transform_image_src(img_src)

        assert transformed_image_src in mailoutbox[i].attachments.keys()

        assert 'cid:%s' % transformed_image_src in mailoutbox[i].html_body

        assert 'cid:%s' % transformed_image_src in mailoutbox[i].text_body

        total_sent += broadcast.result

    storage.delete(image_name)

def test_unsubscription_link(app, categories, announces, mailoutbox):

    unsubscription_link_sentinel = ''

    for category in categories:

        uuid = uuid4()

        scheme = 'mailto:'

        uri = scheme + '%s@example.net' % uuid

        Subscription.objects.create(category=category, identifier=uri, uuid=str(uuid))

        for i, announce in enumerate(announces):

            if announce.category != category:

                continue

            broadcast = Broadcast.objects.get(announce=announce)

            broadcast.send()

            assert broadcast.result

            assert len(mailoutbox) == i+1

            signature = urllib.unquote(re.findall('/unsubscribe/(.*)"', mailoutbox[i].html)[0])

            unsubscription_link = reverse('unsubscribe', kwargs={'unsubscription_token': signature})

            assert signing.loads(signature) == {

                'category': announce.category.pk, 'identifier': uri}

            assert mailoutbox[i].subject == announce.title

            assert unsubscription_link in mailoutbox[i].html

            assert unsubscription_link in mailoutbox[i].text

            assert unsubscription_link_sentinel != unsubscription_link

            # make sure the uri schema is not in the page

            resp = app.get(unsubscription_link)

            assert scheme not in resp.content

        unsubscription_link_sentinel = unsubscription_link