Initial import of the tests files
This commit is contained in:
parent
01660430af
commit
a53e30ca9e
|
@ -0,0 +1 @@
|
|||
INSTALLED_APPS += ('passerelle.apps.atreal_openads',)
|
|
@ -0,0 +1,505 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
|
||||
# to run it use the following command:
|
||||
# ~> cd ~/src/passerelle/tests
|
||||
# ~> DJANGO_SETTINGS_MODULE=passerelle.settings pytest -W ignore::django.utils.deprecation.RemovedInDjango20Warning test_atreal_openads.py -vv
|
||||
#
|
||||
# and with 'pytest-cov':
|
||||
# ~> 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 urlparse
|
||||
import pytest
|
||||
import mock
|
||||
import json
|
||||
import sys, os
|
||||
sys.path.append('/home/vagrant/src/passerelle/tests')
|
||||
import utils
|
||||
|
||||
from passerelle.apps.atreal_openads.models import AtrealOpenads, ForwardFile
|
||||
from passerelle.base.models import Job
|
||||
from passerelle.utils.jsonresponse import APIError
|
||||
|
||||
import base64
|
||||
import re
|
||||
|
||||
from django.http import Http404
|
||||
from django.http.request import HttpRequest, QueryDict
|
||||
from django.http.response import HttpResponse, JsonResponse
|
||||
from django.core.files import File
|
||||
from requests import Response
|
||||
|
||||
try:
|
||||
from StringIO import StringIO
|
||||
except ImportError:
|
||||
from io import StringIO
|
||||
|
||||
|
||||
CONNECTOR_NAME = 'atreal-openads'
|
||||
CONNECTOR_SLUG = 'atreal'
|
||||
COLLECTIVITE = 3
|
||||
OPENADS_API_KEY = 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))
|
||||
|
||||
TEST_FILE_TRAC_ICO = '/vagrant/test_files/trac.ico'
|
||||
TEST_FILE_CERFA_DIA = '/vagrant/test_files/cerfa_10072-02.pdf'
|
||||
|
||||
|
||||
def get_file_data(path, b64=True):
|
||||
"""Return the content of a file as a string, in base64 if specified."""
|
||||
with open(path, 'r') as f:
|
||||
if b64:
|
||||
return base64.b64encode(f.read())
|
||||
return f.read()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def atreal_openads(db):
|
||||
return AtrealOpenads.objects.create(
|
||||
slug = CONNECTOR_SLUG,
|
||||
collectivite = COLLECTIVITE,
|
||||
openADS_API_key = OPENADS_API_KEY,
|
||||
openADS_API_url = OPENADS_API_URL
|
||||
)
|
||||
|
||||
|
||||
def test_openads_hello(app, atreal_openads):
|
||||
resp = atreal_openads.hello(None)
|
||||
assert resp['data']['hello'] == 'world'
|
||||
|
||||
resp = atreal_openads.hello(None, 'toto')
|
||||
assert resp['data']['hello'] == 'toto'
|
||||
|
||||
|
||||
def test_openads_echo(app, atreal_openads):
|
||||
meta = {
|
||||
'Host' : 'passerelle.dev.publik.love',
|
||||
'User-Agent' : 'Mozilla Firefox',
|
||||
'Accept' : 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
|
||||
'Accept-Language': 'fr,fr-FR;q=0.8,en-US;q=0.5,en;q=0.3',
|
||||
'Accept-Encoding': 'gzip, deflate, br',
|
||||
'Referer' : 'https://passerelle.dev.publik.love/%s/%s/' % (CONNECTOR_NAME, CONNECTOR_SLUG),
|
||||
'Cookie' : 'publik_portal_agent_url=https%3A%2F%2Fagent-combo.dev.publik.love%2F; '
|
||||
'publik_portal_agent_title=Portail%20Agent; csrftoken-2baa71=' + FAKE_COOKIE_CRSF,
|
||||
'DNT' : '1',
|
||||
'Connection' : 'keep-alive',
|
||||
'Pragma' : 'no-cache',
|
||||
'Cache-Control' : 'no-cache'
|
||||
}
|
||||
|
||||
req = HttpRequest()
|
||||
req._body = ""
|
||||
req.path = '/test'
|
||||
req.method = 'GET'
|
||||
req.encoding = 'utf-8'
|
||||
req.GET = QueryDict(mutable=True) # required because of encoding setter
|
||||
req.POST = QueryDict(mutable=True) # required because of encoding setter
|
||||
req.content_type = 'text/plain'
|
||||
req.content_params = None
|
||||
req.COOKIES = {}
|
||||
req.META = meta
|
||||
req._read_started = False
|
||||
resp = atreal_openads.echo(req, body=False, cookies=False, meta=False, empty=False)
|
||||
jresp = json.loads(resp['data']['received'])
|
||||
assert jresp['scheme'] == 'http'
|
||||
assert jresp['method'] == req.method
|
||||
assert jresp['content_type'] == req.content_type
|
||||
assert jresp['encoding'] == req.encoding
|
||||
|
||||
resp = atreal_openads.echo(req, body=True, cookies=True, meta=True, empty=True)
|
||||
jresp = json.loads(resp['data']['received'])
|
||||
assert jresp['body'] == req._body
|
||||
assert jresp['FILES'] == []
|
||||
assert jresp['content_params'] == None
|
||||
assert jresp['encoding'] == req.encoding
|
||||
assert jresp['GET'] == {}
|
||||
assert jresp['COOKIES'] == {}
|
||||
assert jresp['content_type'] == req.content_type
|
||||
assert jresp['POST'] == {}
|
||||
assert jresp['scheme'] == 'http'
|
||||
assert jresp['method'] == req.method
|
||||
for k,v in req.META.items():
|
||||
jresp['META'][k] == v
|
||||
|
||||
req = HttpRequest()
|
||||
req._body = '{"data": "test"}'
|
||||
req.path = '/test'
|
||||
req.method = 'POST'
|
||||
req.encoding = 'utf-8'
|
||||
req.GET = QueryDict(mutable=True) # required because of encoding setter
|
||||
req.POST = QueryDict(mutable=True) # required because of encoding setter
|
||||
req.content_type = 'application/json; charset=UTF-8'
|
||||
req.content_params = None
|
||||
req.COOKIES = {}
|
||||
req.META = meta
|
||||
req._read_started = False
|
||||
resp = atreal_openads.echo(req, body=False, cookies=False, meta=False, empty=False)
|
||||
jresp = json.loads(resp['data']['received'])
|
||||
assert jresp['scheme'] == 'http'
|
||||
assert jresp['method'] == req.method
|
||||
assert jresp['content_type'] == req.content_type
|
||||
assert jresp['encoding'] == req.encoding
|
||||
|
||||
resp = atreal_openads.echo(req, body=True, cookies=True, meta=True, empty=True)
|
||||
jresp = json.loads(resp['data']['received'])
|
||||
assert jresp['body'] == req._body
|
||||
assert jresp['FILES'] == []
|
||||
assert jresp['content_params'] == None
|
||||
assert jresp['encoding'] == req.encoding
|
||||
assert jresp['GET'] == {}
|
||||
assert jresp['COOKIES'] == {}
|
||||
assert jresp['content_type'] == req.content_type
|
||||
assert jresp['POST'] == {}
|
||||
assert jresp['scheme'] == 'http'
|
||||
assert jresp['method'] == req.method
|
||||
for k,v in req.META.items():
|
||||
jresp['META'][k] == v
|
||||
|
||||
|
||||
def test_openads_echofile(app, atreal_openads):
|
||||
test_file_b64 = get_file_data(TEST_FILE_TRAC_ICO)
|
||||
test_file_json = {
|
||||
'filename' : os.path.basename(TEST_FILE_TRAC_ICO),
|
||||
'content_type' : 'image/x-icon',
|
||||
'b64_content' : 'ccc'
|
||||
}
|
||||
req = HttpRequest()
|
||||
req._body = json.dumps(test_file_json)
|
||||
req.path = '/test'
|
||||
req.method = 'POST'
|
||||
req.encoding = 'utf-8'
|
||||
req.GET = QueryDict(mutable=True) # required because of encoding setter
|
||||
req.POST = QueryDict(mutable=True) # required because of encoding setter
|
||||
req.content_type = 'application/json'
|
||||
req.content_params = None
|
||||
req.COOKIES = {}
|
||||
req.META = {}
|
||||
req._read_started = False
|
||||
jresp = atreal_openads.echofile(req)
|
||||
assert jresp['filename'] == test_file_json['filename']
|
||||
assert jresp['content_type'] == test_file_json['content_type']
|
||||
assert jresp['content'] == test_file_json['b64_content']
|
||||
|
||||
|
||||
def test_openads_afile(app, atreal_openads):
|
||||
test_file_b64 = get_file_data(TEST_FILE_TRAC_ICO)
|
||||
jresp = atreal_openads.afile(None, format='json')
|
||||
assert jresp['afile']['filename'] == 'trac.ico'
|
||||
assert jresp['afile']['content_type'] == 'image/x-icon'
|
||||
assert jresp['afile']['b64_content'] == test_file_b64
|
||||
assert jresp['extra_info'] == 'blabla'
|
||||
|
||||
|
||||
def test_openads_check_status(app, atreal_openads):
|
||||
fake_resp_json = {
|
||||
'message': 'Service online'
|
||||
}
|
||||
fake_resp = JsonResponse(fake_resp_json)
|
||||
with mock.patch('passerelle.utils.Request.get') as requests_get:
|
||||
requests_get.return_value = mock.Mock(content=fake_resp, status_code=200)
|
||||
jresp = atreal_openads.check_status()
|
||||
assert jresp['response'] == 200
|
||||
|
||||
|
||||
def test_openads_create_dossier(app, atreal_openads):
|
||||
fake_resp_json = {
|
||||
'numero_dossier' : FAKE_NUMERO_DOSSIER,
|
||||
'files': [{
|
||||
'b64_content' : get_file_data(TEST_FILE_CERFA_DIA),
|
||||
'content_type' : 'text/plain',
|
||||
'filename' : 'recepisse_depot_%s.pdf' % FAKE_NUMERO_DOSSIER,
|
||||
}]
|
||||
}
|
||||
fake_resp = Response()
|
||||
fake_resp.status_code = 200
|
||||
fake_resp.headers = {'Content-Type': 'application/json'}
|
||||
fake_resp.encoding = 'utf-8'
|
||||
fake_resp.reason = 'OK'
|
||||
fake_resp._content = json.dumps(fake_resp_json)
|
||||
with mock.patch('passerelle.utils.Request.post') as requests_post:
|
||||
requests_post.return_value = fake_resp
|
||||
jresp = atreal_openads.create_dossier(None, 'DIA')
|
||||
assert jresp['data']['numero_dossier'] == fake_resp_json['numero_dossier']
|
||||
assert jresp['data']['files'][0]['b64_content'] == fake_resp_json['files'][0]['b64_content']
|
||||
assert jresp['data']['files'][0]['content_type'] == fake_resp_json['files'][0]['content_type']
|
||||
assert jresp['data']['files'][0]['filename'] == fake_resp_json['files'][0]['filename']
|
||||
|
||||
job = Job.objects.filter(natural_id=FAKE_NUMERO_DOSSIER).last()
|
||||
assert job
|
||||
job_id = job.id
|
||||
assert job.status == 'registered'
|
||||
assert job.method_name == 'upload_user_files'
|
||||
assert job.natural_id == FAKE_NUMERO_DOSSIER
|
||||
assert job.parameters is not None
|
||||
assert len(job.parameters) == 3
|
||||
assert 'file_ids' in job.parameters
|
||||
assert len(job.parameters['file_ids']) == 2
|
||||
file_ids = job.parameters['file_ids']
|
||||
|
||||
FFs = ForwardFile.objects.filter(id__in=file_ids)
|
||||
for FF in FFs:
|
||||
assert len(FF.numero_demande) > 0
|
||||
assert FF.numero_dossier == FAKE_NUMERO_DOSSIER
|
||||
assert len(FF.file_hash) > 0
|
||||
assert FF.upload_status == 'pending'
|
||||
|
||||
fake_resp_json = "You want add some files on %s " % FAKE_NUMERO_DOSSIER
|
||||
fake_resp = Response()
|
||||
fake_resp.status_code = 200
|
||||
fake_resp.headers = {'Content-Type': 'application/json'}
|
||||
fake_resp.encoding = 'utf-8'
|
||||
fake_resp.reason = 'OK'
|
||||
fake_resp._content = json.dumps(fake_resp_json)
|
||||
with mock.patch('passerelle.utils.Request.post') as requests_post:
|
||||
requests_post.return_value = fake_resp
|
||||
atreal_openads.jobs()
|
||||
|
||||
job = Job.objects.get(id=job_id)
|
||||
assert job.status == 'completed'
|
||||
|
||||
FFs = ForwardFile.objects.filter(id__in=file_ids)
|
||||
for FF in FFs:
|
||||
assert FF.upload_status == 'success'
|
||||
|
||||
|
||||
def test_openads_upload2ForwardFile(app, atreal_openads):
|
||||
FF = atreal_openads.upload2ForwardFile(None, None)
|
||||
assert FF is None
|
||||
|
||||
FF = atreal_openads.upload2ForwardFile(TEST_FILE_CERFA_DIA, FAKE_NUMERO_DOSSIER)
|
||||
assert isinstance(FF, ForwardFile)
|
||||
assert len(FF.numero_demande) > 0
|
||||
assert FF.numero_dossier == FAKE_NUMERO_DOSSIER
|
||||
assert FF.type_fichier == 'CERFA'
|
||||
assert FF.orig_filename == os.path.basename(TEST_FILE_CERFA_DIA)
|
||||
assert FF.content_type == 'application/pdf'
|
||||
assert len(FF.file_hash) > 0
|
||||
assert isinstance(FF.upload_file, File)
|
||||
assert FF.upload_status == 'pending'
|
||||
|
||||
|
||||
def test_openads_get_dossier(app, atreal_openads):
|
||||
fake_resp_json = {
|
||||
'etat' : u"Non préemption en cours",
|
||||
'date_depot' : "24/04/2019",
|
||||
'date_decision' : "",
|
||||
'decision' : "",
|
||||
'date_limite_instruction': "24/06/2019"
|
||||
}
|
||||
fake_resp = Response()
|
||||
fake_resp.status_code = 200
|
||||
fake_resp.headers = {'Content-Type': 'application/json'}
|
||||
fake_resp.encoding = 'utf-8'
|
||||
fake_resp.reason = 'OK'
|
||||
fake_resp._content = json.dumps(fake_resp_json)
|
||||
with mock.patch('passerelle.utils.Request.get') as requests_get:
|
||||
requests_get.return_value = fake_resp
|
||||
jresp = atreal_openads.get_dossier(None, 'DIA', FAKE_NUMERO_DOSSIER)
|
||||
assert jresp['data']['etat'] == fake_resp_json['etat']
|
||||
assert jresp['data']['date_depot'] == fake_resp_json['date_depot']
|
||||
assert jresp['data']['date_decision'] == fake_resp_json['date_decision']
|
||||
assert jresp['data']['decision'] == fake_resp_json['decision']
|
||||
assert jresp['data']['date_limite_instruction'] == fake_resp_json['date_limite_instruction']
|
||||
|
||||
fake_resp_json = {
|
||||
'errors' : [{
|
||||
'location' : 'path',
|
||||
'name' : 'Invalid Type',
|
||||
'description' : '"invalid_type" is not one of DIA, PC, DP, AT, PD'
|
||||
}]
|
||||
}
|
||||
fake_resp = Response()
|
||||
fake_resp.status_code = 404
|
||||
fake_resp.headers = {'Content-Type': 'application/json'}
|
||||
fake_resp.encoding = 'utf-8'
|
||||
fake_resp.reason = 'Resource not found'
|
||||
fake_resp._content = json.dumps(fake_resp_json)
|
||||
with pytest.raises(APIError) as e:
|
||||
with mock.patch('passerelle.utils.Request.get') as requests_get:
|
||||
requests_get.return_value = fake_resp
|
||||
jresp = atreal_openads.get_dossier(None, 'invalid_type', FAKE_NUMERO_DOSSIER)
|
||||
assert re.search(r'^HTTP error: 404, \[path\] "invalid_type" is not one of DIA, PC, DP, AT, PD$', str(e.value))
|
||||
|
||||
|
||||
def test_openads_add_file(app, atreal_openads):
|
||||
fake_resp_json = "You want add some files on %s " % FAKE_NUMERO_DOSSIER
|
||||
fake_resp = Response()
|
||||
fake_resp.status_code = 200
|
||||
fake_resp.headers = {'Content-Type': 'application/json'}
|
||||
fake_resp.encoding = 'utf-8'
|
||||
fake_resp.reason = 'OK'
|
||||
fake_resp._content = json.dumps(fake_resp_json)
|
||||
with mock.patch('passerelle.utils.Request.post') as requests_post:
|
||||
requests_post.return_value = fake_resp
|
||||
jresp = atreal_openads.add_file(None, 'DIA', FAKE_NUMERO_DOSSIER)
|
||||
assert jresp['data'] == fake_resp_json
|
||||
|
||||
|
||||
def test_openads_add_file_async(app, atreal_openads):
|
||||
jresp = atreal_openads.add_file_async(None, 'DIA', FAKE_NUMERO_DOSSIER)
|
||||
assert jresp['data']['message'] == 'upload is pending (async)'
|
||||
assert jresp['data']['job_id'] is not None
|
||||
job_id = jresp['data']['job_id']
|
||||
|
||||
job = Job.objects.get(id=job_id)
|
||||
assert job.status == 'registered'
|
||||
assert job.method_name == 'upload_user_files'
|
||||
assert job.natural_id == FAKE_NUMERO_DOSSIER
|
||||
assert job.parameters is not None
|
||||
assert len(job.parameters) == 3
|
||||
assert 'file_ids' in job.parameters
|
||||
assert len(job.parameters['file_ids']) == 1
|
||||
file_id = job.parameters['file_ids'][0]
|
||||
|
||||
FF = ForwardFile.objects.get(id=file_id)
|
||||
assert len(FF.numero_demande) > 0
|
||||
assert FF.numero_dossier == FAKE_NUMERO_DOSSIER
|
||||
assert FF.type_fichier == 'CERFA'
|
||||
assert FF.orig_filename == os.path.basename(TEST_FILE_CERFA_DIA)
|
||||
assert FF.content_type == 'application/pdf'
|
||||
assert len(FF.file_hash) > 0
|
||||
assert FF.upload_status == 'pending'
|
||||
|
||||
|
||||
fake_resp_json = "You want add some files on %s " % FAKE_NUMERO_DOSSIER
|
||||
fake_resp = Response()
|
||||
fake_resp.status_code = 200
|
||||
fake_resp.headers = {'Content-Type': 'application/json'}
|
||||
fake_resp.encoding = 'utf-8'
|
||||
fake_resp.reason = 'OK'
|
||||
fake_resp._content = json.dumps(fake_resp_json)
|
||||
with mock.patch('passerelle.utils.Request.post') as requests_post:
|
||||
requests_post.return_value = fake_resp
|
||||
atreal_openads.jobs()
|
||||
|
||||
job = Job.objects.get(id=job_id)
|
||||
assert job.status == 'completed'
|
||||
|
||||
FF = ForwardFile.objects.get(id=file_id)
|
||||
assert FF.upload_status == 'success'
|
||||
|
||||
|
||||
def test_openads_get_fwd_files_status(app, atreal_openads):
|
||||
with pytest.raises(Http404) as e:
|
||||
resp404 = atreal_openads.get_fwd_files_status(None, FAKE_NUMERO_DOSSIER, fichier_id=18, summary=None)
|
||||
assert re.search(r"^No file matches 'numero_dossier=[^']+' and 'id=[^']+'.$", str(e.value))
|
||||
|
||||
resp404 = atreal_openads.get_fwd_files_status(None, FAKE_NUMERO_DOSSIER, fichier_id=None, summary=None)
|
||||
assert resp404['data'] is not None
|
||||
assert len(resp404['data']) == 0
|
||||
|
||||
FF = atreal_openads.upload2ForwardFile(TEST_FILE_TRAC_ICO, FAKE_NUMERO_DOSSIER)
|
||||
FF.save()
|
||||
assert isinstance(FF, ForwardFile)
|
||||
|
||||
jresp = atreal_openads.get_fwd_files_status(None, FAKE_NUMERO_DOSSIER, fichier_id=None, summary=None)
|
||||
assert jresp['data'] is not None
|
||||
assert len(jresp['data']) == 1
|
||||
assert jresp['data'][0]['id'] == FF.id
|
||||
for k in ['numero_dossier', 'type_fichier', 'file_hash', 'orig_filename', 'content_type', 'upload_status', 'upload_msg']:
|
||||
assert jresp['data'][0][k] == getattr(FF, k)
|
||||
assert jresp['data'][0]['b64_content'] == get_file_data(FF.upload_file.path)
|
||||
assert jresp['data'][0]['last_update_datetime'] == FF.last_update_datetime
|
||||
|
||||
jresp = atreal_openads.get_fwd_files_status(None, FAKE_NUMERO_DOSSIER, fichier_id=FF.id, summary=None)
|
||||
assert jresp['data'] is not None
|
||||
assert len(jresp['data']) == 1
|
||||
assert jresp['data'][0]['id'] == FF.id
|
||||
for k in ['numero_dossier', 'type_fichier', 'file_hash', 'orig_filename', 'content_type', 'upload_status', 'upload_msg']:
|
||||
assert jresp['data'][0][k] == getattr(FF, k)
|
||||
assert jresp['data'][0]['b64_content'] == get_file_data(FF.upload_file.path)
|
||||
assert jresp['data'][0]['last_update_datetime'] == FF.last_update_datetime
|
||||
|
||||
jresp = atreal_openads.get_fwd_files_status(None, FAKE_NUMERO_DOSSIER, fichier_id=None, summary='1')
|
||||
assert jresp['data'] is not None
|
||||
assert jresp['data']['all_forwarded'] == False
|
||||
status_msg = '[%s] %s => %s' % (FF.id, FF.orig_filename, FF.upload_msg)
|
||||
assert len(jresp['data']['pending']) == 1
|
||||
assert jresp['data']['pending'][0] == status_msg
|
||||
assert len(jresp['data']['uploading']) == 0
|
||||
assert len(jresp['data']['success']) == 0
|
||||
assert len(jresp['data']['failed']) == 0
|
||||
|
||||
jresp = atreal_openads.get_fwd_files_status(None, FAKE_NUMERO_DOSSIER, fichier_id=FF.id, summary='1')
|
||||
assert jresp['data'] is not None
|
||||
assert jresp['data']['all_forwarded'] == False
|
||||
status_msg = '[%s] %s => %s' % (FF.id, FF.orig_filename, FF.upload_msg)
|
||||
assert len(jresp['data']['pending']) == 1
|
||||
assert jresp['data']['pending'][0] == status_msg
|
||||
assert len(jresp['data']['uploading']) == 0
|
||||
assert len(jresp['data']['success']) == 0
|
||||
assert len(jresp['data']['failed']) == 0
|
||||
|
||||
|
||||
def test_openads_createForwardFile(app, atreal_openads):
|
||||
jresp = atreal_openads.createForwardFile(None, FAKE_NUMERO_DOSSIER)
|
||||
assert jresp['data'] is not None
|
||||
m = re.search(r"^ForwardFile '(\d+)' created$", jresp['data'])
|
||||
assert m
|
||||
assert int(m.group(1)) > 0
|
||||
file_id = int(m.group(1))
|
||||
|
||||
FF = ForwardFile.objects.get(id=file_id)
|
||||
assert isinstance(FF, ForwardFile)
|
||||
assert len(FF.numero_demande) > 0
|
||||
assert FF.numero_dossier == FAKE_NUMERO_DOSSIER
|
||||
assert FF.type_fichier == 'CERFA'
|
||||
assert FF.orig_filename == os.path.basename(TEST_FILE_CERFA_DIA)
|
||||
assert FF.content_type == 'application/pdf'
|
||||
assert len(FF.file_hash) > 0
|
||||
assert isinstance(FF.upload_file, File)
|
||||
assert FF.upload_status == 'pending'
|
||||
|
||||
|
||||
def test_openads_get_courrier(app, atreal_openads):
|
||||
fake_resp_json = {
|
||||
'files': [{
|
||||
'filename' : "instruction_4.pdf",
|
||||
'content_type' : "text/plain",
|
||||
'b64_content' : get_file_data(TEST_FILE_CERFA_DIA)
|
||||
}]
|
||||
}
|
||||
fake_resp = Response()
|
||||
fake_resp.status_code = 200
|
||||
fake_resp.headers = {'Content-Type': 'application/json'}
|
||||
fake_resp.encoding = 'utf-8'
|
||||
fake_resp.reason = 'OK'
|
||||
fake_resp._content = json.dumps(fake_resp_json)
|
||||
with mock.patch('passerelle.utils.Request.get') as requests_get:
|
||||
requests_get.return_value = fake_resp
|
||||
jresp = atreal_openads.get_courrier(None, 'DIA', FAKE_NUMERO_DOSSIER)
|
||||
assert len(jresp['data']['files']) == len(fake_resp_json['files'])
|
||||
assert jresp['data']['files'][0]['filename'] == fake_resp_json['files'][0]['filename']
|
||||
assert jresp['data']['files'][0]['content_type'] == fake_resp_json['files'][0]['content_type']
|
||||
assert jresp['data']['files'][0]['b64_content'] == fake_resp_json['files'][0]['b64_content']
|
||||
|
||||
|
||||
def test_openads_upload_user_files(app, atreal_openads):
|
||||
FF = atreal_openads.upload2ForwardFile(TEST_FILE_TRAC_ICO, FAKE_NUMERO_DOSSIER)
|
||||
FF.save()
|
||||
assert isinstance(FF, ForwardFile)
|
||||
assert FF.upload_status == 'pending'
|
||||
file_id = FF.id
|
||||
assert file_id
|
||||
|
||||
fake_resp_json = "You want add some files on %s " % FAKE_NUMERO_DOSSIER
|
||||
fake_resp = Response()
|
||||
fake_resp.status_code = 200
|
||||
fake_resp.headers = {'Content-Type': 'application/json'}
|
||||
fake_resp.encoding = 'utf-8'
|
||||
fake_resp.reason = 'OK'
|
||||
fake_resp._content = json.dumps(fake_resp_json)
|
||||
with mock.patch('passerelle.utils.Request.post') as requests_post:
|
||||
requests_post.return_value = fake_resp
|
||||
atreal_openads.upload_user_files('DIA', FAKE_NUMERO_DOSSIER, file_ids=[file_id])
|
||||
|
||||
FFup = ForwardFile.objects.get(id=file_id)
|
||||
assert isinstance(FFup, ForwardFile)
|
||||
for k in ['numero_dossier', 'type_fichier', 'file_hash', 'orig_filename', 'content_type']:
|
||||
assert getattr(FFup, k) == getattr(FF, k)
|
||||
assert FFup.upload_status == 'success'
|
||||
|
||||
|
Reference in New Issue