X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/2e31a473f22dcea24eaa2e1dbf1b7078647548e1..e4f350462118a86831f92eb375c2bc89cbf85920:/src/common/redirect-to.test.ts

diff --git a/src/common/redirect-to.test.ts b/src/common/redirect-to.test.ts
index d1426c2916..0168fd803f 100644
--- a/src/common/redirect-to.test.ts
+++ b/src/common/redirect-to.test.ts
@@ -6,7 +6,11 @@ import { storeRedirects, handleRedirects } from './redirect-to';
 
 describe('redirect-to', () => {
     const { location } = window;
-    const redirectTo = 'http://localhost/test123';
+    const config: any = {
+        keepWebServiceUrl: 'http://localhost',
+        keepWebServiceInlineUrl: 'http://localhost-inline'
+    };
+    const redirectTo = '/test123';
     const locationTemplate = {
         hash: '',
         hostname: '',
@@ -16,10 +20,10 @@ describe('redirect-to', () => {
         port: '80',
         protocol: 'http',
         search: '',
-        reload: () => {},
-        replace: () => {},
-        assign: () => {},
-        ancestorOrigins: new DOMStringList(),
+        reload: () => { },
+        replace: () => { },
+        assign: () => { },
+        ancestorOrigins: [],
         href: '',
     };
 
@@ -32,9 +36,9 @@ describe('redirect-to', () => {
             delete window.location;
             window.location = {
                 ...locationTemplate,
-                href: `${location.href}?redirectTo=${redirectTo}`,
-            };
-            Object.defineProperty(window, 'sessionStorage', {
+                href: `${location.href}?redirectToDownload=${redirectTo}`,
+            } as any;
+            Object.defineProperty(window, 'localStorage', {
                 value: {
                     setItem: jest.fn(),
                 },
@@ -47,7 +51,7 @@ describe('redirect-to', () => {
             storeRedirects();
 
             // then
-            expect(window.sessionStorage.setItem).toHaveBeenCalledWith('redirectTo', redirectTo);
+            expect(window.localStorage.setItem).toHaveBeenCalledWith('redirectToDownload', redirectTo);
         });
     });
 
@@ -56,9 +60,9 @@ describe('redirect-to', () => {
             delete window.location;
             window.location = {
                 ...locationTemplate,
-                href: `${location.href}?redirectTo=${redirectTo}`,
-            };
-            Object.defineProperty(window, 'sessionStorage', {
+                href: `${location.href}?redirectToDownload=${redirectTo}`,
+            } as any;;
+            Object.defineProperty(window, 'localStorage', {
                 value: {
                     getItem: () => redirectTo,
                     removeItem: jest.fn(),
@@ -68,14 +72,11 @@ describe('redirect-to', () => {
         });
 
         it('should redirect to page when it is present in session storage', () => {
-            // given
-            const token = 'testToken';
-
             // when
-            handleRedirects(token);
+            handleRedirects("abcxyz", config);
 
             // then
-            expect(window.location.href).toBe(`${redirectTo}?api_token=${token}`);
+            expect(window.location.href).toBe(`${config.keepWebServiceUrl}${redirectTo}?api_token=abcxyz`);
         });
     });
-});
\ No newline at end of file
+});