コード例 #1
0
 def test_11_tokeninfo(self):
     db_token = Token.query.filter_by(serial=self.serial1).first()
     token = TotpTokenClass(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())
コード例 #2
0
 def test_11_tokeninfo(self):
     db_token = Token.query.filter_by(serial=self.serial1).first()
     token = TotpTokenClass(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())
コード例 #3
0
 def test_11_tokeninfo(self):
     db_token = Token.query.filter_by(serial=self.serial1).first()
     token = TotpTokenClass(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("30/12/14 16:00")
     end = token.get_validity_period_end()
     self.assertTrue(end == "30/12/14 16:00", end)
     self.assertRaises(Exception,
                       token.set_validity_period_end, "wrong date")
     # handle validity start date
     token.set_validity_period_start("30/12/13 16:00")
     start = token.get_validity_period_start()
     self.assertTrue(start == "30/12/13 16:00", 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() + datetime.timedelta(5)
     end = end_date.strftime(DATE_FORMAT)
     token.set_validity_period_end(end)
     # - 5 days
     start_date = datetime.datetime.now() - 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() + datetime.timedelta(5)
     end = end_date.strftime(DATE_FORMAT)
     token.set_validity_period_end(end)
     # + 2 days
     start_date = datetime.datetime.now() + 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() - datetime.timedelta(1)
     end = end_date.strftime(DATE_FORMAT)
     token.set_validity_period_end(end)
     # - 10 days
     start_date = datetime.datetime.now() - datetime.timedelta(10)
     start = start_date.strftime(DATE_FORMAT)
     token.set_validity_period_start(start)
     self.assertFalse(token.check_validity_period())
コード例 #4
0
    def test_11_tokeninfo(self):
        db_token = Token.query.filter_by(serial=self.serial1).first()
        token = TotpTokenClass(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("30/12/14 16:00")
        end = token.get_validity_period_end()
        self.assertTrue(end == "30/12/14 16:00", end)
        self.assertRaises(Exception, token.set_validity_period_end,
                          "wrong date")
        # handle validity start date
        token.set_validity_period_start("30/12/13 16:00")
        start = token.get_validity_period_start()
        self.assertTrue(start == "30/12/13 16:00", 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() + datetime.timedelta(5)
        end = end_date.strftime(DATE_FORMAT)
        token.set_validity_period_end(end)
        # - 5 days
        start_date = datetime.datetime.now() - 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() + datetime.timedelta(5)
        end = end_date.strftime(DATE_FORMAT)
        token.set_validity_period_end(end)
        # + 2 days
        start_date = datetime.datetime.now() + 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() - datetime.timedelta(1)
        end = end_date.strftime(DATE_FORMAT)
        token.set_validity_period_end(end)
        # - 10 days
        start_date = datetime.datetime.now() - datetime.timedelta(10)
        start = start_date.strftime(DATE_FORMAT)
        token.set_validity_period_start(start)
        self.assertFalse(token.check_validity_period())