Java Code Examples for org.mockito.invocation.InvocationOnMock#getMock()

private Answer<?> getKeyAnswer(final String segmentName,
    final boolean isCompressedInput) {
  return new Answer<Object>() {
    int i = 0;

    @SuppressWarnings("unchecked")
    public Boolean answer(InvocationOnMock invocation) {
      if (i++ == 3) {
        return false;
      }
      Reader<Text,Text> mock = (Reader<Text,Text>) invocation.getMock();
      int multiplier = isCompressedInput ? 100 : 1;
      mock.bytesRead += 10 * multiplier;
      Object[] args = invocation.getArguments();
      DataInputBuffer key = (DataInputBuffer) args[0];
      key.reset(("Segment Key " + segmentName + i).getBytes(), 20);
      return true;
    }
  };
}
 

private Answer<?> getKeyAnswer(final String segmentName,
    final boolean isCompressedInput) {
  return new Answer<Object>() {
    int i = 0;

    @SuppressWarnings("unchecked")
    public Boolean answer(InvocationOnMock invocation) {
      if (i++ == 3) {
        return false;
      }
      Reader<Text,Text> mock = (Reader<Text,Text>) invocation.getMock();
      int multiplier = isCompressedInput ? 100 : 1;
      mock.bytesRead += 10 * multiplier;
      Object[] args = invocation.getArguments();
      DataInputBuffer key = (DataInputBuffer) args[0];
      key.reset(("Segment Key " + segmentName + i).getBytes(), 20);
      return true;
    }
  };
}
 

@Override
public Object answer(InvocationOnMock invocation) throws Throwable {
  // Make JavaScriptObject.cast work in most cases by forcing it to return the underlying mock
  // instead of a new mock of type JavaScriptObject. This allows cast to be used in situations
  // that don't violate the Java type system, but not in situations that do (even though
  // javascript would allow them).
  String methodName = invocation.getMethod().getName();
  if (invocation.getMock() instanceof JavaScriptObject && methodName.equals("cast")) {
    return invocation.getMock();
  } else if (invocation.getMock() instanceof Element && methodName.equals("getTagName")) {
    String className = invocation.getMock().getClass().getSimpleName();
    return className.substring(0, className.indexOf("Element")).toLowerCase();
  } else if (invocation.getMock() instanceof InputElement && methodName.equals("getType")) {
    return "text";
  } else {
    return super.answer(invocation);
  }
}
 

public Object answer(InvocationOnMock invocation) {
    if (methodsGuru.isToString(invocation.getMethod())) {
        Object mock = invocation.getMock();
        MockName name = new MockUtil().getMockName(mock);
        if (name.isSurrogate()) {
            return "Mock for " + ClassNameFinder.classNameForMock(mock) + ", hashCode: " + mock.hashCode();
        } else {
            return name.toString();
        }
    } else if (methodsGuru.isCompareToMethod(invocation.getMethod())) {
        //see issue 184.
        //mocks by default should not return 0 for compareTo because they are not the same. Hence we return 1 (anything but 0 is good).
        //Only for compareTo() method by the Comparable interface
        return 1;
    }
    
    Class<?> returnType = invocation.getMethod().getReturnType();
    return returnValueFor(returnType);
}
 

public Object answer(InvocationOnMock invocation) {
    if (methodsGuru.isToString(invocation.getMethod())) {
        Object mock = invocation.getMock();
        MockName name = mockUtil.getMockName(mock);
        if (name.isDefault()) {
            return "Mock for " + mockUtil.getMockSettings(mock).getTypeToMock().getSimpleName() + ", hashCode: " + mock.hashCode();
        } else {
            return name.toString();
        }
    } else if (methodsGuru.isCompareToMethod(invocation.getMethod())) {
        //see issue 184.
        //mocks by default should return 0 if references are the same, otherwise some other value because they are not the same. Hence we return 1 (anything but 0 is good).
        //Only for compareTo() method by the Comparable interface
        return invocation.getMock() == invocation.getArguments()[0] ? 0 : 1;
    }
    
    Class<?> returnType = invocation.getMethod().getReturnType();
    return returnValueFor(returnType);
}
 

@Override
public Object answer(final InvocationOnMock invocationOnMock) throws Throwable
// CS:ON
{
    final Object mock = invocationOnMock.getMock();
    if (invocationOnMock.getMethod().getReturnType().isInstance(mock)) {
        return mock;
    }

    return RETURNS_DEFAULTS.answer(invocationOnMock);
}
 

@Override
public Object answer(InvocationOnMock invocation) throws Throwable {
  Class<?> returnType = invocation.getMethod().getReturnType();

  Object mock = invocation.getMock();
  if (returnType.isAssignableFrom(mock.getClass())) {
    return mock;
  }
  else {
    return null;
  }
}
 

@Override
public Object answer(final InvocationOnMock invocation) throws Throwable {
    if (Modifier.isAbstract(invocation.getMethod().getModifiers())) {
        throw new UnstubbedMethodException(invocation.getMethod(), invocation.getMock());
    }
    return invocation.callRealMethod();
}
 

public Object createSubscriberAnswer(InvocationOnMock invocation) {
    final AbstractMqttAttributeClass attributes = (AbstractMqttAttributeClass) invocation.getMock();
    final ScheduledExecutorService scheduler = (ScheduledExecutorService) invocation.getArguments()[0];
    final Field field = (Field) invocation.getArguments()[1];
    final String topic = (String) invocation.getArguments()[2];
    final boolean mandatory = (boolean) invocation.getArguments()[3];
    final SubscribeFieldToMQTTtopic s = spy(
            new SubscribeFieldToMQTTtopic(scheduler, field, attributes, topic, mandatory));
    doReturn(CompletableFuture.completedFuture(true)).when(s).subscribeAndReceive(any(), anyInt());
    return s;
}
 

public Object createSubscriberAnswer(InvocationOnMock invocation) {
    final AbstractMqttAttributeClass attributes = (AbstractMqttAttributeClass) invocation.getMock();
    final ScheduledExecutorService scheduler = (ScheduledExecutorService) invocation.getArguments()[0];
    final Field field = (Field) invocation.getArguments()[1];
    final String topic = (String) invocation.getArguments()[2];
    final boolean mandatory = (boolean) invocation.getArguments()[3];
    final SubscribeFieldToMQTTtopic s = spy(
            new SubscribeFieldToMQTTtopic(scheduler, field, attributes, topic, mandatory));
    doReturn(CompletableFuture.completedFuture(true)).when(s).subscribeAndReceive(any(), anyInt());
    return s;
}
 

public Node createSpyNode(InvocationOnMock invocation) {
    final Device device = (Device) invocation.getMock();
    final String id = (String) invocation.getArguments()[0];
    // Create the node
    Node node = spy(device.createNode(id, spy(new NodeAttributes())));
    // Intercept creating a property in the next call and inject a spy'ed property.
    doAnswer(this::createSpyProperty).when(node).createProperty(any());
    return node;
}
 

/**
 * Returns the mock itself if return type and mock type match (assume fluent
 * api).
 *
 * @param invocation
 *          the method invocation. If its return type equals the mock type,
 *          just return the mock.
 * @return the mock itself or null (meaning further stubbing is required).
 * @throws Throwable
 *           when anything fails
 */
@Override
@SuppressWarnings("unchecked")
public T answer(@Nonnull final InvocationOnMock invocation) throws Throwable {
  final Class<?> methodReturnType = invocation.getMethod().getReturnType();
  if (type.isAssignableFrom(methodReturnType) // fluent api methods
    || Query.class.isAssignableFrom(methodReturnType) // asc/desc
  ) {
    Object mock = invocation.getMock();
    return (T)mock ;
  }
  return null;
}
 

/**
 * Returns the mock itself if return type and mock type match (assume fluent
 * api).
 *
 * @param invocation the method invocation. If its return type equals the mock type,
 *                   just return the mock.
 *
 * @return the mock itself or null (meaning further stubbing is required).
 *
 * @throws Throwable when anything fails
 */
@Override
@SuppressWarnings("unchecked")
public MessageCorrelationBuilder answer(@Nonnull final InvocationOnMock invocation) throws Throwable {
  final Class<?> methodReturnType = invocation.getMethod().getReturnType();
  if (MessageCorrelationBuilder.class.isAssignableFrom(methodReturnType)) {
    // fluent api methods
    Object mock = invocation.getMock();
    return (MessageCorrelationBuilder) mock;
  }
  return null;
}
 

public Object answer(InvocationOnMock invocation) throws Throwable {
  Object mock = invocation.getMock();
  if (invocation.getMethod().getReturnType().isInstance(mock)) {
    return mock;
  } else {
    return Mockito.RETURNS_DEFAULTS.answer(invocation);
  }
}
 

@Override
public Object answer(InvocationOnMock invocation) throws Throwable {
  Class<?> returnType = invocation.getMethod().getReturnType();
  if (returnType == clazz) {
    return invocation.getMock();
  } else {
    return delegate.answer(invocation);
  }
}
 

public Property createSpyProperty(InvocationOnMock invocation) {
    final Node node = (Node) invocation.getMock();
    final String id = (String) invocation.getArguments()[0];
    Property property = spy(node.createProperty(id, spy(new PropertyAttributes())));
    return property;
}
 

public Object answer(InvocationOnMock invocation) throws Throwable {
    invocation.getArguments();
    invocation.getMock();
    return string;
}
 

public Object answer(InvocationOnMock invocation) throws Throwable {
    invocation.getArguments();
    invocation.getMock();
    return string;
}
 

public Object answer(InvocationOnMock invocation) throws Throwable {
    invocation.getArguments();
    invocation.getMock();
    return string;
}