Starts to add tests for TokenGenerator