1 files changed, 4 insertions, 0 deletions
diff --git a/spec/javascripts/vue_shared/components/diff_viewer/diff_viewer_spec.js b/spec/javascripts/vue_shared/components/diff_viewer/diff_viewer_spec.js
index 6add6cdac4d..660eaddf01f 100644
--- a/spec/javascripts/vue_shared/components/diff_viewer/diff_viewer_spec.js
+++ b/spec/javascripts/vue_shared/components/diff_viewer/diff_viewer_spec.js
@@ -22,6 +22,7 @@ describe('DiffViewer', () => {
 
     createComponent({
       diffMode: 'replaced',
+      diffViewerMode: 'image',
       newPath: GREEN_BOX_IMAGE_URL,
       newSha: 'ABC',
       oldPath: RED_BOX_IMAGE_URL,
@@ -45,6 +46,7 @@ describe('DiffViewer', () => {
   it('renders fallback download diff display', done => {
     createComponent({
       diffMode: 'replaced',
+      diffViewerMode: 'added',
       newPath: 'test.abc',
       newSha: 'ABC',
       oldPath: 'testold.abc',
@@ -72,6 +74,7 @@ describe('DiffViewer', () => {
   it('renders renamed component', () => {
     createComponent({
       diffMode: 'renamed',
+      diffViewerMode: 'renamed',
       newPath: 'test.abc',
       newSha: 'ABC',
       oldPath: 'testold.abc',
@@ -84,6 +87,7 @@ describe('DiffViewer', () => {
   it('renders mode changed component', () => {
     createComponent({
       diffMode: 'mode_changed',
+      diffViewerMode: 'image',
       newPath: 'test.abc',
       newSha: 'ABC',
       oldPath: 'testold.abc',