mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-15 12:32:18 -05:00
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
|
# This file is part of CherryPy <http://www.cherrypy.org/>
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
# vim:ts=4:sw=4:expandtab:fileencoding=utf-8
|
||
|
|
|
||
|
|
import cherrypy
|
||
|
|
from cherrypy._cpcompat import md5, ntob
|
||
|
|
from cherrypy.lib import auth_basic
|
||
|
|
from cherrypy.test import helper
|
||
|
|
|
||
|
|
|
||
|
|
class BasicAuthTest(helper.CPWebCase):
|
||
|
|
|
||
|
|
def setup_server():
|
||
|
|
class Root:
|
||
|
|
|
||
|
|
def index(self):
|
||
|
|
return "This is public."
|
||
|
|
index.exposed = True
|
||
|
|
|
||
|
|
class BasicProtected:
|
||
|
|
|
||
|
|
def index(self):
|
||
|
|
return "Hello %s, you've been authorized." % (
|
||
|
|
cherrypy.request.login)
|
||
|
|
index.exposed = True
|
||
|
|
|
||
|
|
class BasicProtected2:
|
||
|
|
|
||
|
|
def index(self):
|
||
|
|
return "Hello %s, you've been authorized." % (
|
||
|
|
cherrypy.request.login)
|
||
|
|
index.exposed = True
|
||
|
|
|
||
|
|
userpassdict = {'xuser': 'xpassword'}
|
||
|
|
userhashdict = {'xuser': md5(ntob('xpassword')).hexdigest()}
|
||
|
|
|
||
|
|
def checkpasshash(realm, user, password):
|
||
|
|
p = userhashdict.get(user)
|
||
|
|
return p and p == md5(ntob(password)).hexdigest() or False
|
||
|
|
|
||
|
|
basic_checkpassword_dict = auth_basic.checkpassword_dict(userpassdict)
|
||
|
|
conf = {
|
||
|
|
'/basic': {
|
||
|
|
'tools.auth_basic.on': True,
|
||
|
|
'tools.auth_basic.realm': 'wonderland',
|
||
|
|
'tools.auth_basic.checkpassword': basic_checkpassword_dict
|
||
|
|
},
|
||
|
|
'/basic2': {
|
||
|
|
'tools.auth_basic.on': True,
|
||
|
|
'tools.auth_basic.realm': 'wonderland',
|
||
|
|
'tools.auth_basic.checkpassword': checkpasshash
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
root = Root()
|
||
|
|
root.basic = BasicProtected()
|
||
|
|
root.basic2 = BasicProtected2()
|
||
|
|
cherrypy.tree.mount(root, config=conf)
|
||
|
|
setup_server = staticmethod(setup_server)
|
||
|
|
|
||
|
|
def testPublic(self):
|
||
|
|
self.getPage("/")
|
||
|
|
self.assertStatus('200 OK')
|
||
|
|
self.assertHeader('Content-Type', 'text/html;charset=utf-8')
|
||
|
|
self.assertBody('This is public.')
|
||
|
|
|
||
|
|
def testBasic(self):
|
||
|
|
self.getPage("/basic/")
|
||
|
|
self.assertStatus(401)
|
||
|
|
self.assertHeader('WWW-Authenticate', 'Basic realm="wonderland"')
|
||
|
|
|
||
|
|
self.getPage('/basic/',
|
||
|
|
[('Authorization', 'Basic eHVzZXI6eHBhc3N3b3JX')])
|
||
|
|
self.assertStatus(401)
|
||
|
|
|
||
|
|
self.getPage('/basic/',
|
||
|
|
[('Authorization', 'Basic eHVzZXI6eHBhc3N3b3Jk')])
|
||
|
|
self.assertStatus('200 OK')
|
||
|
|
self.assertBody("Hello xuser, you've been authorized.")
|
||
|
|
|
||
|
|
def testBasic2(self):
|
||
|
|
self.getPage("/basic2/")
|
||
|
|
self.assertStatus(401)
|
||
|
|
self.assertHeader('WWW-Authenticate', 'Basic realm="wonderland"')
|
||
|
|
|
||
|
|
self.getPage('/basic2/',
|
||
|
|
[('Authorization', 'Basic eHVzZXI6eHBhc3N3b3JX')])
|
||
|
|
self.assertStatus(401)
|
||
|
|
|
||
|
|
self.getPage('/basic2/',
|
||
|
|
[('Authorization', 'Basic eHVzZXI6eHBhc3N3b3Jk')])
|
||
|
|
self.assertStatus('200 OK')
|
||
|
|
self.assertBody("Hello xuser, you've been authorized.")
|