tests: add tests on mail views (#40816)

This commit is contained in:
Nicolas Roche 2020-03-19 18:42:30 +01:00
parent 3380e678b6
commit 126e137998
1 changed files with 191 additions and 0 deletions

191
tests/test_mail_manager.py Normal file
View File

@ -0,0 +1,191 @@
# 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}