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()))