Skip to content

Mock IA CDX response in test_agg_select_mem_unrewrite_headers #987

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ato merged 1 commit into from
Mar 31, 2026
Merged

Mock IA CDX response in test_agg_select_mem_unrewrite_headers #987

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 9 additions & 3 deletions pywb/warcserver/test/test_handlers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

import webtest
from fakeredis import FakeStrictRedis
from mock import patch
from mock import Mock, patch
import pytest

import os
Expand Down Expand Up @@ -42,6 +42,10 @@
'http://web.archive.org/web/{timestamp}id_/{url}')
}

IA_CDX_IANA = b'''\
org,iana)/ 20161103124134 http://iana.org/ unk 302 3I42H3S6NNFQ2MSVX7XZKYAYSCX5QBYJ 320
org,iana)/ 20161104161551 https://www.iana.org/ warc/revisit - K3MFZ2HC5UGVYQ42CM5RARW7DWQTTEOS 498
'''



Expand Down Expand Up @@ -212,14 +216,16 @@ def test_agg_select_mem_2(self):

assert 'ResErrors' not in resp.headers

def test_agg_select_mem_unrewrite_headers(self):
@patch.object(ia_cdx['ia-cdx'].sesh, 'get')
def test_agg_select_mem_unrewrite_headers(self, mock_get):
mock_get.return_value = Mock(content=IA_CDX_IANA)

resp = self.testapp.get('/cdx_api/resource?closest=20161103124134&url=http://iana.org/')

assert resp.headers['Warcserver-Source-Coll'] == 'ia-cdx'

buff = BytesIO(resp.body)
record = ArcWarcRecordLoader().parse_record_stream(buff, no_record_parse=False)
print(record.http_headers)
assert record.http_headers.get_statuscode() == '200'
#assert record.http_headers.get_header('Location') == 'https://www.iana.org/'

Expand Down
Loading