Add unit tests for Token