Fix model overview tests

gchq · May 9, 2023 · b9c8880 · b9c8880
1 parent 7f8a9ea
commit b9c8880
Showing 1 changed file with 43 additions and 0 deletions.
diff --git a/frontend/src/ModelOverview.spec.tsx b/frontend/src/ModelOverview.spec.tsx
@@ -1,10 +1,20 @@
 import { ThemeProvider } from '@mui/material/styles'
 import { render, screen, waitFor } from '@testing-library/react'
+import { describe, expect, it, vi } from 'vitest'
 
+import { useGetSchema } from '../data/schema'
 import { EntityKind } from '../types/types'
 import ModelOverview from './ModelOverview'
 import { lightTheme } from './theme'
 
+vi.mock('../data/schema', async () => {
+  const actual = await vi.importActual('../data/schema')
+  return {
+    ...(actual as any),
+    useGetSchema: vi.fn(),
+  }
+})
+
 describe('ModelOverview', () => {
   const version: any = {
     metadata: {
@@ -22,6 +32,39 @@ describe('ModelOverview', () => {
   }
 
   it('renders a ModelOverview component', async () => {
+    const mockedSchema: any = {
+      schema: {
+        name: 'upload-schema',
+        reference: 'test-schema',
+        use: 'UPLOAD',
+        schema: {
+          type: 'object',
+          properties: {
+            contacts: {
+              title: 'Contacts',
+              type: 'object',
+              properties: {
+                uploader: {
+                  title: 'Model Developer',
+                },
+                reviewer: {
+                  title: 'Model Technical Reviewer',
+                },
+                manager: {
+                  title: 'Senior Responsible Officer',
+                },
+              },
+            },
+          },
+        },
+      },
+
+      isSchemaLoading: false,
+      isSchemaError: false,
+    }
+
+    vi.mocked(useGetSchema).mockReturnValue(mockedSchema)
+
     render(
       <ThemeProvider theme={lightTheme}>
         <ModelOverview version={version} />