1 files changed, 4 insertions, 0 deletions
diff --git a/app/assets/javascripts/blob/components/blob_header_viewer_switcher.vue b/app/assets/javascripts/blob/components/blob_header_viewer_switcher.vue
index a5b594fbd88..b2546d47694 100644
--- a/app/assets/javascripts/blob/components/blob_header_viewer_switcher.vue
+++ b/app/assets/javascripts/blob/components/blob_header_viewer_switcher.vue
@@ -53,6 +53,8 @@ export default {
       icon="code"
       category="primary"
       variant="default"
+      class="js-blob-viewer-switch-btn"
+      data-viewer="simple"
       @click="switchToViewer($options.SIMPLE_BLOB_VIEWER)"
     />
     <gl-button
@@ -63,6 +65,8 @@ export default {
       icon="document"
       category="primary"
       variant="default"
+      class="js-blob-viewer-switch-btn"
+      data-viewer="rich"
       @click="switchToViewer($options.RICH_BLOB_VIEWER)"
     />
   </gl-button-group>