def testSetAPIToken(self): result = connection.execute(auth_user_table.insert({'email': 'a'})) user_id = result.inserted_primary_key[0] token = generate_api_token() connection.execute(set_api_token(token=token, auth_user_id=user_id)) user = get_auth_user(connection, user_id) self.assertTrue(bcrypt_sha256.verify(token, user.token))
def testGetAuthUser(self): result = connection.execute(auth_user_table.insert({'email': 'a'})) user_id = result.inserted_primary_key[0] user = get_auth_user(connection, user_id) self.assertEqual(user.email, 'a')