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