def test_02_class_methods(self): db_token = Token.query.filter(Token.serial == self.serial1).first() token = RegistrationTokenClass(db_token) info = token.get_class_info() self.assertTrue(info.get("title") == "Registration Code Token", info) info = token.get_class_info("title") self.assertTrue(info == "Registration Code Token", info)
def test_02_class_methods(self): db_token = Token.query.filter(Token.serial == self.serial1).first() token = RegistrationTokenClass(db_token) info = token.get_class_info() self.assertTrue(info.get("title") == "Registration Code Token", info) info = token.get_class_info("title") self.assertTrue(info == "Registration Code Token", info)
def test_01_create_token(self): db_token = Token(self.serial1, tokentype="registration") db_token.save() token = RegistrationTokenClass(db_token) token.update({}) self.assertTrue(token.token.serial == self.serial1, token) self.assertTrue(token.token.tokentype == "registration", token.token.tokentype) self.assertTrue(token.type == "registration", token) class_prefix = token.get_class_prefix() self.assertTrue(class_prefix == "REG", class_prefix) self.assertTrue(token.get_class_type() == "registration", token)
def test_03_check_password(self): db_token = Token.query.filter(Token.serial == self.serial1).first() token = RegistrationTokenClass(db_token) r = token.check_otp("wrong pw") self.assertTrue(r == -1, r) detail = token.get_init_detail() r = token.check_otp(detail.get("registrationcode")) self.assertTrue(r == 0, r) # check if the token was deleted. db_token = Token.query.filter(Token.serial == self.serial1).first() self.assertEqual(db_token, None)
def test_01_create_token(self): db_token = Token(self.serial1, tokentype="registration") db_token.save() token = RegistrationTokenClass(db_token) token.update({}) self.assertTrue(token.token.serial == self.serial1, token) self.assertTrue(token.token.tokentype == "registration", token.token.tokentype) self.assertTrue(token.type == "registration", token) class_prefix = token.get_class_prefix() self.assertTrue(class_prefix == "REG", class_prefix) self.assertTrue(token.get_class_type() == "registration", token)
def test_03_check_password(self): db_token = Token.query.filter(Token.serial == self.serial1).first() token = RegistrationTokenClass(db_token) r = token.check_otp("wrong pw") self.assertTrue(r == -1, r) detail = token.get_init_detail() r = token.check_otp(detail.get("registrationcode")) self.assertTrue(r == 0, r) # check if the token sill exists after check_otp db_token = Token.query.filter(Token.serial == self.serial1).first() self.assertNotEqual(db_token, None) # check if the token is deleted after inc_success token.inc_count_auth_success() db_token = Token.query.filter(Token.serial == self.serial1).first() self.assertEqual(db_token, None)
def test_03_check_password(self): db_token = Token.query.filter(Token.serial == self.serial1).first() token = RegistrationTokenClass(db_token) r = token.check_otp("wrong pw") self.assertTrue(r == -1, r) detail = token.get_init_detail() r = token.check_otp(detail.get("registrationcode")) self.assertTrue(r == 0, r) # check if the token sill exists after check_otp db_token = Token.query.filter(Token.serial == self.serial1).first() self.assertNotEqual(db_token, None) # check if the token is deleted after inc_success token.inc_count_auth_success() db_token = Token.query.filter(Token.serial == self.serial1).first() self.assertEqual(db_token, None)