mock代码片断1

雨点打透心脏的1/2处 2022-07-17 00:22 76阅读 0赞

/*for validator.java*/
    public class DemoLegValida{ 
    
        public void validateCounterparty(Demosnapshotsnapshot) {
            for (demosnapshot demosnapshot : snapshot.getDemoLegs()) {
                ValidationExceptionUtil.notEmpty(demosnapshot.getCounterParty().getCounterPartyPOGroup(), "POGroup must be specified");
                ValidationExceptionUtil.notEmpty(demosnapshot.getCounterParty().getCounterPartyPOUnit(), "POUnit must be specified");
                ValidationExceptionUtil.notEmpty(demosnapshot.getCounterParty().getCounterPartyIdentifier(), "Counterparty must be specified");
            }
        }

--------------------

/* UnitTest.java */
    public class DemoLegValidatorTest { 
    
        @Test
        public void testcounterpartyValidatorWithNoError() {
            DemoSnapshot snapshot = mock(DemoSnapshot.class);
            List<DemoLegSnapshot> legList = MockDemoLegUtil.getMockedDemoLegSnapshot();
            given(snapshot.getDemoLegs()).willReturn(legList);
            DemoLegValidatorDemoLegValidator= new DemoLegValidator();
            DemoLegValidator.validateCounterparty(snapshot);
        }
    
    
        @Test
        public void testcounterpartyValidatorWithEmptyPartyIdentifier() {
            DemoSnapshot snapshot = mock(DemoSnapshot.class);
        List<DemoLegSnapshot> legList = MockDemoLegUtil.getMockedDemoLegSnapshot_emptyCounterPartyIdentifier();
            given(snapshot.getDemoLegs()).willReturn(legList);
            DemoLegValidatorDemoLegValidator= new DemoLegValidator();
            catchException(DemoLegValidator).validateCounterparty(snapshot);
            assertThat(caughtException(),
                    allOf(instanceOf(ValidationException.class),
                            hasMessage("Counterparty must be specified")));
        }
    
        @Test
        public void testcounterpartyValidatorWithEmptyPOGroup() {
            DemoSnapshot snapshot = mock(DemoSnapshot.class);
            List<DemoLegSnapshot> legList = MockDemoLegUtil.getMockedDemoLegSnapshot_emptyCounterPartyPOGroup();
            given(snapshot.getDemoLegs()).willReturn(legList);
            DemoLegValidatorDemoLegValidator= new DemoLegValidator();
            catchException(DemoLegValidator).validateCounterparty(snapshot);
            assertThat(caughtException(),
                    allOf(instanceOf(ValidationException.class),
                            hasMessage("POGroup must be specified")));
        }
    
        @Test
        public void testcounterpartyValidatorWithEmptyPOUnit() {
            DemoSnapshot snapshot = mock(DemoSnapshot.class);
            List<DemoLegSnapshot> legList = MockDemoLegUtil.getMockedDemoLegSnapshot_emptyCounterPartyPOUnit();
            given(snapshot.getDemoLegs()).willReturn(legList);
            DemoLegValidatorDemoLegValidator= new DemoLegValidator();
            catchException(DemoLegValidator).validateCounterparty(snapshot);
            assertThat(caughtException(),
                    allOf(instanceOf(ValidationException.class),
                            hasMessage("POUnit must be specified")));
        }

--------------------

/* mockdata.java */
    public class MockDemoLegUtil { 
        public static List<DemoLegSnapshot> getMockedDemoLegSnapshot() {
            DemoLegSnapshot mock = mock(DemoLegSnapshot.class);
    
            given(mock.getBaseCurrency()).willReturn("GBP");
            given(mock.getPct()).willReturn(new BigDecimal(50.0));
            given(mock.getLegId()).willReturn("66666");
    
            CounterPartyInfo counterPartyInfo_mock = mock(CounterPartyInfo.class);
    
            given(mock.getCounterParty()).willReturn(counterPartyInfo_mock);
            given(counterPartyInfo_mock.getCounterPartyIdentifier()).willReturn("9999");
            given(counterPartyInfo_mock.getCounterPartyPOGroup()).willReturn("LOH");
            given(counterPartyInfo_mock.getCounterPartyPOUnit()).willReturn("PO7");
    
    
            given(mock.getCounterParty()).willReturn(counterPartyInfo_mock);
            given(mock.getBuySell()).willReturn("S");
            given(mock.getTradingAmount()).willReturn(new BigDecimal(5000.0));
            given(mock.getCounterCurrency()).willReturn("USD");
            given(mock.getCounterAmount()).willReturn(new BigDecimal(6170.0));
    
            List<DemoLegSnapshot> result_list = new ArrayList<>();
            result_list.add(mock);
            return result_list;
        }
    
        public static List<DemoLegSnapshot> getMockedDemoLegSnapshot_emptyCounterPartyIdentifier() {
            DemoLegSnapshot mock = mock(DemoLegSnapshot.class);
    
            CounterPartyInfo counterPartyInfo_mock = mock(CounterPartyInfo.class);
            given(mock.getCounterParty()).willReturn(counterPartyInfo_mock);
            given(counterPartyInfo_mock.getCounterPartyIdentifier()).willReturn("");
            given(counterPartyInfo_mock.getCounterPartyPOGroup()).willReturn("LOH");
            given(counterPartyInfo_mock.getCounterPartyPOUnit()).willReturn("PO7");
            List<DemoLegSnapshot> result_list = new ArrayList<>();
            result_list.add(mock);
            return result_list;
        }
    
        public static List<DemoLegSnapshot> getMockedDemoLegSnapshot_emptyCounterPartyPOGroup() {
            DemoLegSnapshot mock = mock(DemoLegSnapshot.class);
    
            CounterPartyInfo counterPartyInfo_mock = mock(CounterPartyInfo.class);
            given(mock.getCounterParty()).willReturn(counterPartyInfo_mock);
            given(counterPartyInfo_mock.getCounterPartyPOGroup()).willReturn("");
            given(counterPartyInfo_mock.getCounterPartyIdentifier()).willReturn("xxxx");
            given(counterPartyInfo_mock.getCounterPartyPOUnit()).willReturn("PO7");
            List<DemoLegSnapshot> result_list = new ArrayList<>();
            result_list.add(mock);
            return result_list;
        }
    
    
        public static List<DemoLegSnapshot> getMockedDemoLegSnapshot_emptyCounterPartyPOUnit() {
            DemoLegSnapshot mock = mock(DemoLegSnapshot.class);
    
            CounterPartyInfo counterPartyInfo_mock = mock(CounterPartyInfo.class);
            given(mock.getCounterParty()).willReturn(counterPartyInfo_mock);
            given(counterPartyInfo_mock.getCounterPartyPOUnit()).willReturn("");
            given(counterPartyInfo_mock.getCounterPartyIdentifier()).willReturn("yyyy");
            given(counterPartyInfo_mock.getCounterPartyPOGroup()).willReturn("LOH");
            List<DemoLegSnapshot> result_list = new ArrayList<>();
            result_list.add(mock);
            return result_list;
        }