use of gov.ca.cwds.data.legacy.cms.entity.StaffPerson in project cals-api by ca-cwds.
the class ChildAssignedWorkerServiceTest method testGetStaffPersonFromActiveCase.
@Test
public void testGetStaffPersonFromActiveCase() {
List<Case> activeCases = new LinkedList<>();
StaffPerson activeCaseStaffPerson = new StaffPerson();
activeCaseStaffPerson.setFirstName(FIRST_NAME);
activeCaseStaffPerson.setLastName(LAST_NAME);
Case case1 = new Case();
case1.setIdentifier(CASE_TEST_ID);
case1.setStaffPerson(activeCaseStaffPerson);
activeCases.add(case1);
CaseDao caseDaoMock = Mockito.mock(CaseDao.class);
ReferralDao referralDaoMock = Mockito.mock(ReferralDao.class);
doReturn(activeCases).when(caseDaoMock).findActiveByClient(TEST_CLIENT_ID);
Whitebox.setInternalState(childAssignedWorkerService, "caseDao", caseDaoMock);
Whitebox.setInternalState(childAssignedWorkerService, "referralDao", referralDaoMock);
Optional<StaffPerson> staffPerson = childAssignedWorkerService.findAssignedWorkerForClient(TEST_CLIENT_ID);
Assert.assertNotNull(staffPerson);
Assert.assertTrue(staffPerson.isPresent());
Assert.assertEquals(FIRST_NAME, staffPerson.get().getFirstName());
Assert.assertEquals(LAST_NAME, staffPerson.get().getLastName());
}
Aggregations