Fix test regressions

onaio · Jan 27, 2025 · fb6b716 · fb6b716
1 parent 6ade1ec
commit fb6b716
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/packages/fhir-flag/src/components/CloseFlagForm/tests/index.test.tsx b/packages/fhir-flag/src/components/CloseFlagForm/tests/index.test.tsx
@@ -5,6 +5,7 @@ import { flag } from '../../Utils/tests/fixtures';
 import { QueryClient, QueryClientProvider } from 'react-query';
 import userEvents from '@testing-library/user-event';
 import { status } from '../../../constants';
+import { ContextProvider } from '@opensrp/react-utils';
 
 jest.mock('@opensrp/notifications', () => ({
   __esModule: true,
@@ -46,7 +47,9 @@ describe('CloseFlagForm component', () => {
 
     render(
       <QueryClientProvider client={queryClient}>
-        <CloseFlagForm {...props} />
+        <ContextProvider>
+          <CloseFlagForm {...props} />
+        </ContextProvider>
       </QueryClientProvider>
     );
 
@@ -83,7 +86,9 @@ describe('CloseFlagForm component', () => {
 
     render(
       <QueryClientProvider client={queryClient}>
-        <CloseFlagForm {...props} />
+        <ContextProvider>
+          <CloseFlagForm {...props} />
+        </ContextProvider>
       </QueryClientProvider>
     );
 

diff --git a/packages/fhir-flag/src/components/LocationFlag/tests/index.test.tsx b/packages/fhir-flag/src/components/LocationFlag/tests/index.test.tsx
@@ -31,6 +31,8 @@ import {
 import userEvents from '@testing-library/user-event';
 import { status } from '../../../constants';
 import * as notifications from '@opensrp/notifications';
+import { RoleContext } from '@opensrp/rbac';
+import { superUserRole } from '@opensrp/react-utils';
 
 jest.mock('@opensrp/notifications', () => ({
   __esModule: true,
@@ -122,9 +124,11 @@ test('renders correctly and fetches data', async () => {
 
   render(
     <QueryClientProvider client={queryClient}>
-      <Router history={history}>
-        <LocationFlag {...defaultProps} />
-      </Router>
+      <RoleContext.Provider value={superUserRole}>
+        <Router history={history}>
+          <LocationFlag {...defaultProps} />
+        </Router>
+      </RoleContext.Provider>
     </QueryClientProvider>
   );