From 11fae781acc2eb78aeb45f3aae0c3ccdceb5404e Mon Sep 17 00:00:00 2001
From: addison <addisonbeck1@gmail.com>
Date: Mon, 1 Nov 2021 16:28:27 -0400
Subject: [PATCH] [bug] Add awaits to newly async calls in TokenService

---
 common/src/services/token.service.ts | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/common/src/services/token.service.ts b/common/src/services/token.service.ts
index 9ca68026..dd60b478 100644
--- a/common/src/services/token.service.ts
+++ b/common/src/services/token.service.ts
@@ -40,9 +40,9 @@ export class TokenService implements TokenServiceAbstraction {
     }
 
     async setToken(token: string): Promise<void> {
-        this.stateService.setAccessToken(token, { storageLocation: StorageLocation.Memory });
+        await this.stateService.setAccessToken(token, { storageLocation: StorageLocation.Memory });
         if (!await this.skipTokenStorage()) {
-           this.stateService.setAccessToken(token, { storageLocation: StorageLocation.Disk });
+           await this.stateService.setAccessToken(token, { storageLocation: StorageLocation.Disk });
         }
     }
 
@@ -54,7 +54,7 @@ export class TokenService implements TokenServiceAbstraction {
         if (await this.skipTokenStorage()) {
             return;
         }
-        return this.stateService.setRefreshToken(refreshToken);
+        return await this.stateService.setRefreshToken(refreshToken);
     }
 
     async getRefreshToken(): Promise<string> {
@@ -81,16 +81,16 @@ export class TokenService implements TokenServiceAbstraction {
         await this.setClientSecret(clientSecret);
     }
 
-    setTwoFactorToken(token: string): Promise<any> {
-        return this.stateService.setTwoFactorToken(token);
+    async setTwoFactorToken(token: string): Promise<any> {
+        return await this.stateService.setTwoFactorToken(token);
     }
 
-    getTwoFactorToken(): Promise<string> {
-        return this.stateService.getTwoFactorToken();
+    async getTwoFactorToken(): Promise<string> {
+        return await this.stateService.getTwoFactorToken();
     }
 
-    clearTwoFactorToken(): Promise<any> {
-        return this.stateService.setTwoFactorToken(null);
+    async clearTwoFactorToken(): Promise<any> {
+        return await this.stateService.setTwoFactorToken(null);
     }
 
     async clearToken(): Promise<any> {