1 files changed, 74 insertions, 25 deletions

diff --git a/spec/frontend/diffs/store/getters_versions_dropdowns_spec.js b/spec/frontend/diffs/store/getters_versions_dropdowns_spec.js
index 3e5ba66d5e4..0343ef75732 100644
--- a/spec/frontend/diffs/store/getters_versions_dropdowns_spec.js
+++ b/spec/frontend/diffs/store/getters_versions_dropdowns_spec.js
@@ -1,6 +1,9 @@
 import * as getters from '~/diffs/store/getters';
 import state from '~/diffs/store/modules/diff_state';
-import { DIFF_COMPARE_BASE_VERSION_INDEX } from '~/diffs/constants';
+import {
+  DIFF_COMPARE_BASE_VERSION_INDEX,
+  DIFF_COMPARE_HEAD_VERSION_INDEX,
+} from '~/diffs/constants';
 import diffsMockData from '../mock_data/merge_request_diffs';
 
 describe('Compare diff version dropdowns', () => {
@@ -37,47 +40,93 @@ describe('Compare diff version dropdowns', () => {
 
   describe('diffCompareDropdownTargetVersions', () => {
     // diffCompareDropdownTargetVersions slices the array at the first position
-    // and appends a "base" version which is why we use diffsMockData[1] below
-    // This is to display "base" at the end of the target dropdown
-    const expectedFirstVersion = {
-      ...diffsMockData[1],
-      href: expect.any(String),
-      versionName: expect.any(String),
+    // and appends a "base" and "head" version at the end of the list so that
+    // "base" and "head" appear at the bottom of the dropdown
+    // this is also why we use diffsMockData[1] for the "first" version
+
+    let expectedFirstVersion;
+    let expectedBaseVersion;
+    let expectedHeadVersion;
+    const originalLocation = window.location;
+
+    const setupTest = includeDiffHeadParam => {
+      const diffHeadParam = includeDiffHeadParam ? '?diff_head=true' : '';
+
+      Object.defineProperty(window, 'location', {
+        writable: true,
+        value: { href: `https://example.gitlab.com${diffHeadParam}` },
+      });
+
+      expectedFirstVersion = {
+        ...diffsMockData[1],
+        href: expect.any(String),
+        versionName: expect.any(String),
+        selected: false,
+      };
+
+      expectedBaseVersion = {
+        versionName: 'baseVersion',
+        version_index: DIFF_COMPARE_BASE_VERSION_INDEX,
+        href: 'basePath',
+        isBase: true,
+        selected: false,
+      };
+
+      expectedHeadVersion = {
+        versionName: 'baseVersion',
+        version_index: DIFF_COMPARE_HEAD_VERSION_INDEX,
+        href: 'headPath',
+        isHead: true,
+        selected: false,
+      };
     };
 
-    const expectedBaseVersion = {
-      versionName: 'baseVersion',
-      version_index: DIFF_COMPARE_BASE_VERSION_INDEX,
-      href: 'basePath',
-      isBase: true,
+    const assertVersions = targetVersions => {
+      // base and head should be the last two versions in that order
+      const targetBaseVersion = targetVersions[targetVersions.length - 2];
+      const targetHeadVersion = targetVersions[targetVersions.length - 1];
+      expect(targetVersions[0]).toEqual(expectedFirstVersion);
+      expect(targetBaseVersion).toEqual(expectedBaseVersion);
+      expect(targetHeadVersion).toEqual(expectedHeadVersion);
     };
 
+    afterEach(() => {
+      window.location = originalLocation;
+    });
+
     it('base version selected', () => {
-      expectedFirstVersion.selected = false;
+      setupTest();
       expectedBaseVersion.selected = true;
 
-      const targetVersions = getters.diffCompareDropdownTargetVersions(localState, {
-        selectedTargetIndex: DIFF_COMPARE_BASE_VERSION_INDEX,
-      });
+      const targetVersions = getters.diffCompareDropdownTargetVersions(localState, getters);
+      assertVersions(targetVersions);
+    });
 
-      const lastVersion = targetVersions[targetVersions.length - 1];
-      expect(targetVersions[0]).toEqual(expectedFirstVersion);
-      expect(lastVersion).toEqual(expectedBaseVersion);
+    it('head version selected', () => {
+      setupTest(true);
+
+      expectedHeadVersion.selected = true;
+
+      const targetVersions = getters.diffCompareDropdownTargetVersions(localState, getters);
+      assertVersions(targetVersions);
     });
 
     it('first version selected', () => {
-      expectedFirstVersion.selected = true;
-      expectedBaseVersion.selected = false;
+      // NOTE: It should not be possible to have both "diff_head=true" and
+      // have anything other than the head version selected, but the user could
+      // manually add "?diff_head=true" to the url. In this instance we still
+      // want the actual selected version to display as "selected"
+      // Passing in "true" here asserts that first version is still selected
+      // even if "diff_head" is present in the url
+      setupTest(true);
 
+      expectedFirstVersion.selected = true;
       localState.startVersion = expectedFirstVersion;
 
       const targetVersions = getters.diffCompareDropdownTargetVersions(localState, {
         selectedTargetIndex: expectedFirstVersion.version_index,
       });
-
-      const lastVersion = targetVersions[targetVersions.length - 1];
-      expect(targetVersions[0]).toEqual(expectedFirstVersion);
-      expect(lastVersion).toEqual(expectedBaseVersion);
+      assertVersions(targetVersions);
     });
   });