1 files changed, 20 insertions, 1 deletions
diff --git a/spec/frontend/work_items/utils_spec.js b/spec/frontend/work_items/utils_spec.js
index aa24b80cf08..b8af5f10a5a 100644
--- a/spec/frontend/work_items/utils_spec.js
+++ b/spec/frontend/work_items/utils_spec.js
@@ -1,4 +1,9 @@
-import { autocompleteDataSources, markdownPreviewPath } from '~/work_items/utils';
+import {
+  autocompleteDataSources,
+  markdownPreviewPath,
+  getWorkItemTodoOptimisticResponse,
+} from '~/work_items/utils';
+import { workItemResponseFactory } from './mock_data';
 
 describe('autocompleteDataSources', () => {
   beforeEach(() => {
@@ -25,3 +30,17 @@ describe('markdownPreviewPath', () => {
     );
   });
 });
+
+describe('getWorkItemTodoOptimisticResponse', () => {
+  it.each`
+    scenario     | pendingTodo | result
+    ${'empty'}   | ${false}    | ${0}
+    ${'present'} | ${true}     | ${1}
+  `('returns correct response when pending item list is $scenario', ({ pendingTodo, result }) => {
+    const workItem = workItemResponseFactory({ canUpdate: true });
+    expect(
+      getWorkItemTodoOptimisticResponse({ workItem, pendingTodo }).workItemUpdate.workItem
+        .widgets[0].currentUserTodos.edges.length,
+    ).toBe(result);
+  });
+});