chore(mediaviewer): add test

box · Oct 24, 2019 · a91af2a · a91af2a
1 parent 9e874ba
commit a91af2a
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/lib/viewers/media/__tests__/MediaBaseViewer-test.js b/src/lib/viewers/media/__tests__/MediaBaseViewer-test.js
@@ -185,6 +185,30 @@ describe('lib/viewers/media/MediaBaseViewer', () => {
         });
     });
 
+    describe('refreshToken()', () => {
+        it('should return the same token if the tokenGenerator is a string', done => {
+            media.options.file = {
+                id: 'file_123',
+            };
+            media.options.tokenGenerator = 'new_token';
+            media.refreshToken().then(token => {
+                expect(token).to.equal('new_token');
+                done();
+            });
+        });
+
+        it('should return a new token if the tokenGenerator is a function', done => {
+            media.options.file = {
+                id: 'file_123',
+            };
+            media.options.tokenGenerator = id => Promise.resolve({ [id]: 'new_token' });
+            media.refreshToken().then(token => {
+                expect(token).to.equal('new_token');
+                done();
+            });
+        });
+    });
+
     describe('errorHandler()', () => {
         it('should handle download error if the viewer was not yet loaded', () => {
             media.mediaUrl = 'foo';