misc: fix consider-using-with pylint error (#62099)

This commit is contained in:
Lauréline Guérin 2022-03-17 17:56:30 +01:00
parent 1e90f91e99
commit 3dbee7bc63
No known key found for this signature in database
GPG Key ID: 1FAB9B9B4F93D473
19 changed files with 383 additions and 346 deletions

View File

@ -799,42 +799,43 @@ class AbstractCartaDSCS(BaseResource):
def pack(self, dossier_id): def pack(self, dossier_id):
dossier = CartaDSDossier.objects.get(id=dossier_id) dossier = CartaDSDossier.objects.get(id=dossier_id)
zip_filename = os.path.join(default_storage.path('cartads_cs'), '%s.zip' % dossier.tracking_code) zip_filename = os.path.join(default_storage.path('cartads_cs'), '%s.zip' % dossier.tracking_code)
zip_file = zipfile.ZipFile(zip_filename, mode='w') with zipfile.ZipFile(zip_filename, mode='w') as zip_file:
liste_pdf = self.liste_pdf(None, dossier.type_dossier_id) liste_pdf = self.liste_pdf(None, dossier.type_dossier_id)
cerfa_id = liste_pdf['data'][0]['id'] cerfa_id = liste_pdf['data'][0]['id']
for cerfa in liste_pdf['data']: for cerfa in liste_pdf['data']:
if cerfa['id'] == 'AUTRES_DEMANDEURS': if cerfa['id'] == 'AUTRES_DEMANDEURS':
continue
cerfa_id = cerfa['id']
break
cerfa_id = cerfa_id.replace('*', '-')
pieces = self.pieces(None, dossier.type_dossier_id, dossier.objet_demande_id, dossier.tracking_code)
for piece in pieces['data']:
cnt = 1
for file in piece['files']:
if not file.get('id'):
continue continue
cartads_file = CartaDSFile.objects.get(id=file['id']) cerfa_id = cerfa['id']
if piece['id'] == 'cerfa-%s-%s' % (dossier.type_dossier_id, dossier.objet_demande_id): break
zip_file.write(cartads_file.uploaded_file.path, '%s.pdf' % cerfa_id) cerfa_id = cerfa_id.replace('*', '-')
elif piece['id'].startswith('cerfa-autres-'): pieces = self.pieces(
zip_file.write( None, dossier.type_dossier_id, dossier.objet_demande_id, dossier.tracking_code
cartads_file.uploaded_file.path, )
'Fiches_complementaires/Cerfa_autres_demandeurs_%d.pdf' % cnt, for piece in pieces['data']:
) cnt = 1
else: for file in piece['files']:
zip_file.write( if not file.get('id'):
cartads_file.uploaded_file.path, continue
'Pieces/%s-%s%s%s' cartads_file = CartaDSFile.objects.get(id=file['id'])
% ( if piece['id'] == 'cerfa-%s-%s' % (dossier.type_dossier_id, dossier.objet_demande_id):
piece['id'], zip_file.write(cartads_file.uploaded_file.path, '%s.pdf' % cerfa_id)
piece['codePiece'], elif piece['id'].startswith('cerfa-autres-'):
cnt, zip_file.write(
os.path.splitext(cartads_file.uploaded_file.path)[-1], cartads_file.uploaded_file.path,
), 'Fiches_complementaires/Cerfa_autres_demandeurs_%d.pdf' % cnt,
) )
cnt += 1 else:
zip_file.close() zip_file.write(
cartads_file.uploaded_file.path,
'Pieces/%s-%s%s%s'
% (
piece['id'],
piece['codePiece'],
cnt,
os.path.splitext(cartads_file.uploaded_file.path)[-1],
),
)
cnt += 1
dossier.zip_ready = True dossier.zip_ready = True
dossier.save() dossier.save()
self.add_job('send_to_cartads', dossier_id=dossier.id) self.add_job('send_to_cartads', dossier_id=dossier.id)
@ -843,8 +844,9 @@ class AbstractCartaDSCS(BaseResource):
ftp = FTP(self.ftp_server) ftp = FTP(self.ftp_server)
ftp.login(self.ftp_username, self.ftp_password) ftp.login(self.ftp_username, self.ftp_password)
ftp.cwd(self.ftp_client_name) ftp.cwd(self.ftp_client_name)
ftp.storbinary('STOR %s' % os.path.basename(zip_filename), open(zip_filename, 'rb')) with open(zip_filename, 'rb') as fd:
ftp.quit() ftp.storbinary('STOR %s' % os.path.basename(zip_filename), fd)
ftp.quit()
def send_to_cartads(self, dossier_id): def send_to_cartads(self, dossier_id):
dossier = CartaDSDossier.objects.get(id=dossier_id) dossier = CartaDSDossier.objects.get(id=dossier_id)

View File

@ -214,9 +214,11 @@ class Cryptor(BaseResource):
if not os.path.exists(metadata_filename): if not os.path.exists(metadata_filename):
raise APIError('unknown uuid', http_status=404) raise APIError('unknown uuid', http_status=404)
content = read_decrypt(open(content_filename, 'rb'), self.private_key) with open(content_filename, 'rb') as fd:
content = read_decrypt(fd, self.private_key)
metadata = json.load(open(metadata_filename, 'r')) with open(metadata_filename, 'r') as fd:
metadata = json.load(fd)
filename = metadata.get('filename') filename = metadata.get('filename')
content_type = metadata.get('content_type') content_type = metadata.get('content_type')

View File

@ -50,14 +50,15 @@ class Command(BaseCommand):
storage = DefaultStorage() storage = DefaultStorage()
lock_filename = storage.path('family-%s/import-orleans-data.lock' % connector.id) lock_filename = storage.path('family-%s/import-orleans-data.lock' % connector.id)
try: try:
fd = open(lock_filename, 'w') fd = open(lock_filename, 'w') # pylint: disable=consider-using-with
fcntl.lockf(fd, fcntl.LOCK_EX | fcntl.LOCK_NB) fcntl.lockf(fd, fcntl.LOCK_EX | fcntl.LOCK_NB)
except IOError: except IOError:
raise CommandError('Command already running.') raise CommandError('Command already running.')
try: try:
archive_name = os.path.basename(options['archive_file']) archive_name = os.path.basename(options['archive_file'])
connector.archive.save(archive_name, File(open(options['archive_file'], 'rb'))) with open(options['archive_file'], 'rb') as archive_fd:
connector.archive.save(archive_name, File(archive_fd))
except Exception as e: except Exception as e:
raise CommandError('Error occured: %s' % e) raise CommandError('Error occured: %s' % e)
finally: finally:

View File

@ -174,14 +174,14 @@ class GenericFamily(BaseResource):
def clean(self): def clean(self):
if self.archive: if self.archive:
try: try:
archive = zipfile.ZipFile(self.archive) with zipfile.ZipFile(self.archive) as archive:
if self.file_format != 'native':
modname = 'passerelle.apps.family.loaders.%s' % self.file_format
__import__(modname)
module = sys.modules[modname]
module.Loader(self).clean(archive)
except zipfile.BadZipfile: except zipfile.BadZipfile:
raise ValidationError(_('Invalid zip file.')) raise ValidationError(_('Invalid zip file.'))
if self.file_format != 'native':
modname = 'passerelle.apps.family.loaders.%s' % self.file_format
__import__(modname)
module = sys.modules[modname]
module.Loader(self).clean(archive)
return super().clean() return super().clean()
@ -195,101 +195,101 @@ class GenericFamily(BaseResource):
if not os.path.exists(invoices_dir): if not os.path.exists(invoices_dir):
os.makedirs(invoices_dir) os.makedirs(invoices_dir)
archive = zipfile.ZipFile(self.archive.path) with zipfile.ZipFile(self.archive.path) as archive:
if self.file_format != 'native': if self.file_format != 'native':
modname = 'passerelle.apps.family.loaders.%s' % self.file_format modname = 'passerelle.apps.family.loaders.%s' % self.file_format
__import__(modname) __import__(modname)
module = sys.modules[modname] module = sys.modules[modname]
module.Loader(self).load(archive) module.Loader(self).load(archive)
return return
archive_files = archive.namelist() archive_files = archive.namelist()
family_files = [d for d in archive_files if d.endswith('.json')] family_files = [d for d in archive_files if d.endswith('.json')]
families = [] families = []
invoices = [] invoices = []
children = [] children = []
adults = [] adults = []
for f in family_files: for f in family_files:
family_data = json_loads(archive.read(f)) family_data = json_loads(archive.read(f))
families.append(family_data['id']) families.append(family_data['id'])
address = family_data.get('address') or {} address = family_data.get('address') or {}
family_data.update(address) family_data.update(address)
data = dict_cherry_pick(
family_data,
(
'login',
'password',
'family_quotient',
('number', 'street_number'),
('postal_code', 'zipcode'),
('street', 'street_name'),
('complement', 'address_complement'),
),
)
family, dummy = Family.objects.update_or_create(
external_id=family_data['id'], resource=self, defaults=data
)
for adult in family_data.get('adults') or []:
adults.append(adult['id'])
adult_address = adult.get('address') or {}
adult.update(adult_address)
data = dict_cherry_pick( data = dict_cherry_pick(
adult, family_data,
( (
'first_name', 'login',
'last_name', 'password',
'phone', 'family_quotient',
('mobile', 'cellphone'),
'sex',
('number', 'street_number'), ('number', 'street_number'),
('postal_code', 'zipcode'), ('postal_code', 'zipcode'),
('street', 'street_name'), ('street', 'street_name'),
('complement', 'address_complement'), ('complement', 'address_complement'),
'country',
), ),
) )
Adult.objects.update_or_create(family=family, external_id=adult['id'], defaults=data) family, dummy = Family.objects.update_or_create(
# cleanup adults external_id=family_data['id'], resource=self, defaults=data
Adult.objects.exclude(external_id__in=adults).delete()
for child in family_data.get('children') or []:
children.append(child['id'])
data = dict_cherry_pick(child, ('first_name', 'last_name', 'sex', 'birthdate'))
Child.objects.get_or_create(family=family, external_id=child['id'], defaults=data)
# cleanup children
Child.objects.exclude(external_id__in=children).delete()
for invoice in family_data['invoices']:
invoices.append(invoice['id'])
data = dict_cherry_pick(
invoice,
(
'label',
('created', 'issue_date'),
'pay_limit_date',
'litigation_date',
'total_amount',
'payment_date',
'amount',
'autobilling',
),
) )
for date_attribute in data.keys(): for adult in family_data.get('adults') or []:
if not date_attribute.endswith('_date'): adults.append(adult['id'])
continue adult_address = adult.get('address') or {}
if date_attribute == 'payment_date': adult.update(adult_address)
data[date_attribute] = get_datetime(data[date_attribute]) data = dict_cherry_pick(
else: adult,
data[date_attribute] = get_date(data[date_attribute]) (
data['paid'] = bool(data.get('payment_date')) 'first_name',
Invoice.objects.update_or_create( 'last_name',
resource=self, family=family, external_id=invoice['id'], defaults=data 'phone',
) ('mobile', 'cellphone'),
if 'invoices/%s.pdf' % invoice['id'] in archive_files: 'sex',
with open(os.path.join(invoices_dir, '%s.pdf' % invoice['id']), 'wb') as fp: ('number', 'street_number'),
fp.write(archive.read('invoices/%s.pdf' % invoice['id'])) ('postal_code', 'zipcode'),
('street', 'street_name'),
('complement', 'address_complement'),
'country',
),
)
Adult.objects.update_or_create(family=family, external_id=adult['id'], defaults=data)
# cleanup adults
Adult.objects.exclude(external_id__in=adults).delete()
for child in family_data.get('children') or []:
children.append(child['id'])
data = dict_cherry_pick(child, ('first_name', 'last_name', 'sex', 'birthdate'))
Child.objects.get_or_create(family=family, external_id=child['id'], defaults=data)
# cleanup children
Child.objects.exclude(external_id__in=children).delete()
for invoice in family_data['invoices']:
invoices.append(invoice['id'])
data = dict_cherry_pick(
invoice,
(
'label',
('created', 'issue_date'),
'pay_limit_date',
'litigation_date',
'total_amount',
'payment_date',
'amount',
'autobilling',
),
)
for date_attribute in data.keys():
if not date_attribute.endswith('_date'):
continue
if date_attribute == 'payment_date':
data[date_attribute] = get_datetime(data[date_attribute])
else:
data[date_attribute] = get_date(data[date_attribute])
data['paid'] = bool(data.get('payment_date'))
Invoice.objects.update_or_create(
resource=self, family=family, external_id=invoice['id'], defaults=data
)
if 'invoices/%s.pdf' % invoice['id'] in archive_files:
with open(os.path.join(invoices_dir, '%s.pdf' % invoice['id']), 'wb') as fp:
fp.write(archive.read('invoices/%s.pdf' % invoice['id']))
# cleanup invoices # cleanup invoices
Invoice.objects.exclude(external_id__in=invoices).delete() Invoice.objects.exclude(external_id__in=invoices).delete()
@ -526,7 +526,8 @@ class Invoice(models.Model):
if not self.has_pdf: if not self.has_pdf:
raise Http404(_('PDF file not found')) raise Http404(_('PDF file not found'))
response = HttpResponse(open(self.pdf_filename(), 'rb').read(), content_type='application/pdf') with open(self.pdf_filename(), 'rb') as fd:
response = HttpResponse(fd.read(), content_type='application/pdf')
response['Content-Disposition'] = 'attachment; filename=%s.pdf' % self.external_id response['Content-Disposition'] = 'attachment; filename=%s.pdf' % self.external_id
return response return response

View File

@ -232,76 +232,78 @@ class Resource(BaseResource):
def process(self, fd): def process(self, fd):
try: try:
archive = zipfile.ZipFile(fd) with zipfile.ZipFile(fd) as archive:
except Exception: # sort files
doc_files = []
ent_files = []
attachments = {}
for name in archive.namelist():
if ENT_PATTERN.match(name):
ent_files.append(name)
if len(ent_files) != 1:
return False, 'too many/few ent files found: %s' % ent_files
ent_file = ent_files[0]
with archive.open(ent_file) as fd:
document = ET.parse(fd)
for pj_node in PIECE_JOINTE_XPATH(document):
code = CODE_XPATH(pj_node)[0].text
code = 'pj_' + code.lower().replace('-', '_')
fichier = FICHIER_XPATH(pj_node)[0].text
attachments.setdefault(code, []).append(fichier)
for doc_node in DOCUMENTS_XPATH(document):
code = CODE_XPATH(doc_node)[0].text
code = 'doc_' + code.lower().replace('-', '_')
fichier = FICHIER_DONNEES_XPATH(doc_node)[0].text
attachments.setdefault(code, []).append(fichier)
doc_files = [
value for l in attachments.values() for value in l if value.lower().endswith('.xml')
]
if len(doc_files) != 1:
return False, 'too many/few doc files found: %s' % doc_files
for key in attachments:
if len(attachments[key]) > 1:
return False, 'too many attachments of kind %s: %r' % (key, attachments[key])
name = attachments[key][0]
with archive.open(attachments[key][0]) as zip_fd:
content = zip_fd.read()
attachments[key] = {
'filename': name,
'content': base64.b64encode(content).decode('ascii'),
'content_type': 'application/octet-stream',
}
if self.procedure == PROCEDURE_RCO and not attachments:
return False, 'no attachments but RCO requires them'
doc_file = doc_files[0]
insee_codes = ROUTAGE_XPATH(document)
if len(insee_codes) != 1:
return False, 'too many/few insee codes found: %s' % insee_codes
insee_code = insee_codes[0]
email = EMAIL_XPATH(document)
email = email[0] if email else ''
data = {
'insee_code': insee_code,
'email': email,
}
data.update(attachments)
with archive.open(doc_file) as fd:
document = ET.parse(fd)
data.update(self.extract_data(document))
if hasattr(self, 'update_data_%s' % self.procedure):
getattr(self, 'update_data_%s' % self.procedure)(data)
except zipfile.BadZipfile:
return False, 'could not load zipfile' return False, 'could not load zipfile'
# sort files
doc_files = []
ent_files = []
attachments = {}
for name in archive.namelist():
if ENT_PATTERN.match(name):
ent_files.append(name)
if len(ent_files) != 1:
return False, 'too many/few ent files found: %s' % ent_files
ent_file = ent_files[0]
with archive.open(ent_file) as fd:
document = ET.parse(fd)
for pj_node in PIECE_JOINTE_XPATH(document):
code = CODE_XPATH(pj_node)[0].text
code = 'pj_' + code.lower().replace('-', '_')
fichier = FICHIER_XPATH(pj_node)[0].text
attachments.setdefault(code, []).append(fichier)
for doc_node in DOCUMENTS_XPATH(document):
code = CODE_XPATH(doc_node)[0].text
code = 'doc_' + code.lower().replace('-', '_')
fichier = FICHIER_DONNEES_XPATH(doc_node)[0].text
attachments.setdefault(code, []).append(fichier)
doc_files = [value for l in attachments.values() for value in l if value.lower().endswith('.xml')]
if len(doc_files) != 1:
return False, 'too many/few doc files found: %s' % doc_files
for key in attachments:
if len(attachments[key]) > 1:
return False, 'too many attachments of kind %s: %r' % (key, attachments[key])
name = attachments[key][0]
with archive.open(attachments[key][0]) as zip_fd:
content = zip_fd.read()
attachments[key] = {
'filename': name,
'content': base64.b64encode(content).decode('ascii'),
'content_type': 'application/octet-stream',
}
if self.procedure == PROCEDURE_RCO and not attachments:
return False, 'no attachments but RCO requires them'
doc_file = doc_files[0]
insee_codes = ROUTAGE_XPATH(document)
if len(insee_codes) != 1:
return False, 'too many/few insee codes found: %s' % insee_codes
insee_code = insee_codes[0]
email = EMAIL_XPATH(document)
email = email[0] if email else ''
data = {
'insee_code': insee_code,
'email': email,
}
data.update(attachments)
with archive.open(doc_file) as fd:
document = ET.parse(fd)
data.update(self.extract_data(document))
if hasattr(self, 'update_data_%s' % self.procedure):
getattr(self, 'update_data_%s' % self.procedure)(data)
return data, None return data, None
def transfer(self, data): def transfer(self, data):

View File

@ -17,7 +17,8 @@ class Command(BaseCommand):
def handle(self, *args, **options): def handle(self, *args, **options):
if options['output']: if options['output']:
output = open(options['output'], 'w') with open(options['output'], 'w') as output:
else: json.dump(export_site(slugs=options['slugs']), output, indent=4)
output = sys.stdout return
output = sys.stdout
json.dump(export_site(slugs=options['slugs']), output, indent=4) json.dump(export_site(slugs=options['slugs']), output, indent=4)

View File

@ -244,8 +244,6 @@ class ZipTemplate(object):
def diff_zip(one, two): def diff_zip(one, two):
differences = []
def compute_diff(one, two, fd_one, fd_two): def compute_diff(one, two, fd_one, fd_two):
content_one = fd_one.read() content_one = fd_one.read()
content_two = fd_two.read() content_two = fd_two.read()
@ -257,32 +255,40 @@ def diff_zip(one, two):
return ['File %s differs' % one] + diff return ['File %s differs' % one] + diff
return 'File %s differs' % one return 'File %s differs' % one
def run(one, two):
differences = []
with zipfile.ZipFile(one) as one_zip, zipfile.ZipFile(two) as two_zip:
one_nl = set(one_zip.namelist())
two_nl = set(two_zip.namelist())
for name in one_nl - two_nl:
differences.append('File %s only in %s' % (name, one))
for name in two_nl - one_nl:
differences.append('File %s only in %s' % (name, two))
for name in one_nl & two_nl:
with one_zip.open(name) as fd_one:
with two_zip.open(name) as fd_two:
difference = compute_diff(name, name, fd_one, fd_two)
if difference:
differences.append(difference)
if not differences:
# check file order in zip
one_zip_files = [zi.filename for zi in one_zip.infolist()]
two_zip_files = [zi.filename for zi in two_zip.infolist()]
if one_zip_files != two_zip_files:
differences.append('Files are not in the same order')
return differences
if not hasattr(one, 'read'): if not hasattr(one, 'read'):
one = open(one, mode='rb') with open(one, mode='rb') as one:
if not hasattr(two, 'read'):
with open(two, 'rb') as two:
return run(one, two)
with two:
return run(one, two)
with one: with one:
if not hasattr(two, 'read'): if not hasattr(two, 'read'):
two = open(two, 'rb') with open(two, 'rb') as two:
return run(one, two)
with two: with two:
with zipfile.ZipFile(one) as one_zip: return run(one, two)
with zipfile.ZipFile(two) as two_zip:
one_nl = set(one_zip.namelist())
two_nl = set(two_zip.namelist())
for name in one_nl - two_nl:
differences.append('File %s only in %s' % (name, one))
for name in two_nl - one_nl:
differences.append('File %s only in %s' % (name, two))
for name in one_nl & two_nl:
with one_zip.open(name) as fd_one:
with two_zip.open(name) as fd_two:
difference = compute_diff(name, name, fd_one, fd_two)
if difference:
differences.append(difference)
if not differences:
# check file order in zip
one_zip_files = [zi.filename for zi in one_zip.infolist()]
two_zip_files = [zi.filename for zi in two_zip.infolist()]
if one_zip_files != two_zip_files:
differences.append('Files are not in the same order')
return differences

View File

@ -20,9 +20,8 @@ def genesys(db):
@pytest.fixture @pytest.fixture
def mock_codifications_ok(): def mock_codifications_ok():
response = open( with open(os.path.join(os.path.dirname(__file__), 'data', 'genesys_select_codifications.xml')) as fd:
os.path.join(os.path.dirname(__file__), 'data', 'genesys_select_codifications.xml') response = fd.read()
).read()
with tests.utils.mock_url(FAKE_URL, response) as mock: with tests.utils.mock_url(FAKE_URL, response) as mock:
yield mock yield mock
@ -143,9 +142,8 @@ def test_ws_link_created(app, genesys):
assert Link.objects.count() == 0 assert Link.objects.count() == 0
RESPONSE_SELECT_USAGER = open( with open(os.path.join(os.path.dirname(__file__), 'data', 'genesys_select_usager.xml')) as fd:
os.path.join(os.path.dirname(__file__), 'data', 'genesys_select_usager.xml') RESPONSE_SELECT_USAGER = fd.read()
).read()
def test_ws_dossiers(app, genesys): def test_ws_dossiers(app, genesys):

View File

@ -290,11 +290,13 @@ def test_pieces_management(connector, app, cached_data):
resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', b'%PDF...')]) resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', b'%PDF...')])
assert resp.json == [{'error': 'The CERFA should be a PDF file.'}] assert resp.json == [{'error': 'The CERFA should be a PDF file.'}]
pdf_contents = open(os.path.join(os.path.dirname(__file__), 'data', 'minimal.pdf'), 'rb').read() with open(os.path.join(os.path.dirname(__file__), 'data', 'minimal.pdf'), 'rb') as fd:
pdf_contents = fd.read()
resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)]) resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)])
assert resp.json == [{'error': 'The CERFA should not be a scanned document.'}] assert resp.json == [{'error': 'The CERFA should not be a scanned document.'}]
pdf_contents = open(os.path.join(os.path.dirname(__file__), 'data', 'pdf-form.pdf'), 'rb').read() with open(os.path.join(os.path.dirname(__file__), 'data', 'pdf-form.pdf'), 'rb') as fd:
pdf_contents = fd.read()
resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)]) resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)])
cerfa_token = resp.json[0]['token'] cerfa_token = resp.json[0]['token']
@ -554,7 +556,8 @@ def test_doc_pieces_management(connector, app, cached_data):
assert len(data[0]['files']) == 1 assert len(data[0]['files']) == 1
assert list(data[0]['files'][0].keys()) == ['url'] assert list(data[0]['files'][0].keys()) == ['url']
pdf_contents = open(os.path.join(os.path.dirname(__file__), 'data', 'pdf-form.pdf'), 'rb').read() with open(os.path.join(os.path.dirname(__file__), 'data', 'pdf-form.pdf'), 'rb') as fd:
pdf_contents = fd.read()
resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)]) resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)])
assert resp.json[0]['token'] assert resp.json[0]['token']
assert CartaDSFile.objects.filter(tracking_code=dossier.tracking_code, sent_to_cartads=None).count() == 1 assert CartaDSFile.objects.filter(tracking_code=dossier.tracking_code, sent_to_cartads=None).count() == 1
@ -594,12 +597,14 @@ def test_daact_pieces_management(connector, app, cached_data):
assert len(piece['files']) == 1 assert len(piece['files']) == 1
assert list(piece['files'][0].keys()) == ['url'] assert list(piece['files'][0].keys()) == ['url']
pdf_contents = open(os.path.join(os.path.dirname(__file__), 'data', 'pdf-form.pdf'), 'rb').read() with open(os.path.join(os.path.dirname(__file__), 'data', 'pdf-form.pdf'), 'rb') as fd:
pdf_contents = fd.read()
resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)]) resp = app.post(data[0]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)])
assert resp.json[0]['token'] assert resp.json[0]['token']
assert CartaDSFile.objects.filter(tracking_code=dossier.tracking_code, sent_to_cartads=None).count() == 1 assert CartaDSFile.objects.filter(tracking_code=dossier.tracking_code, sent_to_cartads=None).count() == 1
pdf_contents = open(os.path.join(os.path.dirname(__file__), 'data', 'pdf-form.pdf'), 'rb').read() with open(os.path.join(os.path.dirname(__file__), 'data', 'pdf-form.pdf'), 'rb') as fd:
pdf_contents = fd.read()
resp = app.post(data[1]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)]) resp = app.post(data[1]['files'][0]['url'], upload_files=[('files[]', 'test.pdf', pdf_contents)])
assert resp.json[0]['token'] assert resp.json[0]['token']
assert CartaDSFile.objects.filter(tracking_code=dossier.tracking_code, sent_to_cartads=None).count() == 2 assert CartaDSFile.objects.filter(tracking_code=dossier.tracking_code, sent_to_cartads=None).count() == 2

View File

@ -60,9 +60,11 @@ def payload(request):
def assert_xml_doc(filename, assertions): def assert_xml_doc(filename, assertions):
schema = etree.XMLSchema(etree.parse(open(os.path.join(get_test_base_dir('cityweb'), 'cityweb.xsd')))) with open(os.path.join(get_test_base_dir('cityweb'), 'cityweb.xsd')) as fd:
schema = etree.XMLSchema(etree.parse(fd))
content = open(filename).read() with open(filename) as fd:
content = fd.read()
xml_content = etree.fromstring(content) xml_content = etree.fromstring(content)
assert len(xml_content.nsmap) == 1 assert len(xml_content.nsmap) == 1
assert xml_content.nsmap['xs'] == "http://tempuri.org/XMLSchema.xsd" assert xml_content.nsmap['xs'] == "http://tempuri.org/XMLSchema.xsd"

View File

@ -529,15 +529,18 @@ def test_raw_uploadfile(mocked_request, app, setup, debug, caplog):
"""simulate the 3 (ordered) HTTP queries involved""" """simulate the 3 (ordered) HTTP queries involved"""
response = {'status': '200'} response = {'status': '200'}
if method == 'GET' and uri == 'http://example.com/cmisatom': if method == 'GET' and uri == 'http://example.com/cmisatom':
content = open('%s/tests/data/cmis/cmis1.out.xml' % os.getcwd(), 'rb').read() with open('%s/tests/data/cmis/cmis1.out.xml' % os.getcwd(), 'rb') as fd:
content = fd.read()
elif ( elif (
method == 'GET' method == 'GET'
and uri and uri
== 'http://example.com/cmisatom/test/path?path=/test-eo&filter=&includeAllowableActions=false&includeACL=false&includePolicyIds=false&includeRelationships=&renditionFilter=' == 'http://example.com/cmisatom/test/path?path=/test-eo&filter=&includeAllowableActions=false&includeACL=false&includePolicyIds=false&includeRelationships=&renditionFilter='
): ):
content = open('%s/tests/data/cmis/cmis2.out.xml' % os.getcwd(), 'rb').read() with open('%s/tests/data/cmis/cmis2.out.xml' % os.getcwd(), 'rb') as fd:
content = fd.read()
elif method == 'POST' and uri == 'http://example.com/cmisatom/test/children?id=L3Rlc3QtZW8%3D': elif method == 'POST' and uri == 'http://example.com/cmisatom/test/children?id=L3Rlc3QtZW8%3D':
expected_input = open('%s/tests/data/cmis/cmis3.in.xml' % os.getcwd(), 'r').read() with open('%s/tests/data/cmis/cmis3.in.xml' % os.getcwd(), 'r') as fd:
expected_input = fd.read()
expected_input = expected_input.replace('\n', '') expected_input = expected_input.replace('\n', '')
expected_input = re.sub('> *<', '><', expected_input) expected_input = re.sub('> *<', '><', expected_input)
input1 = ET.tostring(ET.XML(expected_input)) input1 = ET.tostring(ET.XML(expected_input))
@ -556,7 +559,8 @@ def test_raw_uploadfile(mocked_request, app, setup, debug, caplog):
if input1 != input2: if input1 != input2:
raise Exception('expect [[%s]] but get [[%s]]' % (body, expected_input)) raise Exception('expect [[%s]] but get [[%s]]' % (body, expected_input))
content = open('%s/tests/data/cmis/cmis3.out.xml' % os.getcwd(), 'rb').read() with open('%s/tests/data/cmis/cmis3.out.xml' % os.getcwd(), 'rb') as fd:
content = fd.read()
else: else:
raise Exception('my fault error, url is not yet mocked: %s' % uri) raise Exception('my fault error, url is not yet mocked: %s' % uri)
return (response, content) return (response, content)

View File

@ -375,68 +375,72 @@ def test_fondettes_concerto_validation():
resource.clean() resource.clean()
filepath = os.path.join(os.path.dirname(__file__), 'data', 'family_data.zip') filepath = os.path.join(os.path.dirname(__file__), 'data', 'family_data.zip')
resource.archive = File(open(filepath), 'family_data.zip') with open(filepath) as fd:
with pytest.raises(ValidationError): resource.archive = File(fd, 'family_data.zip')
resource.clean() with pytest.raises(ValidationError):
resource.clean()
def test_orleans_concerto_loader(): def test_orleans_concerto_loader():
# all related objects will also be deleted # all related objects will also be deleted
Family.objects.all().delete() Family.objects.all().delete()
filepath = os.path.join(os.path.dirname(__file__), 'data', 'orleans', 'family_data_orleans.zip') filepath = os.path.join(os.path.dirname(__file__), 'data', 'orleans', 'family_data_orleans.zip')
resource = GenericFamily( with open(filepath, 'rb') as fd:
title='test orleans', resource = GenericFamily(
slug='test-orleans', title='test orleans',
archive=File(open(filepath, 'rb'), 'family_data_orleans.zip'), slug='test-orleans',
file_format='concerto_orleans', archive=File(fd, 'family_data_orleans.zip'),
) file_format='concerto_orleans',
from passerelle.apps.family.loaders.concerto_orleans import Loader )
from passerelle.apps.family.loaders.concerto_orleans import Loader
loader = Loader(resource) loader = Loader(resource)
loader.archive = zipfile.ZipFile(filepath) with zipfile.ZipFile(filepath) as z:
loader.archive = z
families = loader.build_families() families = loader.build_families()
assert len(families) == 18
for family in families.values():
assert family['external_id']
assert family['adults']
assert family['login']
assert family['password']
assert family['zipcode']
assert family['city']
assert len(family['adults']) > 0
for adult in family['adults']: assert len(families) == 18
assert adult['first_name'] for family in families.values():
assert adult['last_name'] assert family['external_id']
assert adult['sex'] assert family['adults']
assert adult['external_id'] assert family['login']
assert adult['zipcode'] assert family['password']
assert adult['city'] assert family['zipcode']
assert len(family['children']) >= 1 assert family['city']
for child in family['children']: assert len(family['adults']) > 0
assert child['external_id']
assert child['first_name']
assert child['last_name']
assert child['sex']
assert 'birthdate' in child
assert 'invoices' in family for adult in family['adults']:
if family['invoices']: assert adult['first_name']
for invoice in family['invoices']: assert adult['last_name']
assert invoice['external_id'] assert adult['sex']
assert invoice['label'] assert adult['external_id']
assert invoice['issue_date'] assert adult['zipcode']
assert invoice['online_payment'] assert adult['city']
assert 'autobilling' in invoice assert len(family['children']) >= 1
assert 'amount' in invoice for child in family['children']:
assert 'total_amount' in invoice assert child['external_id']
assert child['first_name']
assert child['last_name']
assert child['sex']
assert 'birthdate' in child
# there are 4 families with invoices in test data assert 'invoices' in family
assert len([f for f in families.values() if f['invoices']]) == 4 if family['invoices']:
# and 14 families with no invoices for invoice in family['invoices']:
assert len([f for f in families.values() if not f['invoices']]) == 14 assert invoice['external_id']
resource.save() assert invoice['label']
assert invoice['issue_date']
assert invoice['online_payment']
assert 'autobilling' in invoice
assert 'amount' in invoice
assert 'total_amount' in invoice
# there are 4 families with invoices in test data
assert len([f for f in families.values() if f['invoices']]) == 4
# and 14 families with no invoices
assert len([f for f in families.values() if not f['invoices']]) == 14
resource.save()
assert Family.objects.filter(resource=resource).count() == 18 assert Family.objects.filter(resource=resource).count() == 18
assert Adult.objects.all().count() == 31 assert Adult.objects.all().count() == 31
@ -510,12 +514,13 @@ def test_family_pending_invoices_by_nameid():
def test_incorrect_orleans_data(caplog): def test_incorrect_orleans_data(caplog):
filepath = os.path.join(os.path.dirname(__file__), 'data', 'family_incorrect_data_orleans.zip') filepath = os.path.join(os.path.dirname(__file__), 'data', 'family_incorrect_data_orleans.zip')
GenericFamily.objects.create( with open(filepath, 'rb') as fd:
title='test orleans', GenericFamily.objects.create(
slug='test-orleans', title='test orleans',
archive=File(open(filepath, 'rb'), 'family_incorrect_data_orleans.zip'), slug='test-orleans',
file_format='concerto_orleans', archive=File(fd, 'family_incorrect_data_orleans.zip'),
) file_format='concerto_orleans',
)
for record in caplog.records: for record in caplog.records:
assert 'Error occured while importing data:' in record.message assert 'Error occured while importing data:' in record.message
assert record.name == 'passerelle.resource.family.test-orleans' assert record.name == 'passerelle.resource.family.test-orleans'

View File

@ -56,7 +56,8 @@ DEMAND_STATUS = {'closed': True, 'status': 'accepted', 'comment': 'dossier trait
@mock.patch('passerelle.apps.mdel.models.Demand.create_zip', lambda x, y: '1-14-ILE-LA') @mock.patch('passerelle.apps.mdel.models.Demand.create_zip', lambda x, y: '1-14-ILE-LA')
def test_generic_payload_logging(caplog, app, mdel): def test_generic_payload_logging(caplog, app, mdel):
filename = os.path.join(os.path.dirname(__file__), 'data', 'mdel', 'formdata.json') filename = os.path.join(os.path.dirname(__file__), 'data', 'mdel', 'formdata.json')
payload = json.load(open(filename)) with open(filename) as fd:
payload = json.load(fd)
resp = app.post_json('/mdel/test/create', params=payload, status=200) resp = app.post_json('/mdel/test/create', params=payload, status=200)
assert resp.json['data']['demand_id'] == '1-14-ILE-LA' assert resp.json['data']['demand_id'] == '1-14-ILE-LA'
@ -82,7 +83,8 @@ def test_generic_payload_logging(caplog, app, mdel):
@mock.patch('passerelle.utils.Request.get') @mock.patch('passerelle.utils.Request.get')
def test_proxy_logger(mocked_get, caplog, app, arcgis): def test_proxy_logger(mocked_get, caplog, app, arcgis):
payload = open(os.path.join(os.path.dirname(__file__), 'data', 'nancy_arcgis', 'sigresponse.json')).read() with open(os.path.join(os.path.dirname(__file__), 'data', 'nancy_arcgis', 'sigresponse.json')) as fd:
payload = fd.read()
mocked_get.return_value = tests.utils.FakedResponse(content=payload, status_code=200) mocked_get.return_value = tests.utils.FakedResponse(content=payload, status_code=200)
# simple logger # simple logger
@ -160,7 +162,8 @@ def test_proxy_logger(mocked_get, caplog, app, arcgis):
@mock.patch('requests.Session.send') @mock.patch('requests.Session.send')
def test_proxy_logger_transaction_id(mocked_send, app, arcgis): def test_proxy_logger_transaction_id(mocked_send, app, arcgis):
payload = open(os.path.join(os.path.dirname(__file__), 'data', 'nancy_arcgis', 'sigresponse.json')).read() with open(os.path.join(os.path.dirname(__file__), 'data', 'nancy_arcgis', 'sigresponse.json')) as fd:
payload = fd.read()
mocked_send.return_value = tests.utils.FakedResponse(content=payload, status_code=200) mocked_send.return_value = tests.utils.FakedResponse(content=payload, status_code=200)
arcgis.log_evel = 'DEBUG' arcgis.log_evel = 'DEBUG'
arcgis.base_url = 'https://example.net/' arcgis.base_url = 'https://example.net/'
@ -936,7 +939,8 @@ def test_generic_up_in_endpoints_infos(db, app, connector_class, expected):
def test_generic_endpoint_superuser_access(db, app, admin_user, simple_user): def test_generic_endpoint_superuser_access(db, app, admin_user, simple_user):
MDEL.objects.create(slug='test') MDEL.objects.create(slug='test')
filename = os.path.join(os.path.dirname(__file__), 'data', 'mdel', 'formdata.json') filename = os.path.join(os.path.dirname(__file__), 'data', 'mdel', 'formdata.json')
payload = json.load(open(filename)) with open(filename) as fd:
payload = json.load(fd)
app = login(app, username='user', password='user') app = login(app, username='user', password='user')
resp = app.post_json('/mdel/test/create', params=payload, status=403) resp = app.post_json('/mdel/test/create', params=payload, status=403)

View File

@ -127,7 +127,8 @@ def test_contact_mode_typologies_list_with_invalid_xml(app, setup):
with mock.patch('passerelle.utils.Request.post') as request_post: with mock.patch('passerelle.utils.Request.post') as request_post:
response = mock.Mock() response = mock.Mock()
types_filename = os.path.join(os.path.dirname(__file__), 'data', 'grenoble_gru_typologies.xml') types_filename = os.path.join(os.path.dirname(__file__), 'data', 'grenoble_gru_typologies.xml')
types = open(types_filename).read() with open(types_filename) as fd:
types = fd.read()
response.content = types.replace('Courrier', 'Courrier & autres') response.content = types.replace('Courrier', 'Courrier & autres')
request_post.return_value = response request_post.return_value = response
endpoint = reverse( endpoint = reverse(
@ -145,7 +146,8 @@ def test_contact_mode_typologies_list(app, setup):
with mock.patch('passerelle.utils.Request.post') as request_post: with mock.patch('passerelle.utils.Request.post') as request_post:
response = mock.Mock() response = mock.Mock()
types_filename = os.path.join(os.path.dirname(__file__), 'data', 'grenoble_gru_typologies.xml') types_filename = os.path.join(os.path.dirname(__file__), 'data', 'grenoble_gru_typologies.xml')
response.content = open(types_filename).read() with open(types_filename) as fd:
response.content = fd.read()
request_post.return_value = response request_post.return_value = response
endpoint = reverse( endpoint = reverse(
'generic-endpoint', 'generic-endpoint',
@ -171,7 +173,8 @@ def test_contact_mode_typologies_list(app, setup):
def get_typo_response(): def get_typo_response():
types_filename = os.path.join(os.path.dirname(__file__), 'data', 'grenoble_gru_typologies.xml') types_filename = os.path.join(os.path.dirname(__file__), 'data', 'grenoble_gru_typologies.xml')
types = open(types_filename).read() with open(types_filename) as fd:
types = fd.read()
typo_response = mock.Mock() typo_response = mock.Mock()
typo_response.content = types typo_response.content = types
return typo_response return typo_response
@ -357,9 +360,8 @@ def test_get_pavs(app, setup):
with mock.patch('passerelle.utils.Request.post') as request_post: with mock.patch('passerelle.utils.Request.post') as request_post:
response = mock.Mock() response = mock.Mock()
json_response = mock.Mock() json_response = mock.Mock()
json_response.return_value = json.load( with open(os.path.join(os.path.dirname(__file__), 'data', 'grenoble_gru_pavs.json')) as fd:
open(os.path.join(os.path.dirname(__file__), 'data', 'grenoble_gru_pavs.json')) json_response.return_value = json.load(fd)
)
response.json = json_response response.json = json_response
request_post.return_value = response request_post.return_value = response
response = app.get( response = app.get(

View File

@ -200,7 +200,8 @@ def test_export_to_file(app, setup, filetype):
assert Bdp.objects.count() == 1 assert Bdp.objects.count() == 1
bdp.delete() bdp.delete()
assert Bdp.objects.count() == 0 assert Bdp.objects.count() == 0
import_site(json.load(open(f.name)), overwrite=True) with open(f.name) as fd:
import_site(json.load(fd), overwrite=True)
assert Bdp.objects.count() == 1 assert Bdp.objects.count() == 1

View File

@ -73,7 +73,8 @@ def iph_mocked_get(url, params=None, **kwargs):
else: else:
raise Exception('my fault error, url is not yet mocked: %s' % url) raise Exception('my fault error, url is not yet mocked: %s' % url)
response._content = open(target_file, 'rb').read() with open(target_file, 'rb') as fd:
response._content = fd.read()
response.status_code = 200 response.status_code = 200
return response return response
@ -346,9 +347,8 @@ def test_get_file(mocked_post, mocked_get, app, conn):
response = Response() response = Response()
response.status_code = 200 response.status_code = 200
soap_response = open( with open(os.path.join(os.path.dirname(__file__), 'data/iparapheur_get_file_response.xml'), 'rb') as fd:
os.path.join(os.path.dirname(__file__), 'data/iparapheur_get_file_response.xml'), 'rb' soap_response = fd.read()
).read()
response._content = force_bytes(soap_response) response._content = force_bytes(soap_response)
mocked_post.return_value = response mocked_post.return_value = response
@ -398,9 +398,8 @@ def test_get_file_invalid_appendix(mocked_post, mocked_get, app, conn):
response = Response() response = Response()
response.status_code = 200 response.status_code = 200
soap_response = open( with open(os.path.join(os.path.dirname(__file__), 'data/iparapheur_get_file_response.xml'), 'rb') as fd:
os.path.join(os.path.dirname(__file__), 'data/iparapheur_get_file_response.xml'), 'rb' soap_response = fd.read()
).read()
response._content = soap_response response._content = soap_response
mocked_post.return_value = response mocked_post.return_value = response
@ -421,9 +420,8 @@ def test_get_file_not_found_appendix(mocked_post, mocked_get, app, conn):
response = Response() response = Response()
response.status_code = 200 response.status_code = 200
soap_response = open( with open(os.path.join(os.path.dirname(__file__), 'data/iparapheur_get_file_response.xml'), 'rb') as fd:
os.path.join(os.path.dirname(__file__), 'data/iparapheur_get_file_response.xml'), 'rb' soap_response = fd.read()
).read()
response._content = soap_response response._content = soap_response
mocked_post.return_value = response mocked_post.return_value = response
@ -444,9 +442,8 @@ def test_get_file_appendix(mocked_post, mocked_get, app, conn):
response = Response() response = Response()
response.status_code = 200 response.status_code = 200
soap_response = open( with open(os.path.join(os.path.dirname(__file__), 'data/iparapheur_get_file_response.xml'), 'rb') as fd:
os.path.join(os.path.dirname(__file__), 'data/iparapheur_get_file_response.xml'), 'rb' soap_response = fd.read()
).read()
response._content = soap_response response._content = soap_response
mocked_post.return_value = response mocked_post.return_value = response
@ -521,7 +518,8 @@ def test_call_wsdl(mocked_get, app, conn):
) )
resp = app.get(url) resp = app.get(url)
assert resp.headers['content-type'] == 'text/xml' assert resp.headers['content-type'] == 'text/xml'
assert resp.content == open(wsdl_file(), 'rb').read() with open(wsdl_file(), 'rb') as fd:
assert resp.content == fd.read()
@mock.patch('passerelle.utils.Request.get', side_effect=ConnectionError('mocked error')) @mock.patch('passerelle.utils.Request.get', side_effect=ConnectionError('mocked error'))
@ -548,12 +546,14 @@ def test_no_auth_on_wsdl_imports(mocked_post, mocked_load, mocked_get, app, conn
""" """
response_xmlmime, response_post = Response(), Response() response_xmlmime, response_post = Response(), Response()
response_xmlmime.status_code, response_post.status_code = 200, 200 response_xmlmime.status_code, response_post.status_code = 200, 200
response_xmlmime._content = open(xmlmime(), 'rb').read() with open(xmlmime(), 'rb') as fd:
response_xmlmime._content = fd.read()
response_post._content = force_bytes( response_post._content = force_bytes(
"""<?xml version='1.0' encoding='UTF-8'?><S:Envelope xmlns:S="http://schemas.xmlsoap.org/soap/envelope/"><S:Body><echoResponse xmlns="http://www.adullact.org/spring-ws/iparapheur/1.0" xmlns:xmime="http://www.w3.org/2005/05/xmlmime">[publik_test] m'a dit: &quot;ping&quot;!</echoResponse></S:Body></S:Envelope> """<?xml version='1.0' encoding='UTF-8'?><S:Envelope xmlns:S="http://schemas.xmlsoap.org/soap/envelope/"><S:Body><echoResponse xmlns="http://www.adullact.org/spring-ws/iparapheur/1.0" xmlns:xmime="http://www.w3.org/2005/05/xmlmime">[publik_test] m'a dit: &quot;ping&quot;!</echoResponse></S:Body></S:Envelope>
""" """
) )
mocked_load.return_value = open(wsdl_file(), 'rb').read() with open(wsdl_file(), 'rb') as fd:
mocked_load.return_value = fd.read()
mocked_get.return_value = response_xmlmime mocked_get.return_value = response_xmlmime
mocked_post.return_value = response_post mocked_post.return_value = response_post
url = reverse( url = reverse(

View File

@ -57,8 +57,8 @@ def validate_schema(doc, xsd):
def check_zip_file(zipdir, expected_files): def check_zip_file(zipdir, expected_files):
# check files order in zip # check files order in zip
zipres = zipfile.ZipFile(zipdir, 'r') with zipfile.ZipFile(zipdir, 'r') as zipres:
files = [f.filename for f in zipres.infolist()] files = [f.filename for f in zipres.infolist()]
assert sorted(files[: len(expected_files)]) == sorted(expected_files) assert sorted(files[: len(expected_files)]) == sorted(expected_files)

View File

@ -4,9 +4,12 @@ from passerelle.utils.conversion import to_pdf
def test_pdf_to_pdf_do_nothing(): def test_pdf_to_pdf_do_nothing():
pdf = open(os.path.join(os.path.dirname(__file__), 'data', 'minimal.pdf'), 'rb').read() with open(os.path.join(os.path.dirname(__file__), 'data', 'minimal.pdf'), 'rb') as fd:
pdf = fd.read()
assert to_pdf(pdf) == pdf assert to_pdf(pdf) == pdf
pdf = open(os.path.join(os.path.dirname(__file__), 'data', 'minimal_bom.pdf'), 'rb').read() with open(os.path.join(os.path.dirname(__file__), 'data', 'minimal_bom.pdf'), 'rb') as fd:
pdf = fd.read()
assert to_pdf(pdf) == pdf assert to_pdf(pdf) == pdf
pdf = open(os.path.join(os.path.dirname(__file__), 'data', 'minimal_bomutf8.pdf'), 'rb').read() with open(os.path.join(os.path.dirname(__file__), 'data', 'minimal_bomutf8.pdf'), 'rb') as fd:
pdf = fd.read()
assert to_pdf(pdf) == pdf assert to_pdf(pdf) == pdf

View File

@ -146,20 +146,18 @@ def test_with_parts(tpl_builder, dest):
with full_path.open('rb') as fd: with full_path.open('rb') as fd:
with zipfile.ZipFile(fd) as zi: with zipfile.ZipFile(fd) as zi:
assert zi.namelist() == ['coucou-10-part1.xml', 'coucou-10-dôc.xml'] assert zi.namelist() == ['coucou-10-part1.xml', 'coucou-10-dôc.xml']
assert ( with zi.open('coucou-10-part1.xml') as zfd:
zi.open('coucou-10-part1.xml').read().decode('utf-8') assert zfd.read().decode('utf-8') == '<?xml version="1.0"?><body>blabla</body>'
== '<?xml version="1.0"?><body>blabla</body>' with zi.open('coucou-10-dôc.xml') as zfd:
) assert zfd.read().decode('utf-8') == '<a>Héllo World!</a>'
assert zi.open('coucou-10-dôc.xml').read().decode('utf-8') == '<a>Héllo World!</a>'
with io.BytesIO(z.render_to_bytes()) as fd: with io.BytesIO(z.render_to_bytes()) as fd:
with zipfile.ZipFile(fd) as zi: with zipfile.ZipFile(fd) as zi:
assert zi.namelist() == ['coucou-10-part1.xml', 'coucou-10-dôc.xml'] assert zi.namelist() == ['coucou-10-part1.xml', 'coucou-10-dôc.xml']
assert ( with zi.open('coucou-10-part1.xml') as zfd:
zi.open('coucou-10-part1.xml').read().decode('utf-8') assert zfd.read().decode('utf-8') == '<?xml version="1.0"?><body>blabla</body>'
== '<?xml version="1.0"?><body>blabla</body>' with zi.open('coucou-10-dôc.xml') as zfd:
) assert zfd.read().decode('utf-8') == '<a>Héllo World!</a>'
assert zi.open('coucou-10-dôc.xml').read().decode('utf-8') == '<a>Héllo World!</a>'
def test_xml_error(tpl_builder, dest): def test_xml_error(tpl_builder, dest):