Mirrored_Repos/InvenTree
1
0
Fork 0
mirror of https://github.com/inventree/InvenTree synced 2024-08-30 18:33:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add ability to download file as StringIO or BytesIO

Browse Source
This commit is contained in:
Oliver Walters 2022-05-17 22:28:46 +10:00
parent fd9f25dc0d
commit 55000d5c48
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
InvenTree/InvenTree/api_tester.py
View File

@ -170,7 +170,7 @@ class InvenTreeAPITestCase(APITestCase):
return response
def download_file(self, url, data, expected_code=None, expected_fn=None):
def download_file(self, url, data, expected_code=None, expected_fn=None, decode=False):
"""
Download a file from the server, and return an in-memory file
"""
@ -191,11 +191,20 @@ class InvenTreeAPITestCase(APITestCase):
fn = result.groups()[0]
with io.BytesIO() as fo:
fo.name = fn
fo.write(response.getvalue())
if expected_fn is not None:
self.assertEqual(expected_fn, fn)
return fo
if decode:
# Decode data and return as StringIO file object
fo = io.StringIO()
fo.name = fo
fo.write(response.getvalue().decode('UTF-8'))
else:
# Return a a BytesIO file object
fo = io.BytesIO()
fo.name = fn
fo.write(response.getvalue())
fo.seek(0)
return fo