diff --git a/src/app/services/actions/query-action-creator-util.ts b/src/app/services/actions/query-action-creator-util.ts
index cce501f7a..5c497e040 100644
--- a/src/app/services/actions/query-action-creator-util.ts
+++ b/src/app/services/actions/query-action-creator-util.ts
@@ -53,7 +53,7 @@ export function isImageResponse(contentType: string) {
if (!contentType) { return false; }
return (
contentType === 'application/octet-stream' ||
- contentType.substr(0, 6) === 'image/'
+ contentType.includes('image/')
);
}
diff --git a/src/app/services/actions/query-action-creators.ts b/src/app/services/actions/query-action-creators.ts
index b28d31245..dd9dffcb5 100644
--- a/src/app/services/actions/query-action-creators.ts
+++ b/src/app/services/actions/query-action-creators.ts
@@ -1,9 +1,13 @@
import { MessageBarType } from 'office-ui-fabric-react';
+
import { ContentType } from '../../../types/enums';
import { IHistoryItem } from '../../../types/history';
import { IQuery } from '../../../types/query-runner';
import { writeHistoryData } from '../../views/sidebar/history/history-utils';
-import { anonymousRequest, authenticatedRequest, parseResponse, queryResponse } from './query-action-creator-util';
+import {
+ anonymousRequest, authenticatedRequest,
+ isImageResponse, parseResponse, queryResponse
+} from './query-action-creator-util';
import { setQueryResponseStatus } from './query-status-action-creator';
import { addHistoryItem } from './request-history-action-creators';
@@ -83,7 +87,7 @@ async function createHistory(response: Response, respHeaders: any, query: IQuery
const responseHeaders = { ...respHeaders };
const contentType = respHeaders['content-type'];
- if (contentType === ContentType.Image) {
+ if (isImageResponse(contentType)) {
result = {
message: 'Run the query to view the image'
};
diff --git a/src/app/views/query-response/pivot-items/pivot-items.tsx b/src/app/views/query-response/pivot-items/pivot-items.tsx
index bb986fef6..bd3dbbb9a 100644
--- a/src/app/views/query-response/pivot-items/pivot-items.tsx
+++ b/src/app/views/query-response/pivot-items/pivot-items.tsx
@@ -4,6 +4,7 @@ import React from 'react';
import { ThemeContext } from '../../../../themes/theme-context';
import { ContentType, Mode } from '../../../../types/enums';
import { IQuery } from '../../../../types/query-runner';
+import { isImageResponse } from '../../../services/actions/query-action-creator-util';
import { lookupTemplate } from '../../../utils/adaptive-cards-lookup';
import { Image, Monaco } from '../../common';
import { genericCopy } from '../../common/copy';
@@ -112,14 +113,14 @@ function displayResultComponent(headers: any, body: any, verb: string) {
case ContentType.XML:
return ;
- case ContentType.Image:
- return ;
-
default:
+ if (isImageResponse(contentType)) {
+ return ;
+ }
return ;
}
}