Java MemberModifier类代码示例

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
public void testReplaceView() throws Exception {
    ArrayList<ViewsInfo> mockedViewsInfos = mock(ArrayList.class);
    MemberModifier.field(UnderlabelValidator.class, "mContext").set(mSpiedUnderlabelValidator, mMockedContext);
    MemberModifier.field(UnderlabelValidator.class, "mViewsInfos").set(mSpiedUnderlabelValidator, mockedViewsInfos);
    int index = 128;
    EditText mockedEditText = mock(EditText.class);
    ViewGroup mockedParent = mock(ViewGroup.class);
    LinearLayout mockedContainer = mock(LinearLayout.class);
    ViewsInfo mockedViewsInfo = mock(ViewsInfo.class);
    when(mMockedValidationHolderRegexTypeFail.getEditText()).thenReturn(mockedEditText);
    when(mockedEditText.getParent()).thenReturn(mockedParent);
    when(mockedParent.indexOfChild(mockedEditText)).thenReturn(index);
    PowerMockito.whenNew(LinearLayout.class).withArguments(mMockedContext).thenReturn(mockedContainer);
    PowerMockito.whenNew(ViewsInfo.class).withArguments(index, mockedParent, mockedContainer, mockedEditText).thenReturn(mockedViewsInfo);
    Whitebox.invokeMethod(mSpiedUnderlabelValidator, "replaceView", mMockedValidationHolderRegexTypeFail);
    verify(mockedViewsInfos, times(1)).add(mockedViewsInfo);
    TextView mockedTextView = mock(TextView.class);
    PowerMockito.whenNew(TextView.class).withArguments(mMockedContext).thenReturn(mockedTextView);
    assertEquals(mockedTextView, Whitebox.invokeMethod(mSpiedUnderlabelValidator, "replaceView", mMockedValidationHolderRegexTypeFail));
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
public void testHaltRestoreViewsForAllAnyway() throws IllegalAccessException {
    ArrayList<ViewsInfo> spiedViewsInfos = spy(ArrayList.class);
    ViewsInfo mockedViewsInfo1 = mock(ViewsInfo.class);
    ViewsInfo mockedViewsInfo2 = mock(ViewsInfo.class);
    ViewsInfo mockedViewsInfo3 = mock(ViewsInfo.class);
    ViewsInfo mockedViewsInfo4 = mock(ViewsInfo.class);
    spiedViewsInfos.addAll(Arrays.asList(mockedViewsInfo1, mockedViewsInfo2, mockedViewsInfo3, mockedViewsInfo4));
    MemberModifier.field(UnderlabelValidator.class, "mViewsInfos").set(mSpiedUnderlabelValidator, spiedViewsInfos);
    mSpiedUnderlabelValidator.mValidationHolderList.addAll(Arrays.asList(mMockedValidationHolderRegexTypePass,
            mMockedValidationHolderRegexTypeFail,
            mMockedValidationHolderRangeTypePass,
            mMockedValidationHolderRangeTypeFail));
    mSpiedUnderlabelValidator.halt();
    for (ViewsInfo viewsInfo : spiedViewsInfos) {
        verify(viewsInfo, times(1)).restoreViews();
    }
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testCreateDataCarrier() throws IllegalAccessException {
    DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(5, 100);
    Assert.assertEquals(((Integer) (MemberModifier.field(DataCarrier.class, "bufferSize").get(carrier))).intValue(), 100);
    Assert.assertEquals(((Integer) (MemberModifier.field(DataCarrier.class, "channelSize").get(carrier))).intValue(), 5);

    Channels<SampleData> channels = (Channels<SampleData>) (MemberModifier.field(DataCarrier.class, "channels").get(carrier));
    Assert.assertEquals(channels.getChannelSize(), 5);

    Buffer<SampleData> buffer = channels.getBuffer(0);
    Assert.assertEquals(buffer.getBufferSize(), 100);

    Assert.assertEquals(MemberModifier.field(Buffer.class, "strategy").get(buffer), BufferStrategy.BLOCKING);
    carrier.setBufferStrategy(BufferStrategy.IF_POSSIBLE);
    Assert.assertEquals(MemberModifier.field(Buffer.class, "strategy").get(buffer), BufferStrategy.IF_POSSIBLE);

    Assert.assertEquals(MemberModifier.field(Channels.class, "dataPartitioner").get(channels).getClass(), SimpleRollingPartitioner.class);
    carrier.setPartitioner(new ProducerThreadPartitioner<SampleData>());
    Assert.assertEquals(MemberModifier.field(Channels.class, "dataPartitioner").get(channels).getClass(), ProducerThreadPartitioner.class);
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testProduce() throws IllegalAccessException {
    DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(2, 100);
    Assert.assertTrue(carrier.produce(new SampleData().setName("a")));
    Assert.assertTrue(carrier.produce(new SampleData().setName("b")));
    Assert.assertTrue(carrier.produce(new SampleData().setName("c")));
    Assert.assertTrue(carrier.produce(new SampleData().setName("d")));

    Channels<SampleData> channels = (Channels<SampleData>) (MemberModifier.field(DataCarrier.class, "channels").get(carrier));
    Buffer<SampleData> buffer1 = channels.getBuffer(0);
    List result1 = buffer1.obtain(0, 100);

    Buffer<SampleData> buffer2 = channels.getBuffer(1);
    List result2 = buffer2.obtain(0, 100);

    Assert.assertEquals(2, result1.size());
    Assert.assertEquals(4, result1.size() + result2.size());


}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testOverrideProduce() throws IllegalAccessException {
    DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(2, 100);
    carrier.setBufferStrategy(BufferStrategy.OVERRIDE);

    for (int i = 0; i < 500; i++) {
        Assert.assertTrue(carrier.produce(new SampleData().setName("d" + i)));
    }

    Channels<SampleData> channels = (Channels<SampleData>) (MemberModifier.field(DataCarrier.class, "channels").get(carrier));
    Buffer<SampleData> buffer1 = channels.getBuffer(0);
    List result1 = buffer1.obtain(0, 100);

    Buffer<SampleData> buffer2 = channels.getBuffer(1);
    List result2 = buffer2.obtain(0, 100);
    Assert.assertEquals(200, result1.size() + result2.size());
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testIfPossibleProduce() throws IllegalAccessException {
    DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(2, 100);
    carrier.setBufferStrategy(BufferStrategy.IF_POSSIBLE);

    for (int i = 0; i < 200; i++) {
        Assert.assertTrue(carrier.produce(new SampleData().setName("d" + i)));
    }

    for (int i = 0; i < 200; i++) {
        Assert.assertFalse(carrier.produce(new SampleData().setName("d" + i + "_2")));
    }

    Channels<SampleData> channels = (Channels<SampleData>) (MemberModifier.field(DataCarrier.class, "channels").get(carrier));
    Buffer<SampleData> buffer1 = channels.getBuffer(0);
    List result1 = buffer1.obtain(0, 100);

    Buffer<SampleData> buffer2 = channels.getBuffer(1);
    List result2 = buffer2.obtain(0, 100);
    Assert.assertEquals(200, result1.size() + result2.size());
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testDoGetBadTicket() throws Exception {
    //Mock some objects.
    HttpServletRequest request = createDoGetHttpServletRequest(CONVERSATION_TICKET, TICKET, "false");
    HttpServletResponse response = createMockHttpServletResponse();
    Cas20ServiceTicketValidator ticketValidator = PowerMockito.mock(Cas20ServiceTicketValidator.class);
    PowerMockito.when(ticketValidator.validate(TICKET, URL_WITH_CONVERSATION)).thenThrow(new TicketValidationException("Invalid Ticket"));

    PowerMockito.mockStatic(ExternalAuthentication.class);
    BDDMockito.given(ExternalAuthentication.startExternalAuthentication(request)).willThrow(new ExternalAuthenticationException());

    //Prep our object
    ShibcasAuthServlet shibcasAuthServlet = createShibcasAuthServlet();

    //Override the internal Cas20TicketValidator because we don't want it to call a real server
    MemberModifier.field(ShibcasAuthServlet.class, "ticketValidator").set(shibcasAuthServlet, ticketValidator);

    //Standard request/response - bad ticket
    BDDMockito.given(request.getAttribute(ExternalAuthentication.FORCE_AUTHN_PARAM)).willReturn("false");
    BDDMockito.given(request.getAttribute(ExternalAuthentication.PASSIVE_AUTHN_PARAM)).willReturn("false");
    shibcasAuthServlet.doGet(request, response);

    //Verify
    verify(request).getRequestDispatcher("/no-conversation-state.jsp");
    verify(response).setStatus(404);
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
public static void buildWorkflowModel(String bundleDirectory, AbstractWorkflowDataModel w) throws IllegalAccessException {

        //Need to mock basedir as it is retrieved from the WS
        MemberModifier.field(AbstractWorkflowDataModel.class, "basedir").set(w, bundleDirectory);

        //Get the data from metadata.xml
        Map<String, String> metaInfo = WorkflowV2Utility.parseMetaInfo(FileUtils.getFile(bundleDirectory));

        //Build the workflow model
        AbstractWorkflowDataModel.prepare(w);
        w.setMetadata_output_file_prefix(w.getConfigs().get("output_prefix"));
        w.setMetadata_output_dir(w.getConfigs().get("output_dir"));
        w.setName(metaInfo.get("name"));
        w.setVersion(metaInfo.get("workflow_version"));
        w.setRandom("" + new Random(System.currentTimeMillis()).nextInt(100000000)); //seqware's random int method
        w.setupDirectory();
        w.setupFiles();
        w.setupWorkflow();
        w.setupEnvironment();
        w.buildWorkflow();
        w.wrapup();
    }
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testCreateDataCarrier() throws IllegalAccessException {
    DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(5, 100);
    Assert.assertEquals(((Integer)(MemberModifier.field(DataCarrier.class, "bufferSize").get(carrier))).intValue(), 100);
    Assert.assertEquals(((Integer)(MemberModifier.field(DataCarrier.class, "channelSize").get(carrier))).intValue(), 5);

    Channels<SampleData> channels = (Channels<SampleData>)(MemberModifier.field(DataCarrier.class, "channels").get(carrier));
    Assert.assertEquals(channels.getChannelSize(), 5);

    Buffer<SampleData> buffer = channels.getBuffer(0);
    Assert.assertEquals(buffer.getBufferSize(), 100);

    Assert.assertEquals(MemberModifier.field(Buffer.class, "strategy").get(buffer), BufferStrategy.BLOCKING);
    carrier.setBufferStrategy(BufferStrategy.IF_POSSIBLE);
    Assert.assertEquals(MemberModifier.field(Buffer.class, "strategy").get(buffer), BufferStrategy.IF_POSSIBLE);

    Assert.assertEquals(MemberModifier.field(Channels.class, "dataPartitioner").get(channels).getClass(), SimpleRollingPartitioner.class);
    carrier.setPartitioner(new ProducerThreadPartitioner<SampleData>());
    Assert.assertEquals(MemberModifier.field(Channels.class, "dataPartitioner").get(channels).getClass(), ProducerThreadPartitioner.class);
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testProduce() throws IllegalAccessException {
    DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(2, 100);
    Assert.assertTrue(carrier.produce(new SampleData().setName("a")));
    Assert.assertTrue(carrier.produce(new SampleData().setName("b")));
    Assert.assertTrue(carrier.produce(new SampleData().setName("c")));
    Assert.assertTrue(carrier.produce(new SampleData().setName("d")));

    Channels<SampleData> channels = (Channels<SampleData>)(MemberModifier.field(DataCarrier.class, "channels").get(carrier));
    Buffer<SampleData> buffer1 = channels.getBuffer(0);
    List result1 = buffer1.obtain(0, 100);

    Buffer<SampleData> buffer2 = channels.getBuffer(1);
    List result2 = buffer2.obtain(0, 100);

    Assert.assertEquals(2, result1.size());
    Assert.assertEquals(4, result1.size() + result2.size());

}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testOverrideProduce() throws IllegalAccessException {
    DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(2, 100);
    carrier.setBufferStrategy(BufferStrategy.OVERRIDE);

    for (int i = 0; i < 500; i++) {
        Assert.assertTrue(carrier.produce(new SampleData().setName("d" + i)));
    }

    Channels<SampleData> channels = (Channels<SampleData>)(MemberModifier.field(DataCarrier.class, "channels").get(carrier));
    Buffer<SampleData> buffer1 = channels.getBuffer(0);
    List result1 = buffer1.obtain(0, 100);

    Buffer<SampleData> buffer2 = channels.getBuffer(1);
    List result2 = buffer2.obtain(0, 100);
    Assert.assertEquals(200, result1.size() + result2.size());
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testIfPossibleProduce() throws IllegalAccessException {
    DataCarrier<SampleData> carrier = new DataCarrier<SampleData>(2, 100);
    carrier.setBufferStrategy(BufferStrategy.IF_POSSIBLE);

    for (int i = 0; i < 200; i++) {
        Assert.assertTrue(carrier.produce(new SampleData().setName("d" + i)));
    }

    for (int i = 0; i < 200; i++) {
        Assert.assertFalse(carrier.produce(new SampleData().setName("d" + i + "_2")));
    }

    Channels<SampleData> channels = (Channels<SampleData>)(MemberModifier.field(DataCarrier.class, "channels").get(carrier));
    Buffer<SampleData> buffer1 = channels.getBuffer(0);
    List result1 = buffer1.obtain(0, 100);

    Buffer<SampleData> buffer2 = channels.getBuffer(1);
    List result2 = buffer2.obtain(0, 100);
    Assert.assertEquals(200, result1.size() + result2.size());
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
public void testPutAndGetConnectionInstance() throws Exception {
    ConnectionInfo key = mock(ConnectionInfo.class);
    ConnectionInfo connectionInfo = mock(ConnectionInfo.class);
    PowerMockito.mockStatic(SouthboundMapper.class);
    when(SouthboundMapper.suppressLocalIpPort(key)).thenReturn(connectionInfo);

    clients = new ConcurrentHashMap<>();
    MemberModifier.field(OvsdbConnectionManager.class, "clients").set(ovsdbConnManager, clients);

    // Test putConnectionInstance()
    OvsdbConnectionInstance instance = mock(OvsdbConnectionInstance.class);
    Whitebox.invokeMethod(ovsdbConnManager, "putConnectionInstance", key, instance);
    Map<ConnectionInfo, OvsdbConnectionInstance> testClients = Whitebox.getInternalState(ovsdbConnManager,
            "clients");
    assertEquals("Error, size of the hashmap is incorrect", 1, testClients.size());

    // Test getConnectionInstance(ConnectionInfo key)
    assertEquals("Error, returned incorrect OvsdbConnectionInstance object", instance,
            ovsdbConnManager.getConnectionInstance(key));
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
@SuppressWarnings({ "unchecked", "rawtypes" })
public void testCreateTransactInvokers() throws Exception {
    // transactInvokers not null case
    transactInvokers = new HashMap();
    field(OvsdbConnectionInstance.class, "transactInvokers").set(ovsdbConnectionInstance , transactInvokers);
    ovsdbConnectionInstance.createTransactInvokers();
    verify(ovsdbConnectionInstance, times(0)).getSchema(anyString());

    // transactInvokers null case
    MemberModifier.field(OvsdbConnectionInstance.class, "transactInvokers").set(ovsdbConnectionInstance , null);

    ListenableFuture<DatabaseSchema> listenableDbSchema = mock(ListenableFuture.class);
    DatabaseSchema dbSchema = mock(DatabaseSchema.class);
    doReturn(listenableDbSchema).when(ovsdbConnectionInstance).getSchema(anyString());
    when(listenableDbSchema.get()).thenReturn(dbSchema);

    ovsdbConnectionInstance.createTransactInvokers();
    verify(ovsdbConnectionInstance).getSchema(anyString());

    Map<DatabaseSchema, TransactInvoker> testTransactInvokers = Whitebox.getInternalState(ovsdbConnectionInstance,
            "transactInvokers");
    assertEquals("Error, size of the hashmap is incorrect", 1, testTransactInvokers.size());
}
 

import org.powermock.api.support.membermodification.MemberModifier; //导入依赖的package包/类
@Test
@SuppressWarnings("unchecked")
public void testSetOtherConfig() throws Exception {
    OpenVSwitch oldEntry = mock(OpenVSwitch.class);
    OpenVSwitch openVSwitch = mock(OpenVSwitch.class);

    Column<GenericTableSchema, Map<String, String>> column = mock(Column.class);
    when(openVSwitch.getOtherConfigColumn()).thenReturn(column);
    Map<String, String> map = new HashMap<>();
    when(column.getData()).thenReturn(map);
    when(oldEntry.getOtherConfigColumn()).thenReturn(column);
    MemberModifier.suppress(MemberMatcher.method(OpenVSwitchUpdateCommand.class, "removeOldConfigs",
            ReadWriteTransaction.class, Map.class, OpenVSwitch.class));
    MemberModifier.suppress(MemberMatcher.method(OpenVSwitchUpdateCommand.class, "setNewOtherConfigs",
            OvsdbNodeAugmentationBuilder.class, Map.class));

    ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
    OvsdbNodeAugmentationBuilder ovsdbNodeBuilder = mock(OvsdbNodeAugmentationBuilder.class);
    Whitebox.invokeMethod(openVSwitchUpdateCommand, "setOtherConfig",
            transaction, ovsdbNodeBuilder, oldEntry, openVSwitch);
    verify(openVSwitch, times(2)).getOtherConfigColumn();
    verify(oldEntry, times(2)).getOtherConfigColumn();
    PowerMockito.verifyPrivate(openVSwitchUpdateCommand).invoke("removeOldConfigs",
            any(ReadWriteTransaction.class),
            any(Map.class), any(OpenVSwitch.class));
}