@thoughtspot/visual-embed-sdk 1.25.0 → 1.26.0

package/src/utils/graphql/answerService/answerService.spec.ts

@@ -108,7 +108,7 @@ describe('Answer service tests', () => {
         expect(data.data.foo).toBe(1);
     });
 
-    test('fetchCSVBlob should call the right API', async () => {
+    test('fetchCSVBlob should call the right API', () => {
         fetchMock.once('Bla');
         const answerService = createAnswerService();
         answerService.fetchCSVBlob(undefined, true);
@@ -116,6 +116,18 @@ describe('Answer service tests', () => {
             `https://tshost/prism/download/answer/csv?sessionId=${defaultSession.sessionId}&genNo=${defaultSession.genNo}&userLocale=en-us&exportFileName=data&hideCsvHeader=false`,
             expect.objectContaining({}),
         );
+
+        answerService.fetchCSVBlob('en-uk', true);
+        expect(fetchMock).toHaveBeenCalledWith(
+            `https://tshost/prism/download/answer/csv?sessionId=${defaultSession.sessionId}&genNo=${defaultSession.genNo}&userLocale=en-uk&exportFileName=data&hideCsvHeader=false`,
+            expect.objectContaining({}),
+        );
+
+        answerService.fetchCSVBlob(undefined, false);
+        expect(fetchMock).toHaveBeenCalledWith(
+            `https://tshost/prism/download/answer/csv?sessionId=${defaultSession.sessionId}&genNo=${defaultSession.genNo}&userLocale=en-us&exportFileName=data&hideCsvHeader=true`,
+            expect.objectContaining({}),
+        );
     });
 
     test('getUnderlyingDataForPoint should call the right APIs', async () => {

package/src/utils/graphql/answerService/answerService.ts

@@ -102,12 +102,16 @@ export class AnswerService {
      * @returns Response
      */
     public async fetchCSVBlob(userLocale = 'en-us', includeInfo = false): Promise<Response> {
-        const fetchUrl = `${this.thoughtSpotHost}/prism/download/answer/csv?sessionId=${this.session.sessionId}&genNo=${this.session.genNo}&userLocale=${userLocale}&exportFileName=data&hideCsvHeader=${!includeInfo}`;
+        const fetchUrl = this.getFetchCSVBlobUrl(userLocale, includeInfo);
         return fetch(fetchUrl, {
             credentials: 'include',
         });
     }
 
+    public getFetchCSVBlobUrl(userLocale = 'en-us', includeInfo = false): string {
+        return `${this.thoughtSpotHost}/prism/download/answer/csv?sessionId=${this.session.sessionId}&genNo=${this.session.genNo}&userLocale=${userLocale}&exportFileName=data&hideCsvHeader=${!includeInfo}`;
+    }
+
     /**
      * Get underlying data given a point and the output column names.
      *

package/src/utils/processData.spec.ts

@@ -7,7 +7,7 @@ import { EmbedEvent, AuthType } from '../types';
 describe('Unit test for process data', () => {
     beforeAll(() => {
         base.init({
-            thoughtSpotHost: '',
+            thoughtSpotHost: 'https://tshost',
             authType: AuthType.None,
         });
     });
@@ -153,7 +153,7 @@ describe('Unit test for process data', () => {
         base.init({
             loginFailedMessage: 'Hello',
             autoLogin: true,
-            thoughtSpotHost: '',
+            thoughtSpotHost: 'https://tshost',
             authType: AuthType.None,
         });
         jest.spyOn(base, 'getEmbedConfig').mockRestore();