Mock IA CDX response in test_agg_select_mem_unrewrite_headers

The IA CDX service frequently times out causing transient failures of this test.

Author: ato

pywb/warcserver/test/test_handlers.py

@@ -8,7 +8,7 @@
 
 import webtest
 from fakeredis import FakeStrictRedis
-from mock import patch
+from mock import Mock, patch
 import pytest
 
 import os
@@ -42,6 +42,8 @@
                                 'http://web.archive.org/web/{timestamp}id_/{url}')
 }
 
+IA_CDX_IANA = b'''\
+org,iana)/ 20161103124134 http://iana.org/ unk 302 3I42H3S6NNFQ2MSVX7XZKYAYSCX5QBYJ 320'''
 
 
 
@@ -212,14 +214,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/'