Better mocking of TokenStore