def setUp(self):
# Test for Query String
self.at = EdgeAuth(**{'key': ET_ENCRYPTION_KEY, 'window_seconds': DEFAULT_WINDOW_SECONDS})
# Test for Cookie
self.cat = EdgeAuth(key=ET_ENCRYPTION_KEY, algorithm='sha1', window_seconds=DEFAULT_WINDOW_SECONDS)
# Test for Header
self.hat = EdgeAuth(key=ET_ENCRYPTION_KEY, algorithm='md5', window_seconds=DEFAULT_WINDOW_SECONDS)
def test_times(self):
if not (sys.version_info[0] == 2 and sys.version_info[1] <= 6):
att = EdgeAuth(key=ET_ENCRYPTION_KEY,
window_seconds=DEFAULT_WINDOW_SECONDS,
escape_early=False)
self.assertEqual(None, att.start_time)
self.assertEqual(None, att.end_time)
# start_time
with self.assertRaises(EdgeAuthError):
att.start_time = -1
att.generate_url_token("/")
with self.assertRaises(EdgeAuthError):
att.start_time = "hello"
att.generate_url_token("/")
# end_time
with self.assertRaises(EdgeAuthError):
att.end_time = -1
att.generate_url_token("/")
with self.assertRaises(EdgeAuthError):
att.end_time = "hello"
att.generate_url_token("/")
# window_seconds
with self.assertRaises(EdgeAuthError):
att.window_seconds = -1
att.generate_url_token("/")
with self.assertRaises(EdgeAuthError):
att.window_seconds = "hello"
att.generate_url_token("/")
def test_url_query_escape_on__ignore_yes_with_salt(self):
query_salt_path = "/salt"
ats = EdgeAuth(key=ET_ENCRYPTION_KEY, salt=ET_SALT, window_seconds=DEFAULT_WINDOW_SECONDS, escape_early=True)
token = ats.generateToken(url=query_salt_path)
url = "http://{0}{1}?{2}={3}".format(ET_HOSTNAME, query_salt_path, ats.token_name, token)
response = requests.get(url)
self.assertEqual(404, response.status_code)
def test_acl_asta_escape_on__ignoreQuery_yes(self):
ata = EdgeAuth(key=ET_ENCRYPTION_KEY,
window_seconds=DEFAULT_WINDOW_SECONDS,
escape_early=False)
token = ata.generate_acl_token('/q_escape_ignore/*')
url = "http://{0}{1}?{2}={3}".format(ET_HOSTNAME,
'/q_escape_ignore/hello',
ata.token_name, token)
response = requests.get(url)
self.assertEqual(404, response.status_code)
def test_acl_deli_escape_on__ignoreQuery_yes(self):
atd = EdgeAuth(key=ET_ENCRYPTION_KEY, window_seconds=DEFAULT_WINDOW_SECONDS, escape_early=False)
acl = ['/q_escape_ignore', '/q_escape_ignore/*']
token = atd.generateToken(acl=EdgeAuth.ACL_DELIMITER.join(acl))
url = "http://{0}{1}?{2}={3}".format(ET_HOSTNAME, '/q_escape_ignore', atd.token_name, token)
response = requests.get(url)
self.assertEqual(404, response.status_code)
url = "http://{0}{1}?{2}={3}".format(ET_HOSTNAME, '/q_escape_ignore/world/', atd.token_name, token)
response = requests.get(url)
self.assertEqual(404, response.status_code)