Java Whitebox类代码示例

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void testSendRecordsPropagatesTimestamp() throws Exception {
    final Long timestamp = System.currentTimeMillis();

    createWorkerTask();

    List<SourceRecord> records = Collections.singletonList(
            new SourceRecord(PARTITION, OFFSET, "topic", null, KEY_SCHEMA, KEY, RECORD_SCHEMA, RECORD, timestamp)
    );

    Capture<ProducerRecord<byte[], byte[]>> sent = expectSendRecordAnyTimes();

    PowerMock.replayAll();

    Whitebox.setInternalState(workerTask, "toSend", records);
    Whitebox.invokeMethod(workerTask, "sendRecords");
    assertEquals(timestamp, sent.getValue().timestamp());

    PowerMock.verifyAll();
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void createBinaryFilePositiveRandomContent() throws IOException {

    //Replace the randomGenerator with cutsom one.
    Random random = new Random(1);
    Random originalRandom = (Random) Whitebox.getInternalState(testObject.getClass(),
                                                               "randomGenerator");
    Whitebox.setInternalState(testObject.getClass(), "randomGenerator", random);

    testObject.createBinaryFile(file.getPath(), 10, true);

    //Restore original random generator
    Whitebox.setInternalState(testObject.getClass(), "randomGenerator", originalRandom);

    assertTrue(file.exists());
    assertEquals(10L, file.length());

    byte[] expectedBytes = new byte[]{ 115, -40, 111, -110, -4, -16, -27, -35, -43, 104 };
    byte[] actualBytes = new byte[10];
    BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));
    bis.read(actualBytes);
    assertTrue(Arrays.equals(expectedBytes, actualBytes));
    bis.close();

}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void testSendRecordsConvertsData() throws Exception {
    createWorkerTask();

    List<SourceRecord> records = new ArrayList<>();
    // Can just use the same record for key and value
    records.add(new SourceRecord(PARTITION, OFFSET, "topic", null, KEY_SCHEMA, KEY, RECORD_SCHEMA, RECORD));

    Capture<ProducerRecord<byte[], byte[]>> sent = expectSendRecordAnyTimes();

    PowerMock.replayAll();

    Whitebox.setInternalState(workerTask, "toSend", records);
    Whitebox.invokeMethod(workerTask, "sendRecords");
    assertEquals(SERIALIZED_KEY, sent.getValue().key());
    assertEquals(SERIALIZED_RECORD, sent.getValue().value());

    PowerMock.verifyAll();
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
private SubscriptionBuiltinTopicData createSubscriptionBuiltinTopicData(
    String topic,
    String type,
    String... partitions
) {
  SubscriptionBuiltinTopicData builtinTopicData = mock(SubscriptionBuiltinTopicData.class);
  PartitionQosPolicy partitionQosPolicy = mock(PartitionQosPolicy.class);
  Whitebox.setInternalState(builtinTopicData, "partition", partitionQosPolicy);
  Whitebox.setInternalState(partitionQosPolicy, "name", new StringSeq());

  builtinTopicData.topic_name = topic;
  builtinTopicData.type_name = type;
  builtinTopicData.partition.name.addAll(Arrays.asList(partitions));

  return builtinTopicData;
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void testIgnorePublicationFalseNoParticipantData() throws Exception {
  // setup mock for topic helper
  PowerMockito.mockStatic(BuiltinTopicHelper.class);
  when(BuiltinTopicHelper.getParticipantBuiltinTopicData(
      any(DomainParticipant.class), any(BuiltinTopicKey_t.class))
  ).thenReturn(null);

  // setup mock for publication topic data
  PublicationBuiltinTopicData data = mock(PublicationBuiltinTopicData.class);
  BuiltinTopicKey_t key = PowerMockito.mock(BuiltinTopicKey_t.class);
  Whitebox.setInternalState(data, "participant_key", key);

  // setup mock for instance handle
  InstanceHandle_t instanceHandle = mock(InstanceHandle_t.class);

  // call ignore publication
  assertFalse(filter.ignorePublication(domainParticipant, instanceHandle, data));
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
protected static HttpClient
        getHttpClient( MockHttpURLConnection mockHttpURLConnection ) throws AgentException {

    // construct the client
    List<ActionHeader> httpHeaders = new ArrayList<ActionHeader>();
    httpHeaders.add( new ActionHeader( "User-Agent",
                                       "Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.9.2.3) Gecko/20100401 Firefox/3.6.3" ) );
    NetworkingStopWatch stopWatch = new NetworkingStopWatch( "action" );
    stopWatch.step0_SetNewContext( "my_action[1]" );
    stopWatch.setStateFromBeforeStep1ToAfterStep4(); /// so it can be suspended for getting response
    stopWatch.step5_StartInterimTimer();
    HttpClient client = new HttpClient( "http://", "GET", httpHeaders, stopWatch );

    // inject a mock connection object
    Whitebox.setInternalState( client, "urlConnection", mockHttpURLConnection );

    return client;
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
/**
 * Tests for requestLocation.
 *
 * @throws Exception
 */
@Test
public void requestLocation() throws Exception {
  GoogleApiClient mockGoogleApiClient = mock(GoogleApiClient.class);
  doReturn(true).when(mockGoogleApiClient).isConnected();

  LocationManager mockLocationManager = spy(mLocationManager);
  Whitebox.setInternalState(mockLocationManager, "googleApiClient", mockGoogleApiClient);

  FusedLocationProviderApi mockLocationProviderApi = mock(FusedLocationProviderApi.class);
  Whitebox.setInternalState(LocationServices.class, "FusedLocationApi", mockLocationProviderApi);

  // Testing when a customer did not disableLocationCollection.
  Whitebox.invokeMethod(mockLocationManager, "requestLocation");
  verify(mockLocationProviderApi).requestLocationUpdates(any(GoogleApiClient.class),
      any(LocationRequest.class), any(LocationListener.class));

  // Testing when a customer disableLocationCollection.
  Leanplum.disableLocationCollection();
  Whitebox.invokeMethod(mockLocationManager, "requestLocation");
  verifyNoMoreInteractions(mockLocationProviderApi);
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Override
public void before() throws Exception {
  super.before();

  mMessageMatchResult = new ActionManager.MessageMatchResult();
  Whitebox.setInternalState(mMessageMatchResult, "matchedTrigger", true);
  Whitebox.setInternalState(mMessageMatchResult, "matchedLimit", true);
  assertTrue(mMessageMatchResult.matchedTrigger);
  assertTrue(mMessageMatchResult.matchedLimit);


  mContextualValues = new ActionContext.ContextualValues();
  // create mock object
  mMockActionManager = mock(ActionManager.class);
  // workaround for singleton objects
  Whitebox.setInternalState(ActionManager.class, "instance", mMockActionManager);
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void successGetShiftList() throws Exception {
    LocalTime startTime = LocalTime.of(6, 30);
    LocalTime endTime = LocalTime.of(9, 0);
    List<Shift> shiftListTest = Arrays.asList(
            new Shift(0, DayOfWeek.SUNDAY, startTime, endTime),
            new Shift(1, DayOfWeek.MONDAY, startTime, endTime),
            new Shift(2, DayOfWeek.TUESDAY, startTime, endTime),
            new Shift(3, DayOfWeek.WEDNESDAY, startTime, endTime),
            new Shift(4, DayOfWeek.THURSDAY, startTime, endTime));

    when(requestMock.headers("Set-Cookie")).thenReturn("cookie");
    when(DataStore.getShifts("cookie")).thenReturn(shiftListTest);

    List<Shift> getShiftListResult = Whitebox.invokeMethod(shiftControllerTest, "getShiftList", requestMock, responseMock);
    Assert.assertEquals(shiftListTest, getShiftListResult);
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void successGetFullCourseList() throws Exception {
    List<Course> courseListTest = new ArrayList<>();
    courseListTest.add(new Course(1, "CS302", null));
    courseListTest.add(new Course(2, "CS301", null));

    List<Course> expectedCourseListTest = new ArrayList<>();
    expectedCourseListTest.add(new Course(1, "CS302", null));

    when(requestMock.headers("Set-Cookie")).thenReturn("cookie");
    when(requestMock.queryParamOrDefault("limit", null)).thenReturn("1");
    when(DataStore.getCourses("cookie")).thenReturn(courseListTest);

    List<Course> getCourseListResult = Whitebox.invokeMethod(courseControllerTest, "getCourseList", requestMock, responseMock);

    Assert.assertEquals(expectedCourseListTest, getCourseListResult);
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void testCacheSchemaToConnectConversion() {
    Cache<JsonNode, Schema> cache = Whitebox.getInternalState(converter, "toConnectSchemaCache");
    assertEquals(0, cache.size());

    converter.toConnectData(TOPIC, "{ \"schema\": { \"type\": \"boolean\" }, \"payload\": true }".getBytes());
    assertEquals(1, cache.size());

    converter.toConnectData(TOPIC, "{ \"schema\": { \"type\": \"boolean\" }, \"payload\": true }".getBytes());
    assertEquals(1, cache.size());

    // Different schema should also get cached
    converter.toConnectData(TOPIC, "{ \"schema\": { \"type\": \"boolean\", \"optional\": true }, \"payload\": true }".getBytes());
    assertEquals(2, cache.size());

    // Even equivalent, but different JSON encoding of schema, should get different cache entry
    converter.toConnectData(TOPIC, "{ \"schema\": { \"type\": \"boolean\", \"optional\": false }, \"payload\": true }".getBytes());
    assertEquals(3, cache.size());
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void testCacheSchemaToJsonConversion() {
    Cache<Schema, ObjectNode> cache = Whitebox.getInternalState(converter, "fromConnectSchemaCache");
    assertEquals(0, cache.size());

    // Repeated conversion of the same schema, even if the schema object is different should return the same Java
    // object
    converter.fromConnectData(TOPIC, SchemaBuilder.bool().build(), true);
    assertEquals(1, cache.size());

    converter.fromConnectData(TOPIC, SchemaBuilder.bool().build(), true);
    assertEquals(1, cache.size());

    // Validate that a similar, but different schema correctly returns a different schema.
    converter.fromConnectData(TOPIC, SchemaBuilder.bool().optional().build(), true);
    assertEquals(2, cache.size());
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test(expected = InvalidRecordException.class)
public void testSendRecordsCorruptTimestamp() throws Exception {
    final Long timestamp = -3L;
    createWorkerTask();

    List<SourceRecord> records = Collections.singletonList(
            new SourceRecord(PARTITION, OFFSET, "topic", null, KEY_SCHEMA, KEY, RECORD_SCHEMA, RECORD, timestamp)
    );

    Capture<ProducerRecord<byte[], byte[]>> sent = expectSendRecordAnyTimes();

    PowerMock.replayAll();

    Whitebox.setInternalState(workerTask, "toSend", records);
    Whitebox.invokeMethod(workerTask, "sendRecords");
    assertEquals(null, sent.getValue().timestamp());

    PowerMock.verifyAll();
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void testSendRecordsNoTimestamp() throws Exception {
    final Long timestamp = -1L;
    createWorkerTask();

    List<SourceRecord> records = Collections.singletonList(
            new SourceRecord(PARTITION, OFFSET, "topic", null, KEY_SCHEMA, KEY, RECORD_SCHEMA, RECORD, timestamp)
    );

    Capture<ProducerRecord<byte[], byte[]>> sent = expectSendRecordAnyTimes();

    PowerMock.replayAll();

    Whitebox.setInternalState(workerTask, "toSend", records);
    Whitebox.invokeMethod(workerTask, "sendRecords");
    assertEquals(null, sent.getValue().timestamp());

    PowerMock.verifyAll();
}
 

import org.powermock.reflect.Whitebox; //导入依赖的package包/类
@Test
public void testSendRecordsTaskCommitRecordFail() throws Exception {
    createWorkerTask();

    // Differentiate only by Kafka partition so we can reuse conversion expectations
    SourceRecord record1 = new SourceRecord(PARTITION, OFFSET, "topic", 1, KEY_SCHEMA, KEY, RECORD_SCHEMA, RECORD);
    SourceRecord record2 = new SourceRecord(PARTITION, OFFSET, "topic", 2, KEY_SCHEMA, KEY, RECORD_SCHEMA, RECORD);
    SourceRecord record3 = new SourceRecord(PARTITION, OFFSET, "topic", 3, KEY_SCHEMA, KEY, RECORD_SCHEMA, RECORD);

    // Source task commit record failure will not cause the task to abort
    expectSendRecordOnce(false);
    expectSendRecordTaskCommitRecordFail(false, false);
    expectSendRecordOnce(false);

    PowerMock.replayAll();

    Whitebox.setInternalState(workerTask, "toSend", Arrays.asList(record1, record2, record3));
    Whitebox.invokeMethod(workerTask, "sendRecords");
    assertEquals(false, Whitebox.getInternalState(workerTask, "lastSendFailed"));
    assertNull(Whitebox.getInternalState(workerTask, "toSend"));

    PowerMock.verifyAll();
}