Java Code Examples for org.powermock.reflect.Whitebox#getMethod()

@Test
public void testTupleRangeCheck() throws IllegalAccessException, IllegalArgumentException, InvocationTargetException {
	
	// test private static final int[] rangeCheckFields(int[] fields, int maxAllowedField)
	Method rangeCheckFieldsMethod = Whitebox.getMethod(Keys.class, "rangeCheckFields", int[].class, int.class);

	// valid indexes
	rangeCheckFieldsMethod.invoke(null, new int[]{1, 2, 3, 4}, 4);

	// corner case tests
	rangeCheckFieldsMethod.invoke(null, new int[] {0}, 0);

	Throwable ex = null;
	try {
		// throws illegal argument.
		rangeCheckFieldsMethod.invoke(null, new int[] {5}, 0);
	} catch(Throwable iae) {
		ex = iae;
	}
	Assert.assertNotNull(ex);
}
 

@Test
public void testTupleRangeCheck() throws IllegalAccessException, IllegalArgumentException, InvocationTargetException {
	
	// test private static final int[] rangeCheckFields(int[] fields, int maxAllowedField)
	Method rangeCheckFieldsMethod = Whitebox.getMethod(Keys.class, "rangeCheckFields", int[].class, int.class);

	// valid indexes
	rangeCheckFieldsMethod.invoke(null, new int[]{1, 2, 3, 4}, 4);

	// corner case tests
	rangeCheckFieldsMethod.invoke(null, new int[] {0}, 0);

	Throwable ex = null;
	try {
		// throws illegal argument.
		rangeCheckFieldsMethod.invoke(null, new int[] {5}, 0);
	} catch(Throwable iae) {
		ex = iae;
	}
	Assert.assertNotNull(ex);
}
 

@Test
public void testTupleRangeCheck() throws IllegalAccessException, IllegalArgumentException, InvocationTargetException {
	
	// test private static final int[] rangeCheckFields(int[] fields, int maxAllowedField)
	Method rangeCheckFieldsMethod = Whitebox.getMethod(Keys.class, "rangeCheckFields", int[].class, int.class);

	// valid indexes
	rangeCheckFieldsMethod.invoke(null, new int[]{1, 2, 3, 4}, 4);

	// corner case tests
	rangeCheckFieldsMethod.invoke(null, new int[] {0}, 0);

	Throwable ex = null;
	try {
		// throws illegal argument.
		rangeCheckFieldsMethod.invoke(null, new int[] {5}, 0);
	} catch(Throwable iae) {
		ex = iae;
	}
	Assert.assertNotNull(ex);
}
 

@Test
public void testGetContextWithoutRollUp()
    throws InvocationTargetException, IllegalAccessException {
  Method method = Whitebox.getMethod(TelemetryGenerator.class, "getContext", Map.class);
  Context ctx = (Context) method.invoke(null, context);
  assertEquals("postman", ctx.getDid());
  assertEquals("ORG_001", ctx.getChannel());
  assertEquals("User", ctx.getEnv());
}
 

@Test
public void testGetContextWithRollUp() throws InvocationTargetException, IllegalAccessException {
  rollup.put("id", 1);
  context.put("rollup", rollup);
  Method method = Whitebox.getMethod(TelemetryGenerator.class, "getContext", Map.class);
  Context ctx = (Context) method.invoke(null, context);
  assertTrue(rollup.equals(ctx.getRollup()));
}
 

@Test
public void testRemoveAttributes() throws InvocationTargetException, IllegalAccessException {
  Method method =
      Whitebox.getMethod(TelemetryGenerator.class, "removeAttributes", Map.class, String.class);
  String[] removableProperty = {JsonKey.DEVICE_ID};
  method.invoke(null, context, removableProperty);
  assertFalse(context.containsKey(JsonKey.DEVICE_ID));
}
 

@Test()
public void testGetProducerWithContextNull()
    throws InvocationTargetException, IllegalAccessException {

  Map<String, Object> nullContext = null;
  Method method = Whitebox.getMethod(TelemetryGenerator.class, "getProducer", Map.class);
  Producer producer = (Producer) method.invoke(null, nullContext);
  assertEquals("", producer.getId());
  assertEquals("", producer.getPid());
  assertEquals("", producer.getVer());
}
 

@Test
public void testGetProducerWithAppId() throws InvocationTargetException, IllegalAccessException {
  context.put("appId", "random");
  Method method = Whitebox.getMethod(TelemetryGenerator.class, "getProducer", Map.class);
  Producer producer = (Producer) method.invoke(null, context);
  assertEquals("random", producer.getId());
}
 

@Test
public void testGetProducerWithoutAppId()
    throws InvocationTargetException, IllegalAccessException {
  context.put("telemetry_pdata_id", "local.sunbird.learning.service");
  Method method = Whitebox.getMethod(TelemetryGenerator.class, "getProducer", Map.class);
  Producer producer = (Producer) method.invoke(null, context);
  assertEquals("local.sunbird.learning.service", producer.getId());
}
 

/**
 * Every time the {@link Dependency#getData()} method is invoked we return a
 * custom instance of a {@link DataObject}.
 */
@Override
public void applyInterceptionPolicy(MockPolicyInterceptionSettings settings) {
	final Method getDataMethod = Whitebox.getMethod(Dependency.class);
	final DataObject dataObject = new DataObject("Policy generated data object");
	settings.stubMethod(getDataMethod, dataObject);
}