1
0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-15 00:03:28 +01:00
pywb/tests/test_perms_app.py
Ilya Kreymer a1ab54c340 first pass at memento support #10!
memento support enabled by default, togglable via 'enable_memento' config property
supporting timegate and memento apis, no timemap yet
supporting pattern 2.3 for archival and pattern 1.3 for proxy modes
also:
simplify exception hierarchy a bit more, move down to utils
make WbRequest and WbResponse extensible with mixins (eg for memento)
2014-03-14 10:46:20 -07:00

54 lines
1.4 KiB
Python

import webtest
from pywb.perms.perms_handler import create_perms_checker_app
from pywb.perms.perms_handler import ALLOW, BLOCK
from pywb.framework.wsgi_wrappers import init_app
class TestPermsApp:
TEST_CONFIG = 'tests/test_config.yaml'
def setup(self):
self.app = init_app(create_perms_checker_app,
load_yaml=True,
config_file=self.TEST_CONFIG)
self.testapp = webtest.TestApp(self.app)
def test_allow(self):
resp = self.testapp.get('/check-access/http://example.com')
assert resp.content_type == 'application/json'
assert ALLOW in resp.body
def test_allow_with_timestamp(self):
resp = self.testapp.get('/check-access/20131024000102/http://example.com')
assert resp.content_type == 'application/json'
assert ALLOW in resp.body
def test_block_with_timestamp(self):
resp = self.testapp.get('/check-access/20131024000102/http://www.iana.org/_img/bookmark_icon.ico')
assert resp.content_type == 'application/json'
assert BLOCK in resp.body
def test_bad_url(self):
resp = self.testapp.get('/check-access/@#$', expect_errors=True, status = 400)
assert resp.status_int == 400
assert 'Invalid Url: http://@' in resp.body
def test_not_found(self):
resp = self.testapp.get('/check-access/#abc', expect_errors=True, status = 404)
assert resp.status_int == 404