Java Whitebox.setInternalState方法代码示例

import org.powermock.reflect.Whitebox; //导入方法依赖的package包/类
@Override
public void setup() {
    super.setup();
    Map<String, String> workerProps = new HashMap<>();
    workerProps.put("key.converter", "org.apache.kafka.connect.json.JsonConverter");
    workerProps.put("value.converter", "org.apache.kafka.connect.json.JsonConverter");
    workerProps.put("internal.key.converter", "org.apache.kafka.connect.json.JsonConverter");
    workerProps.put("internal.value.converter", "org.apache.kafka.connect.json.JsonConverter");
    workerProps.put("internal.key.converter.schemas.enable", "false");
    workerProps.put("internal.value.converter.schemas.enable", "false");
    workerProps.put("offset.storage.file.filename", "/tmp/connect.offsets");
    workerProps.put("offset.flush.interval.ms",
            Long.toString(DEFAULT_OFFSET_COMMIT_INTERVAL_MS));
    WorkerConfig config = new StandaloneConfig(workerProps);
    committer = new SourceTaskOffsetCommitter(config, executor, committers);
    Whitebox.setInternalState(SourceTaskOffsetCommitter.class, "log", mockLog);
}
 

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 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 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包/类
/**
 * 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包/类
@Before
public void setup() {
    super.setup();
    // noinspection unchecked
    stateLiveData = mock(SafeMutableLiveData.class);
    mNavigatorHelper = mock(NavigatorHelper.class);
    Whitebox.setInternalState(viewModel, "stateLiveData", stateLiveData);
    Whitebox.setInternalState(viewModel, "navigatorHelper", mNavigatorHelper);
}
 

import org.powermock.reflect.Whitebox; //导入方法依赖的package包/类
@Test
public void testFromSubscriptionDataA() {
  SubscriptionBuiltinTopicData subscriptionBuiltinTopicData = new SubscriptionBuiltinTopicData();
  Whitebox.setInternalState(subscriptionBuiltinTopicData, "participant_key", builtinTopicKeyA);

  ParticipantBuiltinTopicData result = BuiltinTopicHelper.getParticipantBuiltinTopicData(
      domainParticipant, subscriptionBuiltinTopicData);

  assertTrue(("A").equals(result.participant_name.name));
}
 

import org.powermock.reflect.Whitebox; //导入方法依赖的package包/类
private PublicationBuiltinTopicData createPublicationBuiltinTopicData(
    String topic,
    String type,
    String... partitions
) {
  PublicationBuiltinTopicData builtinTopicData = mock(PublicationBuiltinTopicData.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包/类
/**
 * A helper method for testing needToSendLocation.
 *
 * @param lastLocationSentDate Date of the last location sent.
 * @param lastLocationSentAccuracyType Accuracy type of the last location sent.
 * @param currentLocationAccuracyType Accuracy type of the current location.
 * @param expected Expected return value of needToSendLocation given the parameters.
 * @throws Exception
 */
private void verifyNeedToSendLocation(Date lastLocationSentDate,
    LeanplumLocationAccuracyType lastLocationSentAccuracyType,
    LeanplumLocationAccuracyType currentLocationAccuracyType, boolean expected)
    throws Exception {
  Whitebox.setInternalState(mLocationManager, "lastLocationSentDate", lastLocationSentDate);
  Whitebox.setInternalState(mLocationManager, "lastLocationSentAccuracyType",
      lastLocationSentAccuracyType);

  Boolean result = Whitebox.invokeMethod(mLocationManager, "needToSendLocation",
      currentLocationAccuracyType);

  assertEquals(expected, result);
}
 

import org.powermock.reflect.Whitebox; //导入方法依赖的package包/类
@Test
public void testSimpleContextConstructor() {
  Whitebox.setInternalState(ReactBuildConfig.class, "DEBUG", true);
  ReactApplicationContext context = mock(ReactApplicationContext.class);
  ModuleSpec spec = ModuleSpec.simple(SimpleContextModule.class, context);

  NativeModule module = spec.getProvider().get();
  assertThat(module).isInstanceOf(SimpleContextModule.class);
  SimpleContextModule contextModule = (SimpleContextModule) module;
  assertThat(contextModule.getReactApplicationContext()).isSameAs(context);
}
 

import org.powermock.reflect.Whitebox; //导入方法依赖的package包/类
@Test
public void testIgnoreSubscriptionFalse() throws Exception {
  // setup mock for participant data
  ParticipantBuiltinTopicData participantBuiltinTopicData = mock(ParticipantBuiltinTopicData.class);
  {
    ServiceQosPolicy serviceQosPolicy = PowerMockito.mock(ServiceQosPolicy.class);
    serviceQosPolicy.kind = ServiceQosPolicyKind.NO_SERVICE_QOS;
    Whitebox.setInternalState(participantBuiltinTopicData, "service", serviceQosPolicy);
  }
  PowerMockito.whenNew(ParticipantBuiltinTopicData.class).withAnyArguments().thenReturn(participantBuiltinTopicData);

  // setup mock for topic helper
  PowerMockito.mockStatic(BuiltinTopicHelper.class);
  when(BuiltinTopicHelper.getParticipantBuiltinTopicData(
      any(DomainParticipant.class), any(BuiltinTopicKey_t.class))
  ).thenReturn(participantBuiltinTopicData);

  // setup mock for subscription topic data
  SubscriptionBuiltinTopicData data = mock(SubscriptionBuiltinTopicData.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 subscription
  assertFalse(filter.ignoreSubscription(domainParticipant, instanceHandle, data));
}
 

import org.powermock.reflect.Whitebox; //导入方法依赖的package包/类
/**
 * Setup method
 */
@Before
public void setUpTest_GeneralFileSystemOperations() {

    mockedTestObject = new FileTransferClient(TransferProtocol.FTP);

    // create a mock file transfer client
    ftpMock = createMock(FtpClient.class);
    factoryMock = createMock(ClientFactory.class);

    // inject the mock client into the test object
    Whitebox.setInternalState(mockedTestObject, ftpMock);
}
 

import org.powermock.reflect.Whitebox; //导入方法依赖的package包/类
@Test
public void testIgnoreSubscriptionTrue() throws Exception {
  // setup mock for participant data
  ParticipantBuiltinTopicData participantBuiltinTopicData = mock(ParticipantBuiltinTopicData.class);
  PowerMockito.whenNew(ParticipantBuiltinTopicData.class).withAnyArguments().thenReturn(participantBuiltinTopicData);

  PowerMockito.mockStatic(PropertyQosPolicyHelper.class);
  when(PropertyQosPolicyHelper.lookup_property(
      any(), anyString())//eq("rti.routing_service.group_name"))
  ).thenReturn(new Property_t("rti.routing_service.group_name", "TEST", false));

  // setup mock for topic helper
  PowerMockito.mockStatic(BuiltinTopicHelper.class);
  when(BuiltinTopicHelper.getParticipantBuiltinTopicData(
      any(DomainParticipant.class), any(BuiltinTopicKey_t.class))
  ).thenReturn(participantBuiltinTopicData);

  // setup mock for subscription topic data
  SubscriptionBuiltinTopicData data = mock(SubscriptionBuiltinTopicData.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 subscription
  assertTrue(filter.ignoreSubscription(domainParticipant, instanceHandle, data));
}
 

import org.powermock.reflect.Whitebox; //导入方法依赖的package包/类
@Test
public void testIgnoreSubscriptionFalse() throws Exception {
  // setup mock for participant data
  ParticipantBuiltinTopicData participantBuiltinTopicData = mock(ParticipantBuiltinTopicData.class);
  PowerMockito.whenNew(ParticipantBuiltinTopicData.class).withAnyArguments().thenReturn(participantBuiltinTopicData);

  PowerMockito.mockStatic(PropertyQosPolicyHelper.class);
  when(PropertyQosPolicyHelper.lookup_property(
      any(), anyString())//eq("rti.routing_service.group_name"))
  ).thenReturn(new Property_t("rti.routing_service.group_name", "", false));

  // setup mock for topic helper
  PowerMockito.mockStatic(BuiltinTopicHelper.class);
  when(BuiltinTopicHelper.getParticipantBuiltinTopicData(
      any(DomainParticipant.class), any(BuiltinTopicKey_t.class))
  ).thenReturn(participantBuiltinTopicData);

  // setup mock for subscription topic data
  SubscriptionBuiltinTopicData data = mock(SubscriptionBuiltinTopicData.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 subscription
  assertFalse(filter.ignoreSubscription(domainParticipant, instanceHandle, data));
}
 

import org.powermock.reflect.Whitebox; //导入方法依赖的package包/类
@Test
public void testIgnoreSubscriptionFalse() throws Exception {
  // setup mock for subscription topic data
  SubscriptionBuiltinTopicData data = mock(SubscriptionBuiltinTopicData.class);
  BuiltinTopicKey_t key = PowerMockito.mock(BuiltinTopicKey_t.class);
  Whitebox.setInternalState(data, "participant_key", key);
  Whitebox.setInternalState(data, "topic_name", "topic");

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

  // call ignore subscription
  assertFalse(filter.ignoreSubscription(domainParticipant, instanceHandle, data));
}