Fixed unit test

CMS-Enterprise · Dec 6, 2024 · 6ec34e3 · 6ec34e3
1 parent 70cdda4
commit 6ec34e3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/features/ModelPlan/ModelToOperations/_components/MilestoneCard/index.test.tsx b/src/features/ModelPlan/ModelToOperations/_components/MilestoneCard/index.test.tsx
@@ -5,6 +5,8 @@ import { render } from '@testing-library/react';
 import { MtoCommonMilestoneKey } from 'gql/generated/graphql';
 import { suggestedMilestonesMock } from 'tests/mock/mto';
 
+import MessageProvider from 'contexts/MessageContext';
+
 import { MilestoneCardType } from '../../MilestoneLibrary';
 
 import MilestoneCard from './index';
@@ -30,7 +32,9 @@ describe('MilestoneCard Component', () => {
           ]}
         >
           <Route path="/models/:modelID/collaboration-area/model-to-operations/milestone-library">
-            <MilestoneCard milestone={mockMilestone} />
+            <MessageProvider>
+              <MilestoneCard milestone={mockMilestone} />
+            </MessageProvider>
           </Route>
         </MemoryRouter>
       </MockedProvider>