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

from corbo.models import channel_choices

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.delivery_count

    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.delivery_count

        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.delivery_count

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

        total_sent += broadcast.delivery_count

        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)

        mocked_get.return_value = mock.Mock(status_code=200,

                                            headers={'content-type': 'image/png'},

                                            content=storage.open(image_name).read())

        broadcast.send()

        assert broadcast.delivery_count

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

        attachments = [a['filename'] for a in mailoutbox[0].attachments.as_dict()]

        assert image_name in attachments

        assert 'cid:%s' % image_name in mail.outbox[0].html_body

        assert 'cid:%s' % image_name in mail.outbox[0].text_body

        total_sent += broadcast.delivery_count

    storage.delete(image_name)

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

    unsubscription_link_sentinel = ''

    subscriptions_number = 3

    scheme = 'mailto:'

    for category in categories:

        for i in xrange(subscriptions_number):

            uuid = uuid4()

            uri = scheme + '%s@example.com' % 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.delivery_count

            assert len(mail.outbox) == (i+1)*subscriptions_number

            assert mail.outbox[i*subscriptions_number].subject == announce.title

            for counter, destination in enumerate(category.subscription_set.all()):

                index = i*subscriptions_number+counter

                signature = urllib.unquote(re.findall('/unsubscribe/(.*)"', mail.outbox[index].html)[0])

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

                assert mail.outbox[index]._headers['List-Unsubscribe'] == '<http://localhost%s>' % unsubscription_link

                assert unsubscription_link in mail.outbox[index].html

                assert unsubscription_link in mail.outbox[index].text

                assert unsubscription_link_sentinel != unsubscription_link

                assert signing.loads(signature) == {

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

                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