test: new tests added for configurable survey

Author: marslanabdulrauf

src/containers/UnenrollConfirmModal/hooks/index.js

@@ -3,9 +3,9 @@ import React from 'react';
 import { StrictDict } from 'utils';
 import { apiHooks } from 'hooks';
 
+import { configuration } from 'config';
 import { useUnenrollReasons } from './reasons';
 import * as module from '.';
-import { configuration } from 'config';
 
 export const state = StrictDict({
   confirmed: (val) => React.useState(val), // eslint-disable-line

src/containers/UnenrollConfirmModal/hooks/index.test.js

@@ -1,5 +1,6 @@
 import { apiHooks } from 'hooks';
 import { MockUseState } from 'testUtils';
+import { configuration } from 'config';
 
 import * as reasons from './reasons';
 import * as hooks from '.';
@@ -11,13 +12,22 @@ jest.mock('./reasons', () => ({
 jest.mock('hooks', () => ({
   apiHooks: {
     useInitializeApp: jest.fn(),
+    useUnenrollFromCourse: jest.fn(),
+  },
+}));
+
+jest.mock('config', () => ({
+  configuration: {
+    SHOW_UNENROLL_SURVEY: true,
   },
 }));
 
 const state = new MockUseState(hooks);
 const testValue = 'test-value';
 const initializeApp = jest.fn();
+const unenrollFromCourse = jest.fn();
 apiHooks.useInitializeApp.mockReturnValue(initializeApp);
+apiHooks.useUnenrollFromCourse.mockReturnValue(unenrollFromCourse);
 let out;
 
 const mockReason = {
@@ -79,22 +89,66 @@ describe('UnenrollConfirmModal hooks', () => {
         expect(initializeApp).toHaveBeenCalled();
       });
     });
-    describe('modalState', () => {
-      it('returns modalStates.finished if confirmed and submitted', () => {
+  });
+
+  describe('SHOW_UNENROLL_SURVEY configuration tests', () => {
+    beforeEach(() => {
+      state.mock();
+      jest.clearAllMocks();
+    });
+    afterEach(() => {
+      state.restore();
+    });
+
+    describe('when SHOW_UNENROLL_SURVEY is true (default)', () => {
+      beforeEach(() => {
+        configuration.SHOW_UNENROLL_SURVEY = true;
+      });
+
+      test('confirm does not call unenrollFromCourse immediately', () => {
+        out = createUseUnenrollData();
+        out.confirm();
+        expect(unenrollFromCourse).not.toHaveBeenCalled();
+        expect(state.setState.confirmed).toHaveBeenCalledWith(true);
+      });
+
+      test('modalState returns reason when confirmed but not submitted', () => {
+        state.mockVal(state.keys.confirmed, true);
+        reasons.useUnenrollReasons.mockReturnValueOnce({ ...mockReason, isSubmitted: false });
+        out = createUseUnenrollData();
+        expect(out.modalState).toEqual(hooks.modalStates.reason);
+      });
+
+      test('modalState returns finished when confirmed and submitted', () => {
         state.mockVal(state.keys.confirmed, true);
         reasons.useUnenrollReasons.mockReturnValueOnce({ ...mockReason, isSubmitted: true });
         out = createUseUnenrollData();
         expect(out.modalState).toEqual(hooks.modalStates.finished);
       });
-      it('returns modalStates.reason if confirmed and not submitted', () => {
-        state.mockVal(state.keys.confirmed, true);
+    });
+
+    describe('when SHOW_UNENROLL_SURVEY is false', () => {
+      beforeEach(() => {
+        configuration.SHOW_UNENROLL_SURVEY = false;
+      });
+
+      afterEach(() => {
+        // Reset to default
+        configuration.SHOW_UNENROLL_SURVEY = true;
+      });
+
+      test('confirm calls unenrollFromCourse immediately', () => {
         out = createUseUnenrollData();
-        expect(out.modalState).toEqual(hooks.modalStates.reason);
+        out.confirm();
+        expect(unenrollFromCourse).toHaveBeenCalled();
+        expect(state.setState.confirmed).toHaveBeenCalledWith(true);
       });
-      it('returns modalStates.confirm if not confirmed', () => {
-        state.mockVal(state.keys.confirmed, false);
+
+      test('modalState returns finished when confirmed regardless of submission status', () => {
+        state.mockVal(state.keys.confirmed, true);
+        reasons.useUnenrollReasons.mockReturnValueOnce({ ...mockReason, isSubmitted: false });
         out = createUseUnenrollData();
-        expect(out.modalState).toEqual(hooks.modalStates.confirm);
+        expect(out.modalState).toEqual(hooks.modalStates.finished);
       });
     });
   });