feat: add tests

Author: reneshen0328

src/elements/content-sharing/hooks/__tests__/useContactService.test.ts

@@ -0,0 +1,71 @@
+import { renderHook } from '@testing-library/react';
+
+import { convertGroupContactsResponse, convertUserContactsResponse } from '../../utils';
+import { useContactService } from '../useContactService';
+import useContacts from '../useContacts';
+
+jest.mock('../useContacts');
+jest.mock('../../utils');
+
+describe('elements/content-sharing/hooks/useContactService', () => {
+    const mockApi = {
+        getMarkerBasedUsersAPI: jest.fn(),
+        getMarkerBasedGroupsAPI: jest.fn(),
+    };
+    const mockItemID = '123456789';
+    const mockCurrentUserID = '123';
+    const mockGetContacts = jest.fn();
+
+    beforeEach(() => {
+        (useContacts as jest.Mock).mockReturnValue(mockGetContacts);
+        (convertGroupContactsResponse as jest.Mock).mockReturnValue([]);
+        (convertUserContactsResponse as jest.Mock).mockReturnValue([]);
+    });
+
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+
+    test('should return null contactService when currentUserID is null or undefined', () => {
+        [null, undefined].forEach(currentUserID => {
+            const { result } = renderHook(() => useContactService(mockApi, mockItemID, currentUserID));
+
+            expect(result.current.contactService).toBeNull();
+        });
+    });
+
+    test('should return contactService with getContacts function', () => {
+        const { result } = renderHook(() => useContactService(mockApi, mockItemID, mockCurrentUserID));
+
+        expect(useContacts).toHaveBeenCalledWith(mockApi, mockItemID, {
+            transformUsers: expect.any(Function),
+            transformGroups: expect.any(Function),
+        });
+        expect(result.current.contactService).toEqual({
+            getContacts: mockGetContacts,
+        });
+    });
+
+    test('should pass transform functions that call correct conversion functions with params', () => {
+        const mockTransformedUsers = [{ id: 'user1', email: 'user1@test.com' }];
+        const mockTransformedGroups = [{ id: 'group1', name: 'Test Group' }];
+        const mockUserData = { entries: mockTransformedUsers };
+        const mockGroupData = { entries: mockTransformedGroups };
+
+        (convertUserContactsResponse as jest.Mock).mockReturnValue(mockTransformedUsers);
+        (convertGroupContactsResponse as jest.Mock).mockReturnValue(mockTransformedGroups);
+
+        renderHook(() => useContactService(mockApi, mockItemID, mockCurrentUserID));
+
+        // Get the transform functions that were passed to useContacts
+        const transformUsersFn = useContacts.mock.calls[0][2].transformUsers;
+        const transformGroupsFn = useContacts.mock.calls[0][2].transformGroups;
+        const resultUsers = transformUsersFn(mockUserData);
+        const resultGroups = transformGroupsFn(mockGroupData);
+
+        expect(convertUserContactsResponse as jest.Mock).toHaveBeenCalledWith(mockUserData, mockCurrentUserID);
+        expect(convertGroupContactsResponse as jest.Mock).toHaveBeenCalledWith(mockGroupData);
+        expect(resultUsers).toBe(mockTransformedUsers);
+        expect(resultGroups).toBe(mockTransformedGroups);
+    });
+});

src/elements/content-sharing/hooks/useContactService.ts

@@ -1,86 +1,21 @@
 import * as React from 'react';
 
-import { STATUS_INACTIVE } from '../../../constants';
+import { convertGroupContactsResponse, convertUserContactsResponse } from '../utils';
 import useContacts from './useContacts';
-import useContactsByEmail from './useContactsByEmail';
-import { convertUserContactsByEmailResponse } from '../../../features/unified-share-modal/utils/convertData';
 
-const APP_USERS_DOMAIN_REGEXP = /boxdevedition.com/;
-const sortByName = ({ name: nameA = '' }, { name: nameB = '' }) => nameA.localeCompare(nameB);
-
-/**
- * Convert an enterprise users API response into an array of internal USM contacts.
- */
-export const convertUserContactsResponse = (contactsAPIData, currentUserID) => {
-    const { entries = [] } = contactsAPIData;
-
-    // Return all active users except for the current user and app users
-    return entries
-        .filter(
-            ({ id, login: email, status }) =>
-                id !== currentUserID &&
-                email &&
-                !APP_USERS_DOMAIN_REGEXP.test(email) &&
-                status &&
-                status !== STATUS_INACTIVE,
-        )
-        .map(contact => {
-            const { id, login: email, name, type } = contact;
-            return {
-                id,
-                email,
-                name,
-                type,
-                value: email,
-            };
-        })
-        .sort(sortByName);
-};
-
-/**
- * Convert an enterprise groups API response into an array of internal USM contacts.
- */
-export const convertGroupContactsResponse = contactsAPIData => {
-    const { entries = [] } = contactsAPIData;
-
-    // Only return groups with the correct permissions
-    return entries
-        .filter(({ permissions }) => {
-            return permissions && permissions.can_invite_as_collaborator;
-        })
-        .map(contact => {
-            const { id, name, type } = contact;
-            return {
-                id,
-                email: 'Group', // Need this for the avatar to work for isUserContactType
-                name,
-                type,
-                value: 'Group',
-            };
-        })
-        .sort(sortByName);
-};
-
-export const useContactService = (api, itemID, currentUserID) => {
+export const useContactService = (api, itemID, currentUserId) => {
     const getContacts = useContacts(api, itemID, {
-        transformUsers: data => convertUserContactsResponse(data, currentUserID),
+        transformUsers: data => convertUserContactsResponse(data, currentUserId),
         transformGroups: data => convertGroupContactsResponse(data),
     });
 
-    const getContactsByEmail = useContactsByEmail(api, itemID, {
-        transformUsers: data => convertUserContactsByEmailResponse(data),
-    });
-
     const contactService = React.useMemo(() => {
-        if (!currentUserID) {
+        if (!currentUserId) {
             return null;
         }
 
-        return {
-            getContacts,
-            getContactsByEmail,
-        };
-    }, [currentUserID, getContacts, getContactsByEmail]);
+        return { getContacts };
+    }, [currentUserId, getContacts]);
 
     return { contactService };
 };