diff --git a/assemblyline_client/v4_client/module/ingest.py b/assemblyline_client/v4_client/module/ingest.py index dcf6b0b..b056744 100644 --- a/assemblyline_client/v4_client/module/ingest.py +++ b/assemblyline_client/v4_client/module/ingest.py @@ -46,7 +46,7 @@ def __call__(self, fh=None, path=None, content=None, url=None, sha256=None, fnam else: raise ClientError('Could not guess the file name, please provide an fname parameter', 400) fh.seek(0) - files = {'bin': (fname, fh)} + files = {'bin': fh} request = { 'name': fname, } diff --git a/assemblyline_client/v4_client/module/submit.py b/assemblyline_client/v4_client/module/submit.py index 00c0e16..a86037d 100644 --- a/assemblyline_client/v4_client/module/submit.py +++ b/assemblyline_client/v4_client/module/submit.py @@ -41,7 +41,7 @@ def __call__(self, fh=None, path=None, content=None, url=None, sha256=None, fnam else: raise ClientError('Could not guess the file name, please provide an fname parameter', 400) fh.seek(0) - files = {'bin': (fname, fh)} + files = {'bin': fh} request = { 'name': fname, } diff --git a/test/test_submit.py b/test/test_submit.py index f2e397f..ec7302d 100644 --- a/test/test_submit.py +++ b/test/test_submit.py @@ -104,9 +104,9 @@ def test_submit_dynamic(datastore, client): assert res['sid'] is not None assert res == datastore.submission.get(res['sid'], as_obj=False) assert 'Dynamic Analysis' in res['params']['services']['selected'] - for k in res['params']: + for k, v in submission_data['params'].items(): if k not in ['submitter', 'services', 'description', 'quota_item']: - assert res['params'][k] == submission_data['params'][k] + assert res['params'].get(k) == v def test_resubmit(datastore, client): @@ -117,6 +117,6 @@ def test_resubmit(datastore, client): assert res is not None assert res['sid'] is not None assert res == datastore.submission.get(res['sid'], as_obj=False) - for k in res['params']: + for k, v in submission_data['params'].items(): if k not in ['submitter', 'description', 'quota_item']: - assert res['params'][k] == submission_data['params'][k] + assert res['params'].get(k) == v