rename 'generate_csrf_token()' to just 'generate_token()'

and add a 'size' argument
This commit is contained in:
Mike Fährmann
2021-01-11 22:12:40 +01:00
parent f277e48c77
commit 780b6adb91
7 changed files with 28 additions and 11 deletions

View File

@@ -474,6 +474,20 @@ class TestOther(unittest.TestCase):
with self.assertRaises(ValueError):
func(3)
@unittest.skipIf(sys.hexversion < 0x3050000, "missing bytes.hex()")
def test_generate_token(self):
tokens = set()
for _ in range(100):
token = util.generate_token()
tokens.add(token)
self.assertEqual(len(token), 16 * 2)
self.assertRegex(token, r"^[0-9a-f]+$")
self.assertGreaterEqual(len(tokens), 99)
token = util.generate_token(80)
self.assertEqual(len(token), 80 * 2)
self.assertRegex(token, r"^[0-9a-f]+$")
def test_combine_dict(self):
self.assertEqual(
util.combine_dict({}, {}),