def test_totp_token(self):
seed = Util.generate_random_base32()
otp = Util.generate_totp(seed)
self.assertIsNotNone(otp)
result = Util.verify_totp(seed, otp)
self.assertTrue(result)
def authenticate_user(self) -> UserModel:
self.user_model = self.user_model.find_one(
{"username": self.user_model.username})
if bcrypt.checkpw(self.user_model.password.encode('utf-8'),
self.user_model.password.encode('utf-8')):
self.user_model.auth_token = Util.generate_token()
self.user_model.commit(
conditions={'username': self.user_model.username})
return self.user_model
return None
def test_encrypted_string(self):
plaint_text = 'test'
encrypted_string = Util.encrypt3DESData('testkey', plaint_text)
print(encrypted_string)
self.assertNotEqual(plaint_text, encrypted_string)
def test_verify_totp(self):
seed = Util.generate_random_base32()
otp = Util.generate_totp(seed)
result = Util.verify_totp(seed, otp)
self.assertTrue(result)
def test_generate_token_is_unique(self):
token = Util.generate_random_base32()
token2 = Util.generate_random_base32()
self.assertNotEqual(token, token2)
def test_generate_token_is_not_none(self):
token = Util.generate_random_base32()
self.assertIsNotNone(token)
def test_decrypted_string(self):
encrypted_text = 'cinQDDzrFSk='
decrypted_text = Util.decrypt3DESData('testkey', encrypted_text)
self.assertNotEqual(decrypted_text, encrypted_text)
def __init__(self, user_model: UserModel):
self.user_model = user_model
self.util = Util()
def regenerate_user_otp(self, nonce):
user_model = self.user_model.find_one({'auth_nonce': nonce})
user_model_2 = copy.deepcopy(user_model)
user_model.auth_nonce = Util.generate_token()
user_model.commit(conditions={'auth_nonce': user_model_2.auth_nonce})
def __init__(self, user_dict: dict):
self.user_object = UserModel(**user_dict)
self.user_service = UserService(self.user_object)
self.util = Util()