289 lines
9.4 KiB
Python
289 lines
9.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# to run it use the following command in the 'tests' directory:
|
|
# ~> DJANGO_SETTINGS_MODULE=passerelle.settings pytest -W ignore::django.utils.deprecation.RemovedInDjango20Warning test_atreal_openads.py -vv
|
|
#
|
|
# and with 'coverage':
|
|
# ~> DJANGO_SETTINGS_MODULE=passerelle.settings pytest -W ignore::django.utils.deprecation.RemovedInDjango20Warning test_atreal_openads.py -vv --cov=~/src/passerelle/passerelle/apps/atreal_openads
|
|
|
|
import pytest
|
|
import os
|
|
import base64
|
|
import re
|
|
import datetime
|
|
|
|
from django.core.files import File
|
|
from django.core.files.base import ContentFile
|
|
|
|
from atreal_openads.utils import (
|
|
to_dash_case,
|
|
force_encoded_string_output,
|
|
strip_tags,
|
|
clean_spaces,
|
|
normalize,
|
|
get_file_data,
|
|
get_file_digest,
|
|
get_upload_path,
|
|
get_file_extension,
|
|
trunc_str_values,
|
|
DictDumper
|
|
)
|
|
|
|
from atreal_openads.models import (
|
|
ForwardFile,
|
|
Guichet,
|
|
Collectivite,
|
|
AtrealOpenads
|
|
)
|
|
|
|
|
|
CONNECTOR_NAME = 'atreal-openads'
|
|
CONNECTOR_SLUG = 'atreal'
|
|
COLLECTIVITE = 79
|
|
OPENADS_API_LOGIN = 'publik-passerelle'
|
|
OPENADS_API_PASSWORD = base64.urlsafe_b64encode(os.urandom(20))
|
|
OPENADS_API_URL = 'http://openads.api/'
|
|
|
|
FAKE_COOKIE_CRSF = base64.urlsafe_b64encode(os.urandom(20))
|
|
FAKE_NUMERO_DOSSIER = base64.urlsafe_b64encode(os.urandom(10))
|
|
|
|
TESTS_DIR = os.path.dirname(__file__)
|
|
RESOURCES_DIR = os.path.join(TESTS_DIR, 'resources')
|
|
TEST_FILE_CERFA_DIA = os.path.join(RESOURCES_DIR, 'cerfa_10072-02.pdf')
|
|
TEST_FILE_PLAN_CADASTRAL = os.path.join(RESOURCES_DIR, 'plancadastral.pdf')
|
|
|
|
|
|
@pytest.fixture
|
|
def atreal_openads(db):
|
|
return AtrealOpenads.objects.create(
|
|
slug = CONNECTOR_SLUG,
|
|
default_collectivite_openADS_id = COLLECTIVITE,
|
|
openADS_API_url = OPENADS_API_URL,
|
|
basic_auth_username = OPENADS_API_LOGIN,
|
|
basic_auth_password = OPENADS_API_PASSWORD
|
|
)
|
|
|
|
@pytest.fixture
|
|
def collectivite_1(db, atreal_openads):
|
|
return Collectivite.objects.create(
|
|
name = u'Macollectivité',
|
|
connecteur = atreal_openads,
|
|
openADS_id = '3'
|
|
)
|
|
|
|
@pytest.fixture
|
|
def collectivite_1_guichet(db, atreal_openads, collectivite_1):
|
|
return Guichet.objects.create(
|
|
collectivite = collectivite_1,
|
|
ouverture_jour_h = datetime.time(9, 0),
|
|
fermeture_jour_h = datetime.time(17, 0),
|
|
ouverture_sem_d = 1, # Lundi
|
|
fermeture_sem_d = 5, # Vendredi
|
|
ouverture_sem_h = datetime.time(8, 30),
|
|
fermeture_sem_h = datetime.time(12, 15)
|
|
)
|
|
|
|
|
|
def test_to_dash_case():
|
|
s = 'ACamelCaseName'
|
|
assert to_dash_case(s) == 'a-camel-case-name'
|
|
|
|
assert to_dash_case('') == ''
|
|
|
|
|
|
def test_force_encoded_string_output():
|
|
def a_str_function():
|
|
return str('toto')
|
|
ret = force_encoded_string_output(a_str_function)()
|
|
assert isinstance(ret, str)
|
|
ret = force_encoded_string_output(a_str_function, 'latin1')()
|
|
assert isinstance(ret, str)
|
|
|
|
def an_unicode_function():
|
|
return u'toto'
|
|
ret = force_encoded_string_output(an_unicode_function)()
|
|
assert isinstance(ret, str)
|
|
ret = force_encoded_string_output(an_unicode_function, 'latin1')()
|
|
assert isinstance(ret, str)
|
|
|
|
|
|
def test_strip_tags():
|
|
s = 'aaa b cc '
|
|
assert strip_tags(s) == s
|
|
|
|
ss = s + '<em>dd'
|
|
assert strip_tags(ss) == s + 'dd'
|
|
|
|
ss = s + '<em>dd</em>'
|
|
assert strip_tags(ss) == s + 'dd'
|
|
|
|
ss = s + '<em>dd</em>'
|
|
assert strip_tags(ss) == s + 'dd'
|
|
|
|
ss = s + ' 1 < 3'
|
|
assert strip_tags(ss) == s + ' 1 < 3'
|
|
|
|
|
|
def test_clean_spaces():
|
|
s = 'aaa b cc '
|
|
assert clean_spaces(s) == 'aaa b cc'
|
|
|
|
s = 'a\ta b\nb c\rc d\\n\\r\\td'
|
|
assert clean_spaces(s) == 'a a b b c c d d'
|
|
|
|
|
|
def test_normalize():
|
|
assert normalize(None) == ''
|
|
|
|
s = 'aaa b cc '
|
|
assert normalize(s) == 'aaa b cc'
|
|
|
|
s = 'a\ta b\nb c\rc d\\n\\r\\td'
|
|
assert normalize(s) == 'a a b b c c d d'
|
|
|
|
|
|
def test_get_file_data():
|
|
assert get_file_data(TEST_FILE_CERFA_DIA) == base64.b64encode(open(TEST_FILE_CERFA_DIA).read())
|
|
assert get_file_data(TEST_FILE_CERFA_DIA, b64=False) == open(TEST_FILE_CERFA_DIA).read()
|
|
|
|
|
|
def test_get_file_digest():
|
|
with open(TEST_FILE_CERFA_DIA) as fd:
|
|
assert get_file_digest(fd) == 'cc90a620982760fdee16a5b4fe1b5ac3b4fe868fd02d2f70b27f1e46d283ea51'
|
|
|
|
|
|
def test_get_upload_path():
|
|
ff = ForwardFile(
|
|
numero_demande='45641531',
|
|
numero_dossier=FAKE_NUMERO_DOSSIER,
|
|
type_fichier='CERFA',
|
|
orig_filename=os.path.basename(TEST_FILE_CERFA_DIA),
|
|
content_type='application/pdf',
|
|
file_hash='ffdf456fdsvgb4bgfb6g4f5b',
|
|
upload_file=File(open(TEST_FILE_CERFA_DIA, 'r')),
|
|
upload_status='pending',
|
|
connecteur=None,
|
|
collectivite=None
|
|
)
|
|
regex = r"^to_openADS__%s__%s\.pdf$" % (
|
|
'[0-9]{4}-[0-9]{2}-[0-9]{2}_[0-9]{2}h[0-9]{2}m[0-9]{2}s[0-9]+', 'ffdf')
|
|
assert re.search(regex, get_upload_path(ff))
|
|
|
|
|
|
def test_get_file_extension():
|
|
assert get_file_extension('afile.pdf') == '.pdf'
|
|
assert get_file_extension('afile', 'application/pdf') == '.pdf'
|
|
assert get_file_extension('') == ''
|
|
assert get_file_extension('afile') == ''
|
|
|
|
|
|
def test_trunc_str_values():
|
|
d = {}
|
|
assert trunc_str_values(d, 10) == d
|
|
d = {'a': '123456789'}
|
|
assert trunc_str_values(d, 0) == {'a': u'…'}
|
|
d = {'a': '123456789'}
|
|
assert trunc_str_values(d, 1) == {'a': u'1…'}
|
|
d = {'a': '123456789'}
|
|
assert trunc_str_values(d, 2) == {'a': u'12…'}
|
|
d = {'a': '123456789'}
|
|
assert trunc_str_values(d, 5) == {'a': u'12345…'}
|
|
d = {'a': '123456789'}
|
|
assert trunc_str_values(d, 8) == {'a': u'12345678…'}
|
|
d = {'a': '123456789'}
|
|
assert trunc_str_values(d, 9) == {'a': u'123456789'}
|
|
d = {'a': '123456789'}
|
|
assert trunc_str_values(d, 10) == d
|
|
|
|
d = {'a': '123456789', 'b123456789': '987654321'}
|
|
assert trunc_str_values(d, 5) == {'a': u'12345…', 'b123456789': u'98765…'}
|
|
|
|
d = {'a': '123456789', 'b123456789': '987654321', 'c': {'c1':'ABCDEFGHIJK'}}
|
|
assert trunc_str_values(d, 5) == {'a': u'12345…', 'b123456789': u'98765…', 'c': {'c1': u'ABCDE…'}}
|
|
|
|
d = {'a': '123456789', 'b123456789': '987654321', 'c': {'c1':'ABCDEFGHIJK'}, 'd': ['123456789']}
|
|
assert trunc_str_values(d, 5) == {'a': u'12345…', 'b123456789': u'98765…', 'c': {'c1': u'ABCDE…'}, 'd': [u'12345…']}
|
|
|
|
d = {'a': '123456789', 'b123456789': '987654321', 'c': {'c1':'ABCDEFGHIJK'}, 'd': ['123456789', {'eeeeeeeeee':'132456789'}]}
|
|
assert trunc_str_values(d, 5) == {'a': u'12345…', 'b123456789': u'98765…', 'c': {'c1': u'ABCDE…'}, 'd': [u'12345…', {'eeeeeeeeee': u'13245…'}]}
|
|
|
|
|
|
def test_dict_dumper():
|
|
d = {}
|
|
|
|
dd = DictDumper(d, use_json_dumps=False)
|
|
assert repr(dd) == (u'DictDumper(dic=%r,max_str_len=%r,use_json_dumps=%r)' % (
|
|
d, dd.max_str_len, dd.use_json_dumps)).encode('utf-8')
|
|
assert str(dd) == '{}'
|
|
assert unicode(dd) == u'{}'
|
|
|
|
assert d == dd.dic
|
|
assert unicode(d) == unicode(dd)
|
|
dd = DictDumper(d, 0, use_json_dumps=False)
|
|
assert d == dd.dic
|
|
assert unicode(d) == unicode(dd)
|
|
|
|
d = {'a': '123456789'}
|
|
dd = DictDumper(d, 10, use_json_dumps=False)
|
|
assert d == dd.dic
|
|
assert unicode(d) == unicode(dd)
|
|
dd = DictDumper(d, 5, use_json_dumps=False)
|
|
assert d == dd.dic
|
|
assert unicode(dd) == unicode({'a': u'12345…'})
|
|
dd = DictDumper(d, 5, use_json_dumps=True)
|
|
assert d == dd.dic
|
|
assert unicode(dd) == u'{"a": "12345\\u2026"}'
|
|
|
|
|
|
def test_base_model(atreal_openads, collectivite_1, collectivite_1_guichet):
|
|
ff = ForwardFile(
|
|
numero_demande='45641531',
|
|
numero_dossier=FAKE_NUMERO_DOSSIER,
|
|
type_fichier='CERFA',
|
|
orig_filename=os.path.basename(TEST_FILE_CERFA_DIA),
|
|
content_type='application/pdf',
|
|
file_hash='ffdf456fdsvgb4bgfb6g4f5b',
|
|
upload_file=ContentFile('toto'),
|
|
upload_status='pending',
|
|
connecteur=atreal_openads,
|
|
collectivite=None
|
|
)
|
|
|
|
assert ff.get_verbose_name() == 'Forward File'
|
|
assert ff.get_verbose_name_plural() == 'Forward Files'
|
|
|
|
assert ff.get_class_name() == 'ForwardFile'
|
|
assert ff.get_class_name_plural() == 'ForwardFiles'
|
|
|
|
assert ff.get_class_name_dash_case() == 'forward-file'
|
|
assert ff.get_class_name_plural_dash_case() == 'forward-files'
|
|
|
|
assert ff.get_class_name_title() == 'Forward File'
|
|
assert ff.get_class_name_plural_title() == 'Forward Files'
|
|
|
|
assert ff.get_url_name('list', plural=True) == 'list-forward-files'
|
|
|
|
assert ff.get_absolute_url() == '/manage/atreal-openads/atreal/forward-file/None'
|
|
assert ff.get_edit_url() == '/manage/atreal-openads/atreal/edit-forward-file/None'
|
|
assert ff.get_delete_url() == '/manage/atreal-openads/atreal/delete-forward-file/None'
|
|
assert ff.get_list_url() == '/manage/atreal-openads/atreal/forward-files'
|
|
|
|
assert atreal_openads.get_class_name_plural() == 'AtrealOpenads'
|
|
|
|
assert atreal_openads.get_url_name('view') == 'view-connector'
|
|
|
|
params = atreal_openads.get_url_params(True)
|
|
assert params['connector'] == 'atreal-openads'
|
|
assert params['slug'] == atreal_openads.slug
|
|
|
|
with pytest.raises(Exception) as e:
|
|
atreal_openads.get_list_url()
|
|
assert unicode(e.value) == u"AtrealOpenads:get_list_url() method should not be called"
|
|
|
|
# TODO add more collectivite test cases
|
|
|
|
with pytest.raises(Exception) as e:
|
|
collectivite_1_guichet.get_list_url()
|
|
assert unicode(e.value) == u"Guichet:get_list_url() method should not be called"
|
|
|