Add test cases for focus trap options

react-responsive-modal/__tests__/index.test.tsx

@@ -1,5 +1,5 @@
 import * as React from 'react';
-import { render, fireEvent, waitFor } from '@testing-library/react';
+import { fireEvent, render, waitFor } from '@testing-library/react';
 import { Modal } from '../src';
 
 describe('modal', () => {
@@ -441,6 +441,55 @@ describe('modal', () => {
     });
   });
 
+  describe('prop: focusTrapOptions', () => {
+    describe('default settings', () => {
+      it('should focus first tabbable element', () => {
+        const { queryByTestId } = render(
+          <Modal open onClose={() => null}>
+            <a href="https://google.com" data-testid="first-item">
+              modal content
+            </a>
+          </Modal>
+        );
+        expect(queryByTestId('first-item')).toHaveFocus();
+      });
+    });
+
+    describe('when focusOn is firstFocusableElement', () => {
+      it('should focus first tabbable element', () => {
+        const { queryByTestId } = render(
+          <Modal
+            open
+            onClose={() => null}
+            focusTrapOptions={{ focusOn: 'firstFocusableElement' }}
+          >
+            <a href="https://google.com" data-testid="first-item">
+              modal content
+            </a>
+          </Modal>
+        );
+        expect(queryByTestId('first-item')).toHaveFocus();
+      });
+    });
+
+    describe('when focusOn is modalRoot', () => {
+      it('should focus modal root', () => {
+        const { queryByTestId } = render(
+          <Modal
+            open
+            onClose={() => null}
+            focusTrapOptions={{ focusOn: 'modalRoot' }}
+          >
+            <a href="https://google.com" data-testid="first-item">
+              modal content
+            </a>
+          </Modal>
+        );
+        expect(queryByTestId('modal')).toHaveFocus();
+      });
+    });
+  });
+
   describe('prop: onEscKeyDown', () => {
     it('should be called when esc key is pressed', async () => {
       const onEscKeyDown = jest.fn();