test: Add tests for duplicated users load (#15037)

Author: atomrc

src/script/user/UserRepository.test.ts

@@ -21,6 +21,7 @@ import {RECEIPT_MODE} from '@wireapp/api-client/lib/conversation/data';
 import {amplify} from 'amplify';
 import {StatusCodes as HTTP_STATUS} from 'http-status-codes';
 
+import {Availability} from '@wireapp/protocol-messaging';
 import {WebAppEvents} from '@wireapp/webapp-events';
 
 import {entities} from 'test/api/payloads';
@@ -207,7 +208,6 @@ describe('UserRepository', () => {
 
     describe('loadUsers', () => {
       const localUsers = [generateAPIUser(), generateAPIUser(), generateAPIUser()];
-      //const connections = [new ConnectionEntity()];
       beforeEach(async () => {
         jest.resetAllMocks();
         userState.users.removeAll();
@@ -220,7 +220,7 @@ describe('UserRepository', () => {
         const userIds = users.map(user => user.qualified_id!);
         const fetchUserSpy = jest.spyOn(userRepository['userService'], 'getUsers').mockResolvedValue({found: newUsers});
 
-        await userRepository.loadUsers(new User(), [], [], userIds);
+        await userRepository.loadUsers(new User('self'), [], [], userIds);
 
         expect(userState.users()).toHaveLength(users.length + 1);
         expect(fetchUserSpy).toHaveBeenCalledWith(newUsers.map(user => user.qualified_id!));
@@ -230,12 +230,33 @@ describe('UserRepository', () => {
         const userIds = localUsers.map(user => user.qualified_id!);
         const fetchUserSpy = jest.spyOn(userRepository['userService'], 'getUsers').mockResolvedValue({found: []});
 
-        await userRepository.loadUsers(new User(), [], [], userIds);
+        await userRepository.loadUsers(new User('self'), [], [], userIds);
 
         expect(userState.users()).toHaveLength(localUsers.length + 1);
         expect(fetchUserSpy).not.toHaveBeenCalled();
       });
 
+      it('loads users that are partially stored in the DB and maps availability', async () => {
+        const userIds = localUsers.map(user => user.qualified_id!);
+        const partialUsers = [
+          {id: userIds[0].id, availability: Availability.Type.AVAILABLE},
+          {id: userIds[1].id, availability: Availability.Type.BUSY},
+        ];
+
+        jest.spyOn(userRepository['userService'], 'loadUserFromDb').mockResolvedValue(partialUsers as any);
+        const fetchUserSpy = jest
+          .spyOn(userRepository['userService'], 'getUsers')
+          .mockResolvedValue({found: localUsers});
+
+        await userRepository.loadUsers(new User('self'), [], [], userIds);
+
+        expect(userState.users()).toHaveLength(localUsers.length + 1);
+        expect(fetchUserSpy).toHaveBeenCalledWith(userIds);
+
+        const userWithAvailability = userState.users().filter(user => user.availability() !== Availability.Type.NONE);
+        expect(userWithAvailability).toHaveLength(partialUsers.length);
+      });
+
       it('deletes users that are not needed', async () => {
         const newUsers = [generateAPIUser(), generateAPIUser()];
         const userIds = newUsers.map(user => user.qualified_id!);