Add a test for gen_token()