Fixed Download Button Test

1 files changed, 4 insertions, 3 deletions

diff --git a/spec/javascripts/vue_shared/components/content_viewer/content_viewer_spec.js b/spec/javascripts/vue_shared/components/content_viewer/content_viewer_spec.js
index a7b14048fc4..383f0cd29ea 100644
--- a/spec/javascripts/vue_shared/components/content_viewer/content_viewer_spec.js
+++ b/spec/javascripts/vue_shared/components/content_viewer/content_viewer_spec.js
@@ -58,10 +58,11 @@ describe('ContentViewer', () => {
       fileSize: 1024,
     });
 
-    const previewContainer = vm.$el.querySelector('.btn.btn-default');
-
     setTimeout(() => {
-      expect(previewContainer.textContent.trim()).toContain('Download test.abc (1.00 KiB)');
+      expect(vm.$el.querySelector('.file-info').textContent.trim()).toContain(
+        'test.abc (1.00 KiB)',
+      );
+      expect(vm.$el.querySelector('.btn.btn-default').textContent.trim()).toContain('Download');
 
       done();
     });