Add tests for HearthstoneTokenBuilder

1 files changed, 18 insertions, 6 deletions

diff --git a/battleNetApi/src/token.cpp b/battleNetApi/src/token.cpp
index 276c334..a54136c 100644
--- a/battleNetApi/src/token.cpp
+++ b/battleNetApi/src/token.cpp
@@ -1,14 +1,26 @@
 #include "token.hpp"
 
-Token::Token(const std::string &iTokenText, const std::chrono::seconds &iLifetime)
-    : _tokenText{iTokenText}, _expirationTime{iLifetime + currentTime()} {}
+namespace bnet {
+Token::Token() {}
 
-std::string Token::text() const { return _tokenText; }
+std::optional<TokenText> Token::text() const { return _tokenText; }
 
-void Token::setText(const std::string &iTokenText) { _tokenText = iTokenText; }
+std::optional<std::chrono::seconds> Token::expirationTime() const { return _expirationTime; }
 
-void Token::setLifetime(const std::chrono::seconds &iExpirationTime) {
-    _expirationTime = iExpirationTime + currentTime();
+Token &Token::setText(const TokenText &iTokenText) {
+    _tokenText = iTokenText;
+    return *this;
+}
+
+Token &Token::setLifetime(const std::chrono::seconds &iLifetime) {
+    _expirationTime = iLifetime + currentTime();
+    return *this;
+}
+
+Token &Token::setExpirationTime(const std::chrono::seconds &iExpirationTime) {
+    _expirationTime = iExpirationTime;
+    return *this;
 }
 
 bool Token::isExpired() { return _expirationTime <= currentTime(); }
+} // namespace bnet