add primary images query #1081

ral-facilities · Nov 4, 2024 · 5a666bb · 5a666bb
1 parent 5dc385e
commit 5a666bb
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 4 deletions.
diff --git a/src/api/images.test.tsx b/src/api/images.test.tsx
@@ -19,5 +19,17 @@ describe('images api functions', () => {
 
       expect(result.current.data?.length).toEqual(20);
     });
+
+    it('sends request to fetch primary image data and returns successful response', async () => {
+      const { result } = renderHook(() => useGetImages('1', true), {
+        wrapper: hooksWrapperWithProviders(),
+      });
+
+      await waitFor(() => {
+        expect(result.current.isSuccess).toBeTruthy();
+      });
+
+      expect(result.current.data?.length).toEqual(1);
+    });
   });
 });
diff --git a/src/api/images.tsx b/src/api/images.tsx
@@ -9,9 +9,14 @@ export const getImage = async (id: string): Promise<ImageGet> => {
   });
 };
 
-const getImages = async (entityId: string): Promise<Image[]> => {
+const getImages = async (
+  entityId: string,
+  primary?: boolean
+): Promise<Image[]> => {
   const queryParams = new URLSearchParams();
   queryParams.append('entity_id', entityId);
+
+  if (primary !== undefined) queryParams.append('primary', String(primary));
   return storageApi
     .get(`/images`, {
       params: queryParams,
@@ -20,11 +25,12 @@ const getImages = async (entityId: string): Promise<Image[]> => {
 };
 
 export const useGetImages = (
-  entityId?: string
+  entityId?: string,
+  primary?: boolean
 ): UseQueryResult<Image[], AxiosError> => {
   return useQuery({
-    queryKey: ['Images', entityId],
-    queryFn: () => getImages(entityId ?? ''),
+    queryKey: ['Images', entityId, primary],
+    queryFn: () => getImages(entityId ?? '', primary),
     enabled: !!entityId,
   });
 };