Produits Entr'ouvert » Welco

# welco - multichannel request processing

# Copyright (C) 2020  Entr'ouvert

#

# This program is free software: you can redistribute it and/or modify it

# under the terms of the GNU Affero General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import httmock

import json

import requests

from webtest import Upload

from django.contrib.contenttypes.models import ContentType

from django.core.files.base import ContentFile

from django.utils.encoding import force_text

from welco.sources.mail.models import Mail

def test_viewer_view(app):

    resp = app.get('/mail/viewer/', status=302)

    assert resp.location == '?file='

    resp = resp.follow()

    assert resp.html.find('title').text == 'PDF.js viewer'

    resp = app.get('/mail/viewer/', {'file': 'tests/test.pdf'}, status=200)

    assert resp.html.find('title').text == 'PDF.js viewer'

def test_get_feeder_view(app, user):

    resp = app.get('/mail/feeder/', status=302)

    assert resp.location.startswith('/login/?next=')

    app.set_user(user.username)

    resp = app.get('/mail/feeder/', status=200)

    assert resp.html.find('h2').text == 'Mail Feeder'

def test_post_feeder_view(app, user):

    app.set_user(user.username)

    resp = app.post(

        '/mail/feeder/',

        params={'mail': Upload('filename.txt', b'contents')},

        status=302)

    assert resp.location == '/mail/feeder/'

    resp = resp.follow()

    assert resp.html.find('li', {'class': 'info'}).text == '1 files uploaded successfully.'

def test_qualification_save_view(settings, app, db):

    settings.KNOWN_SERVICES = {

        'wcs': {

            'demarches': {

                'url': 'http://wcs.example.net/',

            }

        }

    }

    mail = Mail.objects.create(

        content=ContentFile('foo', name='bar.txt'),

        subject='spam')

    assert not mail.contact_id

    source_type = ContentType.objects.get_for_model(Mail).pk

    resp = app.post(

        '/ajax/qualification-mail-save',

        params={'source_type': source_type, 'source_pk': mail.pk, 'subject': 'eggs'},

        status=302)

    assert resp.location == '/ajax/qualification?source_type=%s&source_pk=%s' % (

        source_type, mail.pk)

    @httmock.urlmatch(netloc='wcs.example.net', path='/api/formdefs/', method='GET')

    def response_get(url, request):

        headers = {'content-type': 'application/json'}

        content = {

            "err": 0,

            "data": [{

                "title": "Foo",

                "slug": "foo",

            }]}

        return httmock.response(200, content, headers)

    with httmock.HTTMock(response_get):

        resp = resp.follow()

    assert resp.html.find('option', {'value': 'demarches:foo'}).text == 'Foo'

    assert Mail.objects.get(id=mail.pk).subject == 'eggs'

def test_edit_note_view(app, user):

    resp = app.get('/ajax/mail/edit-note/', status=302)

    assert resp.location.startswith('/login/?next=')

    app.set_user(user.username)

    mail = Mail.objects.create(

        content=ContentFile('foo', name='bar.txt'),

        note='spam')

    resp = app.get('/ajax/mail/edit-note/', params={'mail': mail.pk}, status=200)

    assert resp.html.find('h2').text == 'Note'

    assert resp.html.find('textarea', {'name': 'note'}).text == 'spam'

    resp.form['note'] = 'eggs'

    resp = resp.form.submit()

    assert resp.content_type == 'text/html'

    assert resp.text == '{"result": "ok"}'

    assert Mail.objects.get(id=mail.pk).note == 'eggs'

def test_note_view(app, user):

    mail = Mail.objects.create(content=ContentFile('foo', name='bar.txt'))

    resp = app.get('/ajax/mail/note/%s' % mail.pk, status=302)

    assert resp.location.startswith('/login/?next=')

    app.set_user(user.username)

    resp = app.get('/ajax/mail/note/%s' % mail.pk, status=200)

    assert resp.text == '+'

    assert not Mail.objects.get(id=mail.pk).note  # mail object is unchanged

def test_reject_view(settings, app, user):

    settings.MAARCH_FEED= {

        'URL': 'http://maarch.example.net',

        'ENABLE': True,

        'USERNAME': 'xxx',

        'PASSWORD': 'yyy',

        'STATUS_REFUSED': 'FOO',

    }

    resp = app.post('/ajax/mail/reject', status=302)

    assert resp.location.startswith('/login/?next=')

    app.set_user(user.username)

    mail = Mail.objects.create(

        content=ContentFile('foo', name='bar.txt'),

        external_id='maarch-42')

    @httmock.urlmatch(netloc='maarch.example.net', path='/rest/res/resource/status', method='PUT')

    def response_ok(url, request):

        assert json.loads(force_text(request.body)) == {'status': 'FOO', 'resId': ['42']}

        headers = {'content-type': 'application/json'}

        content = {"maarch_say": "ok"}

        return httmock.response(200, content, headers)

    with httmock.HTTMock(response_ok):

        resp = app.post('/ajax/mail/reject', params={'source_pk': mail.pk}, status=200)

    assert Mail.objects.count() == 0

    # errors

    mail = Mail.objects.create(

        content=ContentFile('foo', name='bar.txt'),

        external_id='maarch-42')

    @httmock.urlmatch(netloc='maarch.example.net', path='/rest/res/resource/status', method='PUT')

    def response_error1(url, request):

        raise requests.RequestException

    with httmock.HTTMock(response_error1):

        resp = app.post('/ajax/mail/reject', params={'source_pk': mail.pk})

    assert Mail.objects.get(id=mail.pk)

    @httmock.urlmatch(netloc='maarch.example.net', path='/rest/res/resource/status', method='PUT')

    def response_error2(url, request):

        return httmock.response(500)

    with httmock.HTTMock(response_error2):

        resp = app.post('/ajax/mail/reject', params={'source_pk': mail.pk})

    assert Mail.objects.get(id=mail.pk)

    @httmock.urlmatch(netloc='maarch.example.net', path='/rest/res/resource/status', method='PUT')

    def response_error3(url, request):

        return httmock.response(200, 'not a json')

    with httmock.HTTMock(response_error3):

        resp = app.post('/ajax/mail/reject', params={'source_pk': mail.pk})

    assert Mail.objects.get(id=mail.pk)

def test_mail_count_view(app, user):

    resp = app.get('/ajax/count/mail/', status=302)

    assert resp.location.startswith('/login/?next=')

    Mail.objects.create(content=ContentFile('foo', name='bar.txt'), status='done-42')

    Mail.objects.create(content=ContentFile('foo', name='bar.txt'), status='43')

    app.set_user(user.username)

    resp = app.get('/ajax/count/mail/', status=200)

    assert resp.content_type == 'application/json'

    assert resp.json == {'count': 1}