Merge pull request #587 from microsoft/fix-tests

fix: Frontend tests

Author: Roopan-Microsoft

src/frontend/src/components/Answer/Answer.test.tsx

@@ -73,7 +73,7 @@ const mockCitations: Citation[] = [
     }
 ]
 let mockAnswerProps: AskResponse = {
-    answer: 'This is an example answer with citations [doc1] and [doc2] and [doc3].',
+    answer: 'This is an example answer with citations [1] and [2] and [3].',
     message_id: '123',
     feedback: Feedback.Neutral,
     citations: cloneDeep(mockCitations)
@@ -133,7 +133,7 @@ describe('Answer Component', () => {
 
         const answerWithMissingFeedback = {
            ...mockAnswerProps,
-           answer: 'This is an example answer with citations [doc1]',
+           answer: 'This is an example answer with citations [1]',
            citations: [mockCitations[0]] // Only include one citation
         }
 

src/frontend/src/components/Answer/AnswerParser.test.ts

@@ -41,7 +41,7 @@ const sampleCitations: Citation[] = [
 ]
 
 const sampleAnswer: AskResponse = {
-  answer: 'This is an example answer with citations [doc1] and [doc2].',
+  answer: 'This is an example answer with citations [1] and [2].',
   citations: cloneDeep(sampleCitations)
 }
 
@@ -57,13 +57,13 @@ describe('enumerateCitations', () => {
 describe('parseAnswer', () => {
   it('reformats the answer text and reindexes citations', () => {
     const parsed: ParsedAnswer = parseAnswer(sampleAnswer)
-    expect(parsed.markdownFormatText).toBe('This is an example answer with citations [doc1] and [doc2].')
-    expect(parsed.citations.length).toBe(3)
+    expect(parsed.markdownFormatText).toBe('This is an example answer with citations  ^1^  and  ^2^ .')
+    expect(parsed.citations.length).toBe(2)
     expect(parsed.citations[0].id).toBe('doc1')
-    // expect(parsed.citations[0].reindex_id).toBe('1')
-    // expect(parsed.citations[1].id).toBe('2')
-    // // expect(parsed.citations[1].reindex_id).toBe('2')
-    // expect(parsed.citations[0].part_index).toBe(1)
-    // expect(parsed.citations[1].part_index).toBe(2)
+    expect(parsed.citations[0].reindex_id).toBe('1')
+    expect(parsed.citations[1].id).toBe('doc2')
+    expect(parsed.citations[1].reindex_id).toBe('2')
+    expect(parsed.citations[0].part_index).toBe(1)
+    expect(parsed.citations[1].part_index).toBe(2)
   })
 })

src/frontend/src/components/DraftCards/SectionCard.test.tsx

@@ -1,13 +1,13 @@
 import React from 'react'
 import { render, screen, fireEvent, waitFor } from '@testing-library/react'
+import { act } from 'react'
 import '@testing-library/jest-dom'
 import SectionCard from './SectionCard'
 import { AppStateContext } from '../../state/AppProvider'
 import { sectionGenerate } from '../../api'
 import { MemoryRouter } from 'react-router-dom'
 
 import { ChatHistoryLoadingState } from '../../api/models'
-import { act } from 'react-dom/test-utils'
 
 import {defaultMockState} from '../../test/test.utils';
 
@@ -77,17 +77,27 @@ describe('SectionCard Component', () => {
   })
 
   it('When context not available throws an error', async () => {
+    // Suppress console.error for this test to avoid cluttering test output
+    const consoleSpy = jest.spyOn(console, 'error').mockImplementation(() => {})
+    
     expect(() =>
       render(
         <MemoryRouter>
           <SectionCard sectionIdx={0} />
         </MemoryRouter>
       )
     ).toThrow('useAppState must be used within a AppStateProvider')
+    
+    consoleSpy.mockRestore()
   })
 
   it('When no section available in context throws an error', async () => {
+    // Suppress console.error for this test to avoid cluttering test output
+    const consoleSpy = jest.spyOn(console, 'error').mockImplementation(() => {})
+    
     expect(() => renderWithContext(2)).toThrow('Section not found')
+    
+    consoleSpy.mockRestore()
   })
 
   it('renders section title and description', async () => {