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