def test_42_sort_tokens(self): # return pagination tokendata = get_tokens_paginate(sortby=Token.serial, page=1, psize=5) self.assertTrue( len(tokendata.get("tokens")) == 5, len(tokendata.get("tokens"))) # sort ascending tokendata = get_tokens_paginate(sortby=Token.serial, page=1, psize=100, sortdir="asc") self.assertTrue( len(tokendata.get("tokens")) >= 9, len(tokendata.get("tokens"))) tokens = tokendata.get("tokens") for token in tokens: print(token.get("serial")) self.assertTrue(tokens[0].get("serial") == "A8", tokens[0]) self.assertTrue(tokens[-1].get("serial") == "hotptoken", tokens[-1]) # Reverse sorting tokendata = get_tokens_paginate(sortby=Token.serial, page=1, psize=100, sortdir="desc") tokens = tokendata.get("tokens") for token in tokens: print(token.get("serial")) self.assertTrue(tokens[0].get("serial") == "hotptoken") self.assertTrue(tokens[-1].get("serial") == "A8") # sort with string column tokendata = get_tokens_paginate(sortby="serial", page=1, psize=100, sortdir="asc") tokens = tokendata.get("tokens") for token in tokens: print(token.get("serial")) self.assertTrue(tokens[-1].get("serial") == "hotptoken") self.assertTrue(tokens[0].get("serial") == "A8") tokendata = get_tokens_paginate(sortby="serial", page=1, psize=100, sortdir="desc") tokens = tokendata.get("tokens") for token in tokens: print(token.get("serial")) self.assertTrue(tokens[0].get("serial") == "hotptoken") self.assertTrue(tokens[-1].get("serial") == "A8")
def test_42_sort_tokens(self): # return pagination tokendata = get_tokens_paginate(sortby=Token.serial, page=1, psize=5) self.assertTrue(len(tokendata.get("tokens")) == 5, len(tokendata.get("tokens"))) # sort ascending tokendata = get_tokens_paginate(sortby=Token.serial, page=1, psize=100, sortdir="asc") self.assertTrue(len(tokendata.get("tokens")) >= 9, len(tokendata.get("tokens"))) tokens = tokendata.get("tokens") for token in tokens: print(token.get("serial")) self.assertTrue(tokens[0].get("serial") == "A8", tokens[0]) self.assertTrue(tokens[-1].get("serial") == "hotptoken", tokens[-1]) # Reverse sorting tokendata = get_tokens_paginate(sortby=Token.serial, page=1, psize=100, sortdir="desc") tokens = tokendata.get("tokens") for token in tokens: print(token.get("serial")) self.assertTrue(tokens[0].get("serial") == "hotptoken") self.assertTrue(tokens[-1].get("serial") == "A8") # sort with string column tokendata = get_tokens_paginate(sortby="serial", page=1, psize=100, sortdir="asc") tokens = tokendata.get("tokens") for token in tokens: print(token.get("serial")) self.assertTrue(tokens[-1].get("serial") == "hotptoken") self.assertTrue(tokens[0].get("serial") == "A8") tokendata = get_tokens_paginate(sortby="serial", page=1, psize=100, sortdir="desc") tokens = tokendata.get("tokens") for token in tokens: print(token.get("serial")) self.assertTrue(tokens[0].get("serial") == "hotptoken") self.assertTrue(tokens[-1].get("serial") == "A8")
def test_41_get_tokens_paginate(self): # create some tokens for serial in ["S1", "S2", "S3", "A8", "B", "X"]: init_token({ "serial": serial, "type": "hotp", "otpkey": self.otpkey, "realm": self.realm1 }) token_count = 15 # return pagination tokens = get_tokens_paginate(sortby=Token.serial, page=1, psize=5) self.assertTrue( len(tokens.get("tokens")) == 5, len(tokens.get("tokens"))) self.assertEqual(tokens.get("count"), token_count) self.assertTrue(tokens.get("next") == 2, tokens.get("next")) self.assertTrue(tokens.get("prev") is None, tokens.get("prev")) tokens = get_tokens_paginate(sortby=Token.serial, page=2, psize=5) self.assertEqual(len(tokens.get("tokens")), 5) self.assertEqual(tokens.get("count"), token_count) self.assertEqual(tokens.get("next"), 3) self.assertEqual(tokens.get("prev"), 1) tokens = get_tokens_paginate(sortby=Token.serial, page=3, psize=5) self.assertEqual(len(tokens.get("tokens")), 4) self.assertEqual(tokens.get("count"), token_count) self.assertEqual(tokens.get("next"), None) self.assertEqual(tokens.get("prev"), 2) # Test filtering and sorting tokens = get_tokens_paginate(assigned=True, page=1) self.assertTrue( len(tokens.get("tokens")) == 2, len(tokens.get("tokens"))) self.assertTrue(tokens.get("count") == 2, tokens.get("count")) self.assertTrue(tokens.get("next") is None, tokens.get("next")) self.assertTrue(tokens.get("prev") is None, tokens.get("prev")) tokens = get_tokens_paginate(sortby=Token.serial, page=1, sortdir="desc") self.assertTrue(len(tokens.get("tokens")), token_count - 1) self.assertEqual(tokens.get("count"), token_count) self.assertTrue(tokens.get("next") is None, tokens.get("next")) self.assertTrue(tokens.get("prev") is None, tokens.get("prev")) # Test to retrieve tokens of user cornelius tokens = get_tokens_paginate(user=User("cornelius", "realm1")) self.assertTrue( len(tokens.get("tokens")) == 2, len(tokens.get("tokens"))) # test to retrieve tokens with not strict serial matching tokens = get_tokens_paginate(serial="hotp*") self.assertTrue( len(tokens.get("tokens")) == 1, len(tokens.get("tokens")))
def test_41_get_tokens_paginate(self): for serial in ["S1", "S2", "S3", "A8", "B", "X"]: init_token({"serial": "AUTO003", "type": "hotp", "otpkey":self.otpkey, "realm": self.realm1}) # return pagination tokens = get_tokens_paginate(sortby=Token.serial, page=1, psize=5) self.assertTrue(len(tokens.get("tokens")) == 5, len(tokens.get("tokens"))) self.assertTrue(tokens.get("count") == 13, tokens.get("count")) self.assertTrue(tokens.get("next") == 2, tokens.get("next")) self.assertTrue(tokens.get("prev") is None, tokens.get("prev")) tokens = get_tokens_paginate(sortby=Token.serial, page=2, psize=5) self.assertTrue(len(tokens.get("tokens")) == 5, len(tokens.get("tokens"))) self.assertTrue(tokens.get("count") == 13, tokens.get("count")) self.assertTrue(tokens.get("next") == 3, tokens.get("next")) self.assertTrue(tokens.get("prev") == 1, tokens.get("prev")) tokens = get_tokens_paginate(sortby=Token.serial, page=3, psize=5) self.assertTrue(len(tokens.get("tokens")) == 2, len(tokens.get("tokens"))) self.assertTrue(tokens.get("count") == 13, tokens.get("count")) self.assertTrue(tokens.get("next") is None, tokens.get("next")) self.assertTrue(tokens.get("prev") == 2, tokens.get("prev")) # Test filtering and sorting tokens = get_tokens_paginate(assigned=True, page=1) self.assertTrue(len(tokens.get("tokens")) == 2, len(tokens.get("tokens"))) self.assertTrue(tokens.get("count") == 2, tokens.get("count")) self.assertTrue(tokens.get("next") is None, tokens.get("next")) self.assertTrue(tokens.get("prev") is None, tokens.get("prev")) tokens = get_tokens_paginate(sortby=Token.serial, page=1, sortdir="desc") self.assertTrue(len(tokens.get("tokens")) == 12, len(tokens.get("tokens"))) self.assertTrue(tokens.get("count") == 13, tokens.get("count")) self.assertTrue(tokens.get("next") is None, tokens.get("next")) self.assertTrue(tokens.get("prev") is None, tokens.get("prev")) # Test to retrieve tokens of user cornelius tokens = get_tokens_paginate(user=User("cornelius", "realm1")) self.assertTrue(len(tokens.get("tokens")) == 2, len(tokens.get("tokens"))) # test to retrieve tokens with not strict serial matching tokens = get_tokens_paginate(serial="hotp*") self.assertTrue(len(tokens.get("tokens")) == 1, len(tokens.get("tokens")))