even more testing

Author: Veikkosuhonen

e2e/chat.spec.ts

@@ -48,7 +48,7 @@ testMatrix.forEach((testConfig) => {
         await useMockModel(page)
       })
 
-      test('Chat v2 mock response works', async ({ page }) => {
+      test('One message works', async ({ page }) => {
         await acceptDisclaimer(page)
         await useMockModel(page)
 
@@ -60,6 +60,28 @@ testMatrix.forEach((testConfig) => {
         await expect(page.getByTestId('assistant-message')).toContainText('You are calling mock endpoint for streaming mock data')
       })
 
+      test('Multiple messages work', async ({ page }) => {
+        await acceptDisclaimer(page)
+        await useMockModel(page)
+
+        await sendChatMessage(page, 'say perkele')
+
+        await closeSendPreference(page)
+
+        await expect(page.getByTestId('user-message').first()).toContainText('say perkele')
+        await expect(page.getByTestId('assistant-message').first()).toContainText('perkele')
+
+        await sendChatMessage(page, 'say minttuista')
+
+        await expect(page.getByTestId('user-message').nth(1)).toContainText('say minttuista')
+        await expect(page.getByTestId('assistant-message').nth(1)).toContainText('minttuista')
+
+        await sendChatMessage(page, 'say settiä')
+
+        await expect(page.getByTestId('user-message').nth(2)).toContainText('say settiä')
+        await expect(page.getByTestId('assistant-message').nth(2)).toContainText('settiä')
+      })
+
       test('Can empty conversation', async ({ page }) => {
         await acceptDisclaimer(page)
         await useMockModel(page)
@@ -78,6 +100,22 @@ testMatrix.forEach((testConfig) => {
         await expect(page.getByTestId('assistant-message')).not.toBeVisible()
       })
 
+      test('Can save as email', async ({ page }) => {
+        await acceptDisclaimer(page)
+        await useMockModel(page)
+
+        await sendChatMessage(page, 'tää tyhjennetään')
+
+        await closeSendPreference(page)
+
+        await expect(page.getByTestId('user-message')).toContainText('tää tyhjennetään')
+        await expect(page.getByTestId('assistant-message')).toContainText('OVER', { timeout: 6000 })
+
+        await page.getByTestId('email-button').click()
+
+        await expect(page.getByText('Email sent')).toBeVisible()
+      })
+
       test('Custom system prompt can be changed', async ({ page }) => {
         await acceptDisclaimer(page)
         await useMockModel(page)

src/client/components/ChatV2/EmailButton.tsx

@@ -195,7 +195,7 @@ const EmailButton = ({ messages, disabled }: { messages: Message[]; disabled: bo
 
   return (
     <Tooltip arrow placement="right" title={t('chat:email', { email: user.email })}>
-      <OutlineButtonBlack startIcon={<EmailIcon />} onClick={handleSend}>
+      <OutlineButtonBlack startIcon={<EmailIcon />} onClick={handleSend} data-testid="email-button">
         {t('email:save')}
       </OutlineButtonBlack>
     </Tooltip>

src/server/services/langchain/MockModel.ts

@@ -31,6 +31,9 @@ export class MockModel extends FakeStreamingChatModel {
     } else if (firstSystemMessage && (firstSystemMessage.content as string).startsWith('mocktest')) {
       // testing a system message
       // Do nothing. FakeStreamingChatModel echoes the first message.
+    } else if (lastHumanMessage.startsWith('say ')) {
+      const msg = lastHumanMessage.replace('say ', '')
+      this.chunks = [new AIMessageChunk(msg)]
     } else if (lastHumanMessage.startsWith('rag')) {
       // Do a tool call
       this.chunks = toolCallChunks