mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-17 08:54:02 +01:00
54 lines
1.4 KiB
Python
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 = '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
|