Add test for TokenChecker