Add more UT

Signed-off-by: Heng Qian <[email protected]>
qianheng-aws · Jul 23, 2024 · 0fb91ea · 0fb91ea
1 parent 327b68d
commit 0fb91ea
Showing 1 changed file with 44 additions and 0 deletions.
diff --git a/public/tabs/chat/messages/message_content.test.tsx b/public/tabs/chat/messages/message_content.test.tsx
@@ -77,6 +77,50 @@ describe('<MessageContent />', () => {
     expect(screen.queryAllByText('title')).toHaveLength(1);
   });
 
+  it('should display message(markdown + additionalAction)', () => {
+    render(
+      <MessageContent
+        message={{
+          type: 'output',
+          contentType: 'markdown',
+          additionalActions: [
+            {
+              actionType: 'customized_content_type',
+              message: 'customized title',
+              content: 'mock customized content',
+            },
+          ],
+          content: '# title',
+        }}
+      />
+    );
+    expect(screen.queryAllByText('title')).toHaveLength(1);
+    expect(customizedRenderMock.mock.calls[0][0]).toMatchInlineSnapshot(`
+      Object {
+        "content": "mock customized content",
+        "contentType": "customized_content_type",
+        "type": "output",
+      }
+    `);
+    expect(customizedRenderMock.mock.calls[0][1].props).toMatchInlineSnapshot(`
+      Object {
+        "message": Object {
+          "additionalActions": Array [
+            Object {
+              "actionType": "customized_content_type",
+              "content": "mock customized content",
+              "message": "customized title",
+            },
+          ],
+          "content": "# title",
+          "contentType": "markdown",
+          "type": "output",
+        },
+      }
+    `);
+    expect(customizedRenderMock.mock.calls[0][1].chatContext).not.toBeUndefined();
+  });
+
   it('should render customized render content', () => {
     render(
       <MessageContent