1 files changed, 103 insertions, 0 deletions
diff --git a/spec/frontend/vue_mr_widget/stores/get_state_key_spec.js b/spec/frontend/vue_mr_widget/stores/get_state_key_spec.js
new file mode 100644
index 00000000000..b356ea85cad
--- /dev/null
+++ b/spec/frontend/vue_mr_widget/stores/get_state_key_spec.js
@@ -0,0 +1,103 @@
+import getStateKey from '~/vue_merge_request_widget/stores/get_state_key';
+
+describe('getStateKey', () => {
+  it('should return proper state name', () => {
+    const context = {
+      mergeStatus: 'checked',
+      mergeWhenPipelineSucceeds: false,
+      canMerge: true,
+      onlyAllowMergeIfPipelineSucceeds: false,
+      isPipelineFailed: false,
+      hasMergeableDiscussionsState: false,
+      isPipelineBlocked: false,
+      canBeMerged: false,
+    };
+    const data = {
+      project_archived: false,
+      branch_missing: false,
+      commits_count: 2,
+      has_conflicts: false,
+      work_in_progress: false,
+    };
+    const bound = getStateKey.bind(context, data);
+
+    expect(bound()).toEqual(null);
+
+    context.canBeMerged = true;
+
+    expect(bound()).toEqual('readyToMerge');
+
+    context.canMerge = false;
+
+    expect(bound()).toEqual('notAllowedToMerge');
+
+    context.mergeWhenPipelineSucceeds = true;
+
+    expect(bound()).toEqual('mergeWhenPipelineSucceeds');
+
+    context.isSHAMismatch = true;
+
+    expect(bound()).toEqual('shaMismatch');
+
+    context.isPipelineBlocked = true;
+
+    expect(bound()).toEqual('pipelineBlocked');
+
+    context.hasMergeableDiscussionsState = true;
+
+    expect(bound()).toEqual('unresolvedDiscussions');
+
+    context.onlyAllowMergeIfPipelineSucceeds = true;
+    context.isPipelineFailed = true;
+
+    expect(bound()).toEqual('pipelineFailed');
+
+    data.work_in_progress = true;
+
+    expect(bound()).toEqual('workInProgress');
+
+    data.has_conflicts = true;
+
+    expect(bound()).toEqual('conflicts');
+
+    context.mergeStatus = 'unchecked';
+
+    expect(bound()).toEqual('checking');
+
+    data.commits_count = 0;
+
+    expect(bound()).toEqual('nothingToMerge');
+
+    data.branch_missing = true;
+
+    expect(bound()).toEqual('missingBranch');
+
+    data.project_archived = true;
+
+    expect(bound()).toEqual('archived');
+  });
+
+  it('returns rebased state key', () => {
+    const context = {
+      mergeStatus: 'checked',
+      mergeWhenPipelineSucceeds: false,
+      canMerge: true,
+      onlyAllowMergeIfPipelineSucceeds: true,
+      isPipelineFailed: true,
+      hasMergeableDiscussionsState: false,
+      isPipelineBlocked: false,
+      canBeMerged: false,
+      shouldBeRebased: true,
+    };
+    const data = {
+      project_archived: false,
+      branch_missing: false,
+      commits_count: 2,
+      has_conflicts: false,
+      work_in_progress: false,
+    };
+    const bound = getStateKey.bind(context, data);
+
+    expect(bound()).toEqual('rebase');
+  });
+});