Enrich test case of hiding remove choice button

leung018 · May 27, 2024 · c02540a · c02540a
1 parent 2a50ff7
commit c02540a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/app/components/editor.test.tsx b/src/app/components/editor.test.tsx
@@ -111,6 +111,11 @@ class UIServiceInteractor {
     )
   }
 
+  clickRemoveChoice({ choiceNumber }: { choiceNumber: number }) {
+    fireEvent.click(this.queryRemoveChoiceButton({ choiceNumber })!)
+    return this
+  }
+
   clickAddChoice() {
     const addChoiceButtons = screen.getAllByText('Add Choice')
     fireEvent.click(addChoiceButtons[this.questionNumberFocus - 1])
@@ -581,6 +586,14 @@ describe('QuestionSetEditor', () => {
     interactor.setQuestionNumberFocus(1)
     expect(interactor.queryRemoveChoiceButton({ choiceNumber: 1 })).toBeNull()
     expect(interactor.queryRemoveChoiceButton({ choiceNumber: 2 })).toBeNull()
+
+    interactor.clickAddChoice()
+    interactor.clickRemoveChoice({ choiceNumber: 1 })
+
+    //: TODO: Uncomment below when action that is triggered by remove choice button is implemented
+    // expect(interactor.queryRemoveChoiceButton({ choiceNumber: 1 })).toBeNull()
+    // expect(interactor.queryRemoveChoiceButton({ choiceNumber: 2 })).toBeNull()
+    // expect(interactor.queryRemoveChoiceButton({ choiceNumber: 3 })).toBeNull()
   })
 })