def test_rsa_dict_to_key(self): data = open(KEY_FILE_PAIRS[0][0]).read() rsa_dict = top.parse(data) key = top.get_key(rsa_dict) self.assertTrue(key) self.assertTrue(key.e) self.assertTrue(key.d)
def test_x509_dict_to_key(self): data = open(KEY_FILE_PAIRS[0][1]).read() x509_dict = top.parse(data) key = top.get_key(x509_dict) self.assertTrue(key) self.assertTrue(key.e) # "lambda" suppresses exception until called by the test handler self.assertRaises(AttributeError, lambda: key.d)
def test_params(self):
key = 898767890
wrong_key = 'i78657890uoih'
value = 9876890
params_pair = p1.parameter(key, value)
self.assertTrue(key == p1.get_key(params_pair))
self.assertFalse(p1.get_value_by_key(params_pair, wrong_key))
self.assertFalse(p1.parameter('', value))
self.assertEqual(p1.get_value_by_key(params_pair, key), value)
def generate_auth_token(id, expiration=3600):
"""
Generates token with the given id and expiration time
:param id:
:param expiration:
:return:
"""
s = Serializer(config.get_key(), expires_in=expiration)
return s.dumps({'id': id})
def test_create_parameters_list(self):
key1 = 898767890
key2 = 'i78657890uoih'
value1 = 9876890
value2 = '9876890'
param1 = p1.parameter(key1, value1)
param2 = p1.parameter(key2, value2)
params_1 = p1.create_parameters_list(param1, param2, param1, param1)
print(p1.head(p1.car(params_1)))
self.assertEqual(
p1.get_key(
p1.head(
p1.tail(
params_1
)
)
), key2)
def verify_admin_token(token):
"""
:param token:
:param userid:
:return:
"""
s = Serializer(config.get_key())
try:
data = s.loads(token)
except SignatureExpired:
return False # valid token, but expired
except BadSignature:
return False # invalid
if int(data['id']['roleid']) == 3:
return True # valid token and request
return False # valid token but invalid request
def verify_auth_token(token, userid):
"""
:param token:
:param userid:
:return:
"""
s = Serializer(config.get_key())
try:
data = s.loads(token)
except SignatureExpired:
return False # valid token, but expired
except BadSignature:
return False # invalid token
if data['id']['uid'] == int(userid):
return True # valid token and request
return False # valid token but invalid request
def setUp(self):
with open(KEY_FILE_PAIRS[0][1]) as pub:
self.pubkey = top.get_key(top.parse(pub.read()))
with open(KEY_FILE_PAIRS[0][0]) as priv:
self.privkey = top.get_key(top.parse(priv.read()))
def setUp(self): self.pubkey = top.get_key(top.parse(open(KEY_FILE_PAIRS[0][1]).read())) self.privkey = top.get_key(top.parse(open(KEY_FILE_PAIRS[0][0]).read()))
