94 lines
2.3 KiB
Python
94 lines
2.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from base import *
|
||
|
|
from requests import *
|
||
|
|
|
||
|
|
#userI:passI for I in [1..4] with [apr-md5, crypt, plain and apr-sha]
|
||
|
|
PASSWORDS="""user1:$apr1$mhpONdUp$xSRcAbK2F6hLFUzW59tzW/
|
||
|
|
user2:JTMoqfZHCS0aI
|
||
|
|
user3:pass3
|
||
|
|
user4:{SHA}LbTBgR9CRYKpD41+53mVzwGNlEM=
|
||
|
|
"""
|
||
|
|
|
||
|
|
# user5:pass5 in realm 'Realm1'
|
||
|
|
DIGESTPASSWORDS="""user5:Realm1:b0590e8c95605dd708226b552fc86a22
|
||
|
|
"""
|
||
|
|
|
||
|
|
class TestAprMd5Fail(CurlRequest):
|
||
|
|
URL = "/test.txt"
|
||
|
|
EXPECT_RESPONSE_CODE = 401
|
||
|
|
AUTH = "user1:test1"
|
||
|
|
|
||
|
|
class TestAprMd5Success(CurlRequest):
|
||
|
|
URL = "/test.txt"
|
||
|
|
EXPECT_RESPONSE_CODE = 200
|
||
|
|
AUTH = "user1:pass1"
|
||
|
|
|
||
|
|
class TestCryptFail(CurlRequest):
|
||
|
|
URL = "/test.txt"
|
||
|
|
EXPECT_RESPONSE_CODE = 401
|
||
|
|
AUTH = "user2:test2"
|
||
|
|
|
||
|
|
class TestCryptSuccess(CurlRequest):
|
||
|
|
URL = "/test.txt"
|
||
|
|
EXPECT_RESPONSE_CODE = 200
|
||
|
|
AUTH = "user2:pass2"
|
||
|
|
|
||
|
|
class TestPlainFail(CurlRequest):
|
||
|
|
URL = "/test.txt?plain"
|
||
|
|
EXPECT_RESPONSE_CODE = 401
|
||
|
|
AUTH = "user3:test3"
|
||
|
|
|
||
|
|
class TestPlainSuccess(CurlRequest):
|
||
|
|
URL = "/test.txt?plain"
|
||
|
|
EXPECT_RESPONSE_CODE = 200
|
||
|
|
AUTH = "user3:pass3"
|
||
|
|
|
||
|
|
class TestAprSha1Fail(CurlRequest):
|
||
|
|
URL = "/test.txt"
|
||
|
|
EXPECT_RESPONSE_CODE = 401
|
||
|
|
AUTH = "user4:test4"
|
||
|
|
|
||
|
|
class TestAprSha1Success(CurlRequest):
|
||
|
|
URL = "/test.txt"
|
||
|
|
EXPECT_RESPONSE_CODE = 200
|
||
|
|
AUTH = "user4:pass4"
|
||
|
|
|
||
|
|
class TestDigestFail(CurlRequest):
|
||
|
|
URL = "/test.txt?digest"
|
||
|
|
EXPECT_RESPONSE_CODE = 401
|
||
|
|
AUTH = "user5:test5"
|
||
|
|
|
||
|
|
class TestDigestSuccess(CurlRequest):
|
||
|
|
URL = "/test.txt?digest"
|
||
|
|
EXPECT_RESPONSE_CODE = 200
|
||
|
|
AUTH = "user5:pass5"
|
||
|
|
|
||
|
|
class Test(GroupTest):
|
||
|
|
group = [
|
||
|
|
TestAprMd5Fail, TestAprMd5Success,
|
||
|
|
TestCryptFail, TestCryptSuccess,
|
||
|
|
TestPlainFail, TestPlainSuccess,
|
||
|
|
TestAprSha1Fail, TestAprSha1Success,
|
||
|
|
TestDigestFail, TestDigestSuccess,
|
||
|
|
]
|
||
|
|
|
||
|
|
def Prepare(self):
|
||
|
|
passwdfile = self.PrepareFile("conf/mod-auth.htpasswd", PASSWORDS)
|
||
|
|
digestfile = self.PrepareFile("conf/mod-auth.htdigest", DIGESTPASSWORDS)
|
||
|
|
|
||
|
|
self.config = """
|
||
|
|
setup {{ module_load ( "mod_auth" ); }}
|
||
|
|
|
||
|
|
auth.debug = true;
|
||
|
|
if req.query == "plain" {{
|
||
|
|
auth.plain ["method": "basic", "realm": "Basic Auth Realm", "file": "{passwdfile}", "ttl": 10];
|
||
|
|
}} else if req.query == "digest" {{
|
||
|
|
auth.htdigest ["method": "basic", "realm": "Realm1", "file": "{digestfile}", "ttl": 10];
|
||
|
|
}} else {{
|
||
|
|
auth.htpasswd ["method": "basic", "realm": "Basic Auth Realm", "file": "{passwdfile}", "ttl": 10];
|
||
|
|
}}
|
||
|
|
|
||
|
|
defaultaction;
|
||
|
|
""".format(passwdfile = passwdfile, digestfile = digestfile)
|