@Test
public void testCalculateUnitFundingSourceBadId() throws Exception {
protocolFundingSourceService = new ProtocolFundingSourceServiceImpl();
protocolFundingSourceService.setUnitService(getUnitService());
protocolFundingSourceService.setFundingSourceTypeService(getFundingSourceTypeService());
ProtocolFundingSource fundingSource =
(ProtocolFundingSource)
protocolFundingSourceService.updateProtocolFundingSource(
UNIT_SOURCE_TYPE_ID, UNIT_NUMBER_BAD, null);
assertNotNull(fundingSource);
assertNull(fundingSource.getFundingSourceName());
}
@Test
public void testIsValidIdForTypeUnit() throws Exception {
protocolFundingSourceService = new ProtocolFundingSourceServiceImpl();
protocolFundingSourceService.setFundingSourceTypeService(getFundingSourceTypeService());
protocolFundingSourceService.setUnitService(getUnitService());
ProtocolFundingSource fundingSource =
new ProtocolFundingSource(UNIT_NUMBER_GOOD, FundingSourceType.UNIT, null, null);
assertTrue(protocolFundingSourceService.isValidIdForType(fundingSource));
fundingSource = new ProtocolFundingSource(UNIT_NUMBER_BAD, FundingSourceType.UNIT, null, null);
assertFalse(protocolFundingSourceService.isValidIdForType(fundingSource));
}
@Test
public void testCalculateUnitFundingSourceEmptyId() throws Exception {
protocolFundingSourceService = new ProtocolFundingSourceServiceImpl();
final UnitService unitService = context.mock(UnitService.class);
protocolFundingSourceService.setUnitService(unitService);
protocolFundingSourceService.setFundingSourceTypeService(getFundingSourceTypeService());
ProtocolFundingSource fundingSource =
(ProtocolFundingSource)
protocolFundingSourceService.updateProtocolFundingSource(
UNIT_SOURCE_TYPE_ID, EMPTY_NUMBER, null);
assertNull(fundingSource);
}
@Test
public void testCalculateUnitFunding() throws Exception {
protocolFundingSourceService = new ProtocolFundingSourceServiceImpl();
protocolFundingSourceService.setUnitService(getUnitService());
protocolFundingSourceService.setFundingSourceTypeService(getFundingSourceTypeService());
ProtocolFundingSource fundingSource =
(ProtocolFundingSource)
protocolFundingSourceService.updateProtocolFundingSource(
UNIT_SOURCE_TYPE_ID, UNIT_NUMBER_GOOD, null);
assertNotNull(fundingSource);
assertNotNull(fundingSource.getFundingSourceName());
assertTrue(fundingSource.getFundingSourceName().equalsIgnoreCase(UNIT_NAME_GOOD));
}
