@Test public void testSuppressMethodInParentOnlyWhenObjectIsSpy() throws Exception { suppress(method(SuppressMethodParent.class, "myMethod")); SuppressMethod tested = spy(new SuppressMethod()); assertEquals(20, tested.myMethod()); }
@SuppressWarnings("unchecked") @Test public void testExecute() throws Exception { protocolRemovedCommand = mock(ProtocolRemovedCommand.class, Mockito.CALLS_REAL_METHODS); PowerMockito.suppress(MemberMatcher.methodsDeclaredIn(InstanceIdentifier.class)); ProtocolEntry protocol = mock(ProtocolEntry.class); when(protocol.getProtocol()) .thenAnswer( new Answer<Class<? extends OvsdbBridgeProtocolBase>>() { public Class<? extends OvsdbBridgeProtocolBase> answer(InvocationOnMock invocation) throws Throwable { return OvsdbBridgeProtocolOpenflow10.class; } }); Optional<ProtocolEntry> operationalProtocolEntryOptional = mock(Optional.class); when(operationalProtocolEntryOptional.isPresent()).thenReturn(true); when(operationalProtocolEntryOptional.get()).thenReturn(protocol); BridgeOperationalState bridgeOpState = mock(BridgeOperationalState.class); when(bridgeOpState.getProtocolEntry(any(InstanceIdentifier.class))) .thenReturn(operationalProtocolEntryOptional); MemberModifier.suppress( MemberMatcher.method(ProtocolUpdateCommand.class, "getOperationalState")); when(protocolRemovedCommand.getOperationalState()).thenReturn(bridgeOpState); InstanceIdentifier<ProtocolEntry> protocolIid = mock(InstanceIdentifier.class); removed.add(protocolIid); MemberModifier.field(ProtocolRemovedCommand.class, "removed") .set(protocolRemovedCommand, removed); MemberModifier.field(ProtocolRemovedCommand.class, "updatedBridges") .set(protocolRemovedCommand, updatedBridges); when(updatedBridges.get(any(InstanceIdentifier.class))) .thenReturn(mock(OvsdbBridgeAugmentation.class)); Operations op = (Operations) setField("op"); Mutate<GenericTableSchema> mutate = mock(Mutate.class); when(op.mutate(any(Bridge.class))).thenReturn(mutate); Column<GenericTableSchema, Set<String>> column = mock(Column.class); Bridge bridge = mock(Bridge.class); when(bridge.getProtocolsColumn()).thenReturn(column); when(column.getSchema()).thenReturn(mock(ColumnSchema.class)); when(column.getData()).thenReturn(new HashSet<String>()); when(mutate.addMutation(any(ColumnSchema.class), any(Mutator.class), any(Set.class))) .thenReturn(mutate); PowerMockito.mockStatic(TyperUtils.class); when(TyperUtils.getTypedRowWrapper(any(DatabaseSchema.class), any(Class.class))) .thenReturn(bridge); TransactionBuilder transaction = mock(TransactionBuilder.class); protocolRemovedCommand.execute(transaction); Mockito.verify(transaction).add(any(Operation.class)); }