diff --git a/src/lib/viewers/error/PreviewErrorViewer.js b/src/lib/viewers/error/PreviewErrorViewer.js index 097376160..955876b07 100644 --- a/src/lib/viewers/error/PreviewErrorViewer.js +++ b/src/lib/viewers/error/PreviewErrorViewer.js @@ -80,8 +80,7 @@ class PreviewErrorViewer extends BaseViewer { const { displayMessage, details, message } = error; const { file } = this.options; - const { extension } = file || {}; - const iconExtension = extension === 'flv' || extension === 'tgz' || extension === 'zip' ? extension : undefined; + const { extension: iconExtension } = file || {}; // Display the default or file-specific error icon this.errorIcon = new ErrorIcon({ containerEl: this.iconEl }); diff --git a/src/lib/viewers/error/__tests__/PreviewErrorViewer-test.js b/src/lib/viewers/error/__tests__/PreviewErrorViewer-test.js index c1c282d8d..919a93d19 100644 --- a/src/lib/viewers/error/__tests__/PreviewErrorViewer-test.js +++ b/src/lib/viewers/error/__tests__/PreviewErrorViewer-test.js @@ -55,7 +55,8 @@ describe('lib/viewers/error/PreviewErrorViewer', () => { ['zip', 'zip'], ['tgz', 'tgz'], ['flv', 'flv'], - ['blah', undefined], + ['pdf', 'pdf'], + ['pptx', 'pptx'], ])('should set appropriate icon', (fileExtension, iconExtension) => { const err = new PreviewError('some_code'); error.options.file.extension = fileExtension;