Update tests for context checking in auth tokens