def test_documents_processor_basic(): urls = [] docs = documents_processor(urls, 'JobID', client_id) assert docs == { 'client_id': client_id, "type": "docs", 'data': [], 'job_id': 'JobID', 'version': version }
def test_documents_processor_empty(): urls = [] docs = documents_processor(urls, 'JobID', client_id) assert docs == { 'client_id': client_id, 'type': 'docs', 'data': [], 'job_id': 'JobID', 'version': version }
def test_documents_processor(mock_req): urls = [base_url, base_url2] mock_req.get(add_api_key(base_url), status_code=200, text='{"documents": ' '[{"documentId": "CMS-2005-0001-0001", "attachmentCount": 4},\ {"documentId": "CMS-2005-0001-0002", "attachmentCount": 999}]}' ) mock_req.get( add_api_key(base_url2), status_code=200, text='{"documents": ' '[{"documentId": "CMS-2005-0001-0003", "attachmentCount": 88},\ {"documentId": "CMS-2005-0001-0004", "attachmentCount": 666}]}' ) docs = documents_processor(urls, 'Job ID', client_id) assert docs == ({ 'job_id': 'Job ID', 'type': 'docs', 'data': [[{ 'id': 'CMS-2005-0001-0001', 'count': 5 }], [{ 'id': 'CMS-2005-0001-0002', 'count': 1000 }], [{ 'id': 'CMS-2005-0001-0003', 'count': 89 }, { 'id': 'CMS-2005-0001-0004', 'count': 667 }]], 'version': version, 'client_id': str(client_id) })