fix: fixup

memori-ai · Jun 6, 2024 · 8be6b69 · 8be6b69
1 parent 8eee08a
commit 8be6b69
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/jest.setup.js b/jest.setup.js
@@ -83,3 +83,4 @@ Object.defineProperty(window, 'MathJax', {
 });
 
 jest.spyOn(window, 'MathJax', 'typesetPromise').mockResolvedValue();
+jest.spyOn(global.MathJax, 'typesetPromise').mockResolvedValue();
diff --git a/src/components/ChatBubble/ChatBubble.tsx b/src/components/ChatBubble/ChatBubble.tsx
@@ -26,9 +26,6 @@ import markedLinkifyIt from 'marked-linkify-it';
 import markedKatex from 'marked-katex-extension';
 import markedExtendedTables from '../../helpers/markedExtendedTables';
 
-// eslint-disable-next-line no-undef
-const isTest = process?.env.NODE_ENV === 'test';
-
 export interface Props {
   message: Message;
   memori: Memori;
@@ -116,10 +113,13 @@ const ChatBubble: React.FC<Props> = ({
     .replace(/(<br>)+/g, '<br>');
 
   useLayoutEffect(() => {
-    if (!isTest && typeof window !== 'undefined' && !message.fromUser) {
+    if (typeof window !== 'undefined' && !message.fromUser) {
       // @ts-ignore
       // eslint-disable-next-line no-undef
-      MathJax.typesetPromise();
+      if ('MathJax' in window && window.MathJax.typesetPromise)
+        // @ts-ignore
+        // eslint-disable-next-line no-undef
+        window.MathJax.typesetPromise();
     }
   }, [message.text, message.fromUser]);
Original file line number	Diff line number	Diff line change
Expand Up		@@ -83,3 +83,4 @@ Object.defineProperty(window, 'MathJax', {
		});

		jest.spyOn(window, 'MathJax', 'typesetPromise').mockResolvedValue();
		jest.spyOn(global.MathJax, 'typesetPromise').mockResolvedValue();