Create TokenBuilder class

1 files changed, 11 insertions, 5 deletions

diff --git a/battleNetApi/src/main.cpp b/battleNetApi/src/main.cpp
index 2b6596a..ccab84c 100644
--- a/battleNetApi/src/main.cpp
+++ b/battleNetApi/src/main.cpp
@@ -23,7 +23,7 @@ class DefaultDownloadMethod : public DownloadMethod {
         aApiUrl.setPageSize("100");
         std::optional<Token> aToken{TokenPool::availableToken()};
         if (aToken.has_value()) {
-            HearthstoneApi::downloadCards(aApiUrl, aToken.value());
+            return HearthstoneApi::downloadCards(aApiUrl, aToken.value());
         }
         return json({});
     };
@@ -32,8 +32,11 @@ class DefaultDownloadMethod : public DownloadMethod {
 class DownloaderContext {
   public:
     DownloaderContext() = default;
-    DownloaderContext(std::unique_ptr<DownloadMethod> iDownloader) : mDownloader(std::move(iDownloader)) {};
-    void setDownloadMethod(std::unique_ptr<DownloadMethod> iDownloader) { mDownloader = std::move(iDownloader); }
+    DownloaderContext(std::unique_ptr<DownloadMethod> iDownloader)
+        : mDownloader(std::move(iDownloader)) {};
+    void setDownloadMethod(std::unique_ptr<DownloadMethod> iDownloader) {
+        mDownloader = std::move(iDownloader);
+    }
     json download() {
         if (mDownloader) {
             return mDownloader->download();
@@ -50,13 +53,16 @@ class DownloaderContext {
 int main(int argc, char *argv[]) {
     DownloaderContext aDownloaderContext;
     std::string aOutputJsonPath;
-    if (argc < 4) {
+    if (argc < 3) {
         TokenPool::feedTokenSource("tests/resources/credentials.json");
-        aDownloaderContext.setDownloadMethod(std::make_unique<DefaultDownloadMethod>());
+        aDownloaderContext.setDownloadMethod(
+            std::make_unique<DefaultDownloadMethod>());
         aOutputJsonPath = "tests/output/list_of_cards.json";
     } else {
         std::string aCredentialsFilePath{argv[1]};
         TokenPool::feedTokenSource(aCredentialsFilePath);
+        aDownloaderContext.setDownloadMethod(
+            std::make_unique<DefaultDownloadMethod>());
         aOutputJsonPath = argv[2];
     }
     std::ofstream aOutputJsonFile(aOutputJsonPath);