Produits Entr'ouvert » Corbo

import pytest

from uuid import uuid4

from django.core.urlresolvers import reverse

from django.utils.http import urlencode

from django.contrib.auth import get_user_model

from django.utils.text import slugify

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

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='By email')

        Broadcast.objects.create(announce=a)

        announces.append(a)

        a = Announce.objects.create(category=category, title='Another one')

        Broadcast.objects.create(announce=a)

        announces.append(a)

    return announces

@pytest.fixture

def user():

    User = get_user_model()

    user = User.objects.create(username='john.doe', first_name=u'John',

                               last_name=u'Doe', email='john.doe@example.net')

    user.set_password('password')

    user.save()

    return user

def test_get_newsletters(app, categories, announces, user):

    resp = app.get(reverse('newsletters'), status=403)

    app.authorization = ('Basic', ('john.doe', 'password'))

    resp = app.get(reverse('newsletters'))

    data = resp.json

    assert data['data']

    for category in data['data']:

        assert 'id' in category

        assert 'text' in category

        assert category['id'] in [slugify(c) for c in CATEGORIES]

        assert category['text'] in CATEGORIES

        assert 'transports' in category

        assert category['transports'] == [{'id': 'mailto', 'text': 'Email'}]

def test_get_subscriptions_by_email(app, categories, announces, user):

    resp = app.get(reverse('subscriptions'), status=403)

    foo = 'foo@example.com'

    resp = app.get(reverse('subscriptions'), {'email': foo}, status=403)

    app.authorization = ('Basic', ('john.doe', 'password'))

    for identifier, name in channel_choices[:1]:

        for category in categories:

            uri = '%s:%s' % (identifier, foo)

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

            resp = app.get(reverse('subscriptions'), {'email': foo})

            assert 'data' in resp.json

            data = resp.json['data']

            for d in data:

                assert d['id'] in [category.slug for category in categories]

                assert d['text'] in [category.name for category in categories]

                for t in d['transports']:

                    assert t['id'] == identifier

def test_update_subscriptions(app, categories, announces, user):

    params = urlencode({'email': 'foo@example.com',

                        'uuid': str(uuid4())})

    app.authorization = ('Basic', ('john.doe', 'password'))

    subscriptions_url = reverse('subscriptions') + '?' + params

    for category in categories:

        transports = []

        for identifier, name in channel_choices[:1]:

            transports.append(identifier)

            category_id = str(category.id)

            subscriptions = [{'id': category_id,

                              'text': category.name,

                              'transports': transports}]

            resp = app.post_json(subscriptions_url, subscriptions)

            if resp.json['data']:

                resp = app.get(subscriptions_url)

                for cat in resp.json['data']:

                    if cat['id'] == category_id:

                        sub_transports = [c['id'] for c in cat['transports']]

                        assert sub_transports == transports

def test_delete_subscriptions(app, categories, announces, user):

    params = urlencode({'email': 'foo@example.com', 'uuid': str(uuid4())})

    subscriptions_url = reverse('subscriptions') + '?' + params

    resp = app.delete(subscriptions_url, status=403)

    app.authorization = ('Basic', ('john.doe', 'password'))

    resp = app.delete(subscriptions_url)

    if resp.json['data']:

        resp = app.get(subscriptions_url)

        assert resp.json['data'] == []

def test_simple_subscription(app, categories, user):

    payload = {'category_id': 'alerts'}

    url = '/api/subscribe/?email=john@example.net'

    # anonymous

    resp = app.post_json(url, payload, status=403)

    assert resp.json['detail'] == 'Authentication credentials were not provided.'

    # authenticated

    app.authorization = ('Basic', ('john.doe', 'password'))

    resp = app.post_json(url, payload, status=200)

    assert resp.json['data'] is True

    # with wrong method

    resp = app.get('/api/subscribe/?email=john@example.net', status=405)

    assert resp.json['detail'] == 'Method "GET" not allowed.'

    # right method on right url

    resp = app.get('/api/subscriptions/?email=john@example.net', status=200)

    data = resp.json['data']

    assert len(data) == 1

    assert data[0]['id'] == 'alerts'

    assert data[0]['text'] == 'Alerts'

    assert len(data[0]['transports']) == 1

    transport = data[0]['transports'][0]

    assert transport['id'] == 'mailto'

    assert transport['text'] == 'mailto'