Merged PR 4587: initialized local accessToken fields on setAccessToken post success

May Hartov · May Hartov · commit 5216b20e86f5 · 2018-06-06T08:42:22.000Z
initialized local accessToken fields on setAccessToken post success
diff --git a/src/embed.ts b/src/embed.ts
@@ -382,13 +382,16 @@ export abstract class Embed {
     var embedType = this.config.type;
     return this.service.hpm.post<models.IError[]>('/' + embedType + '/token', accessToken, { uid: this.config.uniqueId }, this.iframe.contentWindow)
       .then(response => {
+        this.config.accessToken = accessToken;
+        this.element.setAttribute(Embed.accessTokenAttribute, accessToken);
+        this.service.accessToken = accessToken;
         return response.body;
       })
       .catch(response => {
         throw response.body;
       });
   }
-  
+
   /**
    * Gets an access token from the first available location: config, attribute, global.
    * 
diff --git a/test/test.spec.ts b/test/test.spec.ts
@@ -3102,11 +3102,15 @@ describe('SDK-to-HPM', function () {
           body: {}
         }));
 
+        let newToken = "newToken"
         // Act
-        report.setAccessToken(accessToken)
+        report.setAccessToken(newToken)
           .then(() => {
             // Assert
-            expect(spyHpm.post).toHaveBeenCalledWith('/report/token', accessToken, { uid: uniqueId }, iframe.contentWindow);
+            expect(spyHpm.post).toHaveBeenCalledWith('/report/token', newToken, { uid: uniqueId }, iframe.contentWindow);
+            expect(report.service.accessToken).toEqual(newToken);
+            expect(report.config.accessToken).toEqual(newToken);
+            expect(report.element.getAttribute(embed.Embed.accessTokenAttribute)).toEqual(newToken);
             done();
           });
       });
