From 992c1ff8d8eee7608b0656104952f65ef245ebb0 Mon Sep 17 00:00:00 2001
From: Tim Zallmann <tzallmann@gitlab.com>
Date: Thu, 5 Apr 2018 22:32:39 +0200
Subject: Fixed Download Button Test

---
 .../vue_shared/components/content_viewer/content_viewer_spec.js    | 7 ++++---
 1 file 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();
     });
-- 
cgit v1.2.1