def test_01_create_token(self): db_token = Token(self.serial1, tokentype="email") db_token.save() token = EmailTokenClass(db_token) token.update({"email": self.email}) token.save() self.assertTrue(token.token.serial == self.serial1, token) self.assertTrue(token.token.tokentype == "email", token.token) self.assertTrue(token.type == "email", token.type) class_prefix = token.get_class_prefix() self.assertTrue(class_prefix == "PIEM", class_prefix) self.assertTrue(token.get_class_type() == "email", token) # create token with dynamic email db_token = Token(self.serial2, tokentype="email") db_token.save() token = EmailTokenClass(db_token) token.update({"dynamic_email": True}) token.save() self.assertTrue(is_true(token.get_tokeninfo("dynamic_email"))) self.assertTrue(token.token.serial == self.serial2, token) self.assertTrue(token.token.tokentype == "email", token.token) self.assertTrue(token.type == "email", token.type) class_prefix = token.get_class_prefix() self.assertTrue(class_prefix == "PIEM", class_prefix) self.assertTrue(token.get_class_type() == "email", token) token.add_user(User(login="******", realm=self.realm1))
def test_01_create_token(self): db_token = Token(self.serial1, tokentype="email") db_token.save() token = EmailTokenClass(db_token) token.update({"email": self.email}) token.save() self.assertTrue(token.token.serial == self.serial1, token) self.assertTrue(token.token.tokentype == "email", token.token) self.assertTrue(token.type == "email", token.type) class_prefix = token.get_class_prefix() self.assertTrue(class_prefix == "PIEM", class_prefix) self.assertTrue(token.get_class_type() == "email", token) # create token with dynamic email db_token = Token(self.serial2, tokentype="email") db_token.save() token = EmailTokenClass(db_token) token.update({"dynamic_email": True}) token.save() self.assertTrue(is_true(token.get_tokeninfo("dynamic_email"))) self.assertTrue(token.token.serial == self.serial2, token) self.assertTrue(token.token.tokentype == "email", token.token) self.assertTrue(token.type == "email", token.type) class_prefix = token.get_class_prefix() self.assertTrue(class_prefix == "PIEM", class_prefix) self.assertTrue(token.get_class_type() == "email", token) token.add_user(User(login="******", realm=self.realm1))
def test_11_tokeninfo(self): db_token = Token.query.filter_by(serial=self.serial1).first() token = EmailTokenClass(db_token) token.add_tokeninfo("key1", "value2") info1 = token.get_tokeninfo() self.assertTrue("key1" in info1, info1) token.add_tokeninfo("key2", "value3") info2 = token.get_tokeninfo() self.assertTrue("key2" in info2, info2) token.set_tokeninfo(info1) info2 = token.get_tokeninfo() self.assertTrue("key2" not in info2, info2) self.assertTrue(token.get_tokeninfo("key1") == "value2", info2) # auth counter token.set_count_auth_success_max(200) token.set_count_auth_max(1000) token.set_count_auth_success(100) token.inc_count_auth_success() token.set_count_auth(200) token.inc_count_auth() self.assertTrue(token.get_count_auth_success_max() == 200) self.assertTrue(token.get_count_auth_success() == 101) self.assertTrue(token.get_count_auth_max() == 1000) self.assertTrue(token.get_count_auth() == 201) self.assertTrue(token.check_auth_counter()) token.set_count_auth_max(10) self.assertFalse(token.check_auth_counter()) token.set_count_auth_max(1000) token.set_count_auth_success_max(10) self.assertFalse(token.check_auth_counter()) # handle validity end date token.set_validity_period_end("2014-12-30T16:00+0200") end = token.get_validity_period_end() self.assertTrue(end == "2014-12-30T16:00+0200", end) self.assertRaises(Exception, token.set_validity_period_end, "wrong date") # handle validity start date token.set_validity_period_start("2013-12-30T16:00+0200") start = token.get_validity_period_start() self.assertTrue(start == "2013-12-30T16:00+0200", start) self.assertRaises(Exception, token.set_validity_period_start, "wrong date") self.assertFalse(token.check_validity_period()) # check validity period # +5 days end_date = datetime.datetime.now(tzlocal()) + datetime.timedelta(5) end = end_date.strftime(DATE_FORMAT) token.set_validity_period_end(end) # - 5 days start_date = datetime.datetime.now(tzlocal()) - datetime.timedelta(5) start = start_date.strftime(DATE_FORMAT) token.set_validity_period_start(start) self.assertTrue(token.check_validity_period()) # check before start date # +5 days end_date = datetime.datetime.now(tzlocal()) + datetime.timedelta(5) end = end_date.strftime(DATE_FORMAT) token.set_validity_period_end(end) # + 2 days start_date = datetime.datetime.now(tzlocal()) + datetime.timedelta(2) start = start_date.strftime(DATE_FORMAT) token.set_validity_period_start(start) self.assertFalse(token.check_validity_period()) # check after enddate # -1 day end_date = datetime.datetime.now(tzlocal()) - datetime.timedelta(1) end = end_date.strftime(DATE_FORMAT) token.set_validity_period_end(end) # - 10 days start_date = datetime.datetime.now(tzlocal()) - datetime.timedelta(10) start = start_date.strftime(DATE_FORMAT) token.set_validity_period_start(start) self.assertFalse(token.check_validity_period())
def test_08_info(self): db_token = Token.query.filter_by(serial=self.serial1).first() token = EmailTokenClass(db_token) token.set_hashlib("sha1") ti = token.get_tokeninfo() self.assertTrue("hashlib" in ti, ti)
def test_11_tokeninfo(self): db_token = Token.query.filter_by(serial=self.serial1).first() token = EmailTokenClass(db_token) token.add_tokeninfo("key1", "value2") info1 = token.get_tokeninfo() self.assertTrue("key1" in info1, info1) token.add_tokeninfo("key2", "value3") info2 = token.get_tokeninfo() self.assertTrue("key2" in info2, info2) token.set_tokeninfo(info1) info2 = token.get_tokeninfo() self.assertTrue("key2" not in info2, info2) self.assertTrue(token.get_tokeninfo("key1") == "value2", info2) # auth counter token.set_count_auth_success_max(200) token.set_count_auth_max(1000) token.set_count_auth_success(100) token.inc_count_auth_success() token.set_count_auth(200) token.inc_count_auth() self.assertTrue(token.get_count_auth_success_max() == 200) self.assertTrue(token.get_count_auth_success() == 101) self.assertTrue(token.get_count_auth_max() == 1000) self.assertTrue(token.get_count_auth() == 201) self.assertTrue(token.check_auth_counter()) token.set_count_auth_max(10) self.assertFalse(token.check_auth_counter()) token.set_count_auth_max(1000) token.set_count_auth_success_max(10) self.assertFalse(token.check_auth_counter()) # handle validity end date token.set_validity_period_end("2014-12-30T16:00+0200") end = token.get_validity_period_end() self.assertTrue(end == "2014-12-30T16:00+0200", end) self.assertRaises(Exception, token.set_validity_period_end, "wrong date") # handle validity start date token.set_validity_period_start("2013-12-30T16:00+0200") start = token.get_validity_period_start() self.assertTrue(start == "2013-12-30T16:00+0200", start) self.assertRaises(Exception, token.set_validity_period_start, "wrong date") self.assertFalse(token.check_validity_period()) # check validity period # +5 days end_date = datetime.datetime.now(tzlocal()) + datetime.timedelta(5) end = end_date.strftime(DATE_FORMAT) token.set_validity_period_end(end) # - 5 days start_date = datetime.datetime.now(tzlocal()) - datetime.timedelta(5) start = start_date.strftime(DATE_FORMAT) token.set_validity_period_start(start) self.assertTrue(token.check_validity_period()) # check before start date # +5 days end_date = datetime.datetime.now(tzlocal()) + datetime.timedelta(5) end = end_date.strftime(DATE_FORMAT) token.set_validity_period_end(end) # + 2 days start_date = datetime.datetime.now(tzlocal()) + datetime.timedelta(2) start = start_date.strftime(DATE_FORMAT) token.set_validity_period_start(start) self.assertFalse(token.check_validity_period()) # check after enddate # -1 day end_date = datetime.datetime.now(tzlocal()) - datetime.timedelta(1) end = end_date.strftime(DATE_FORMAT) token.set_validity_period_end(end) # - 10 days start_date = datetime.datetime.now(tzlocal()) - datetime.timedelta(10) start = start_date.strftime(DATE_FORMAT) token.set_validity_period_start(start) self.assertFalse(token.check_validity_period())
def test_08_info(self): db_token = Token.query.filter_by(serial=self.serial1).first() token = EmailTokenClass(db_token) token.set_hashlib("sha1") ti = token.get_tokeninfo() self.assertTrue("hashlib" in ti, ti)