diff options
author | matt <matt@xcolour.net> | 2013-01-28 11:32:18 -0500 |
---|---|---|
committer | matt <matt@xcolour.net> | 2013-01-28 11:32:18 -0500 |
commit | 1afcb52d73271bbbd78f885451aa1b0e78c09871 (patch) | |
tree | 9145840d6036fcbc0b6647c88f679a567fa8c54d /tests/test_grantip.py | |
download | paste-git-stringio.tar.gz |
Import StringIO so it can be used.stringio
Diffstat (limited to 'tests/test_grantip.py')
-rw-r--r-- | tests/test_grantip.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/tests/test_grantip.py b/tests/test_grantip.py new file mode 100644 index 0000000..8d74280 --- /dev/null +++ b/tests/test_grantip.py @@ -0,0 +1,34 @@ +from paste.auth import grantip +from paste.fixture import * + +def test_make_app(): + def application(environ, start_response): + start_response('200 OK', [('content-type', 'text/plain')]) + return [ + str(environ.get('REMOTE_USER')), + ':', + str(environ.get('REMOTE_USER_TOKENS')), + ] + ip_map = { + '127.0.0.1': (None, 'system'), + '192.168.0.0/16': (None, 'worker'), + '192.168.0.5<->192.168.0.8': ('bob', 'editor'), + '192.168.0.8': ('__remove__', '-worker'), + } + app = grantip.GrantIPMiddleware(application, ip_map) + app = TestApp(app) + return app + +def test_req(): + app = test_make_app() + def doit(remote_addr): + res = app.get('/', extra_environ={'REMOTE_ADDR': remote_addr}) + return res.body + assert doit('127.0.0.1') == 'None:system' + assert doit('192.168.15.12') == 'None:worker' + assert doit('192.168.0.4') == 'None:worker' + result = doit('192.168.0.5') + assert result.startswith('bob:') + assert 'editor' in result and 'worker' in result + assert result.count(',') == 1 + assert doit('192.168.0.8') == 'None:editor' |