From 59a9eb3835006745be62342b72a3baf292c9a380 Mon Sep 17 00:00:00 2001
From: Matvei Andrienko <m.y.andrienko@outlook.com>
Date: Thu, 1 Feb 2024 15:25:48 +0100
Subject: [PATCH] feat: add bounce modal tests

---
 .../Message/__tests__/MessageSimple.test.js   | 105 +++++++++++++++++-
 1 file changed, 101 insertions(+), 4 deletions(-)

diff --git a/src/components/Message/__tests__/MessageSimple.test.js b/src/components/Message/__tests__/MessageSimple.test.js
index ce201b14a..599422fb2 100644
--- a/src/components/Message/__tests__/MessageSimple.test.js
+++ b/src/components/Message/__tests__/MessageSimple.test.js
@@ -36,6 +36,7 @@ import {
   generateUser,
   getTestClientWithUser,
 } from '../../../mock-builders';
+import { MessageBounceOptions } from '../../MessageBounce';
 
 Dayjs.extend(calendar);
 
@@ -44,8 +45,8 @@ jest.mock('../MessageText', () => ({ MessageText: jest.fn(() => <div />) }));
 jest.mock('../../MML', () => ({ MML: jest.fn(() => <div />) }));
 jest.mock('../../Avatar', () => ({ Avatar: jest.fn(() => <div />) }));
 jest.mock('../../MessageInput', () => ({
-  EditMessageForm: jest.fn(() => <div />),
-  MessageInput: jest.fn(() => <div />),
+  EditMessageForm: jest.fn(() => <div data-testid='edit-message-form' />),
+  MessageInput: jest.fn(() => <div data-testid='message-input' />),
 }));
 jest.mock('../../Modal', () => ({ Modal: jest.fn((props) => <div>{props.children}</div>) }));
 
@@ -55,6 +56,7 @@ const carol = generateUser();
 const openThreadMock = jest.fn();
 const tDateTimeParserMock = jest.fn((date) => Dayjs(date));
 const retrySendMessageMock = jest.fn();
+const removeMessageMock = jest.fn();
 
 async function renderMessageSimple({
   message,
@@ -63,6 +65,7 @@ async function renderMessageSimple({
   channelCapabilities = { 'send-reaction': true },
   components = {},
   renderer = render,
+  themeVersion = '1',
 }) {
   const channel = generateChannel({
     getConfig: () => channelConfigOverrides,
@@ -73,10 +76,14 @@ async function renderMessageSimple({
   const client = await getTestClientWithUser(alice);
 
   return renderer(
-    <ChatProvider value={{ client, themeVersion: '1' }}>
+    <ChatProvider value={{ client, themeVersion }}>
       <ChannelStateProvider value={{ channel, channelCapabilities, channelConfig }}>
         <ChannelActionProvider
-          value={{ openThread: openThreadMock, retrySendMessage: retrySendMessageMock }}
+          value={{
+            openThread: openThreadMock,
+            removeMessage: removeMessageMock,
+            retrySendMessage: retrySendMessageMock,
+          }}
         >
           <TranslationProvider value={{ t: (key) => key, tDateTimeParser: tDateTimeParserMock }}>
             <ComponentProvider
@@ -636,4 +643,94 @@ describe('<MessageSimple />', () => {
     const results = await axe(container);
     expect(results).toHaveNoViolations();
   });
+
+  describe('bounced message', () => {
+    const bouncedMessageOptions = {
+      moderation_details: {
+        action: 'MESSAGE_RESPONSE_ACTION_BOUNCE',
+      },
+      type: 'error',
+    };
+
+    it('should render error badge for bounced messages', async () => {
+      const message = generateAliceMessage(bouncedMessageOptions);
+      const { queryByTestId } = await renderMessageSimple({ message, themeVersion: '2' });
+      expect(queryByTestId('error')).toBeInTheDocument();
+    });
+
+    it('should render open bounce modal on click', async () => {
+      const message = generateAliceMessage(bouncedMessageOptions);
+      const { getByTestId, queryByTestId } = await renderMessageSimple({ message });
+      fireEvent.click(getByTestId('message-inner'));
+      expect(queryByTestId('message-bounce-options')).toBeInTheDocument();
+    });
+
+    it('should switch to message editing', async () => {
+      const message = generateAliceMessage(bouncedMessageOptions);
+      const { getByTestId, queryByTestId } = await renderMessageSimple({
+        message,
+      });
+      fireEvent.click(getByTestId('message-inner'));
+      fireEvent.click(getByTestId('message-bounce-edit'));
+      expect(queryByTestId('message-input')).toBeInTheDocument();
+    });
+
+    it('should retry sending message', async () => {
+      const message = generateAliceMessage(bouncedMessageOptions);
+      const { getByTestId } = await renderMessageSimple({
+        message,
+      });
+      fireEvent.click(getByTestId('message-inner'));
+      fireEvent.click(getByTestId('message-bounce-send'));
+      expect(retrySendMessageMock).toHaveBeenCalledWith(
+        expect.objectContaining({
+          id: message.id,
+        }),
+      );
+    });
+
+    it('should remove message', async () => {
+      const message = generateAliceMessage(bouncedMessageOptions);
+      const { getByTestId } = await renderMessageSimple({
+        message,
+      });
+      fireEvent.click(getByTestId('message-inner'));
+      fireEvent.click(getByTestId('message-bounce-delete'));
+      expect(removeMessageMock).toHaveBeenCalledWith(
+        expect.objectContaining({
+          id: message.id,
+        }),
+      );
+    });
+
+    it('should use overriden modal content component', async () => {
+      const message = generateAliceMessage(bouncedMessageOptions);
+      const CustomMessageBounceOptions = () => (
+        <div data-testid='custom-message-bounce-options'>Overriden</div>
+      );
+      const { getByTestId, queryByTestId } = await renderMessageSimple({
+        components: {
+          MessageBounceOptions: CustomMessageBounceOptions,
+        },
+        message,
+      });
+      fireEvent.click(getByTestId('message-inner'));
+      expect(queryByTestId('custom-message-bounce-options')).toBeInTheDocument();
+    });
+
+    it('should use overriden modal content text', async () => {
+      const message = generateAliceMessage(bouncedMessageOptions);
+      const CustomMessageBounceOptions = () => (
+        <MessageBounceOptions>Overriden</MessageBounceOptions>
+      );
+      const { getByTestId, queryByText } = await renderMessageSimple({
+        components: {
+          MessageBounceOptions: CustomMessageBounceOptions,
+        },
+        message,
+      });
+      fireEvent.click(getByTestId('message-inner'));
+      expect(queryByText('Overriden')).toBeInTheDocument();
+    });
+  });
 });