pywb/tests/test_prefer_header.py

from .base_config_test import BaseConfigTest, fmod

from pywb.warcserver.index.cdxobject import CDXObject


# ============================================================================
class TestPreferWithRedirects(BaseConfigTest):
    @classmethod
    def setup_class(cls):
        super(TestPreferWithRedirects, cls).setup_class('config_test_redirect_classic.yaml')

    def _assert_pref_headers(self, resp, pref):
        assert resp.headers['Preference-Applied'] == pref
        assert 'Prefer' in resp.headers['Vary']

    def _assert_raw_memento(self, resp):
        self._assert_pref_headers(resp, 'raw')
        assert '"/time-zones"' in resp.text, resp.text
        assert 'wombat.js' not in resp.text

    def _assert_rewritten(self, resp, fmod):
        self._assert_pref_headers(resp, 'rewritten')

        assert '"20140127171238"' in resp.text
        assert 'wombat.js' in resp.text
        assert 'new _WBWombat' in resp.text, resp.text
        assert '/20140127171238{0}/http://www.iana.org/time-zones"'.format(fmod) in resp.text

    def _assert_redir_to_raw(self, resp):
        self._assert_pref_headers(resp, 'raw')

        assert resp.location.endswith('/pywb/20140127171238id_/http://www.iana.org/')
        resp = resp.follow()

        self._assert_raw_memento(resp)

    def _assert_redir_to_rewritten(self, resp, fmod):
        self._assert_pref_headers(resp, 'rewritten')

        assert resp.location.endswith('/pywb/20140127171238{0}/http://www.iana.org/'.format(fmod))
        resp = resp.follow()

        self._assert_rewritten(resp, fmod)

    def test_prefer_redir_timegate_raw(self, fmod):
        headers = {'Prefer': 'raw'}
        fmod_slash = fmod + '/' if fmod else ''
        resp = self.get('/pywb/{0}http://www.iana.org/', fmod_slash, headers=headers, status=307)

        self._assert_redir_to_raw(resp)

    def test_prefer_redir_timegate_rewritten(self, fmod):
        headers = {'Prefer': 'rewritten'}
        fmod_slash = fmod + '/' if fmod else ''
        resp = self.get('/pywb/{0}http://www.iana.org/', fmod_slash, headers=headers, status=307)

        self._assert_redir_to_rewritten(resp, fmod)

    def test_prefer_redir_memento_to_raw(self, fmod):
        headers = {'Prefer': 'raw'}
        resp = self.get('/pywb/20140127171238{0}/http://www.iana.org/', fmod, headers=headers, status=307)

        self._assert_redir_to_raw(resp)

    def test_prefer_redir_memento_redir_to_rewritten_diff_mod(self, fmod):
        headers = {'Prefer': 'rewritten'}
        resp = self.get('/pywb/20140127171238js_/http://www.iana.org/', fmod, headers=headers, status=307)

        self._assert_redir_to_rewritten(resp, fmod)

    def test_prefer_redir_memento_matches_rewritten(self, fmod):
        headers = {'Prefer': 'rewritten'}
        resp = self.get('/pywb/20140127171238{0}/http://www.iana.org/', fmod, headers=headers, status=200)

        self._assert_rewritten(resp, fmod)

    def test_prefer_redir_memento_matches_raw(self):
        headers = {'Prefer': 'raw'}
        resp = self.testapp.get('/pywb/20140127171238id_/http://www.iana.org/', headers=headers, status=200)

        self._assert_raw_memento(resp)

    def test_prefer_redir_invalid(self, fmod):
        headers = {'Prefer': 'unknown'}
        resp = self.get('/pywb/20140127171238{0}/http://www.iana.org/', fmod, headers=headers, status=400)


# ============================================================================
class TestPreferWithNoRedirects(BaseConfigTest):
    @classmethod
    def setup_class(cls):
        super(TestPreferWithNoRedirects, cls).setup_class('config_test.yaml',
                                        custom_config={'enable_prefer': True})

    def _assert_pref_headers(self, resp, pref):
        assert resp.headers['Preference-Applied'] == pref
        assert 'Prefer' in resp.headers['Vary']

    def _assert_raw(self, resp):
        self._assert_pref_headers(resp, 'raw')
        assert '"/time-zones"' in resp.text, resp.text
        assert 'wombat.js' not in resp.text

        assert resp.headers['Content-Location'].endswith('/pywb/20140127171238id_/http://www.iana.org/')

    def _assert_rewritten(self, resp, fmod):
        self._assert_pref_headers(resp, 'rewritten')

        assert '"20140127171238"' in resp.text
        assert 'wombat.js' in resp.text
        assert 'new _WBWombat' in resp.text, resp.text

        assert resp.headers['Content-Location'].endswith('/pywb/20140127171238{0}/http://www.iana.org/'.format(fmod))

    def test_prefer_timegate_raw(self, fmod):
        headers = {'Prefer': 'raw'}
        fmod_slash = fmod + '/' if fmod else ''
        resp = self.get('/pywb/{0}http://www.iana.org/', fmod_slash, headers=headers, status=200)

        self._assert_raw(resp)

    def test_prefer_timegate_rewritten(self, fmod):
        headers = {'Prefer': 'rewritten'}
        fmod_slash = fmod + '/' if fmod else ''
        resp = self.get('/pywb/{0}http://www.iana.org/', fmod_slash, headers=headers, status=200)

        assert '/pywb/{0}http://www.iana.org/time-zones"'.format(fmod_slash) in resp.text
        self._assert_rewritten(resp, fmod)

    def test_prefer_memento_raw(self, fmod):
        headers = {'Prefer': 'raw'}
        resp = self.get('/pywb/20140127171238{0}/http://www.iana.org/', fmod, headers=headers, status=200)

        self._assert_raw(resp)

    def test_prefer_memento_rewritten(self, fmod):
        headers = {'Prefer': 'rewritten'}
        resp = self.get('/pywb/20140127171238{0}/http://www.iana.org/', fmod, headers=headers, status=200)

        self._assert_rewritten(resp, fmod)

    def test_prefer_memento_raw_id_mod(self):
        headers = {'Prefer': 'raw'}
        resp = self.testapp.get('/pywb/20140127171238id_/http://www.iana.org/', headers=headers, status=200)

        self._assert_raw(resp)

    def test_prefer_memento_rewritten_id_mod(self, fmod):
        headers = {'Prefer': 'rewritten'}
        resp = self.get('/pywb/20140127171238id_/http://www.iana.org/', fmod, headers=headers, status=200)

        self._assert_rewritten(resp, fmod)

    def test_prefer_memento_rewritten_diff_mod(self):
        headers = {'Prefer': 'raw'}
        resp = self.get('/pywb/20140127171238js_/http://www.iana.org/', fmod, headers=headers, status=200)

        self._assert_raw(resp)

    def test_prefer_invalid(self, fmod):
        headers = {'Prefer': 'unknown'}
        resp = self.get('/pywb/20140127171238{0}/http://www.iana.org/', fmod, headers=headers, status=400)