How to use capture method of org.mockito.ArgumentCaptor class

Best Mockito code snippet using org.mockito.ArgumentCaptor.capture

Source:TestMsckCreatePartitionsInBatches.java

...141    // there should be 2 calls to create partitions with each batch size of 5142    ArgumentCaptor<Boolean> ifNotExistsArg = ArgumentCaptor.forClass(Boolean.class);143    ArgumentCaptor<Boolean> needResultsArg = ArgumentCaptor.forClass(Boolean.class);144    ArgumentCaptor<List<Partition>> argParts = ArgumentCaptor.forClass((Class) List.class);145    Mockito.verify(spyDb, Mockito.times(2)).add_partitions(argParts.capture(), ifNotExistsArg.capture(), needResultsArg.capture());146    // confirm the batch sizes were 5, 5 in the two calls to create partitions147    List<List<Partition>> apds = argParts.getAllValues();148    int retryAttempt = 1;149    Assert.assertEquals(String.format("Unexpected batch size in retry attempt %d ", retryAttempt++),150        5, apds.get(0).size());151    Assert.assertEquals(String.format("Unexpected batch size in retry attempt %d ", retryAttempt++),152        5, apds.get(1).size());153    assertTrue(ifNotExistsArg.getValue());154    assertFalse(needResultsArg.getValue());155  }156  /**157   * Tests the number of times Hive.createPartitions calls are executed with total number of158   * partitions to be added are not exactly divisible by batch size159   *160   * @throws Exception161   */162  @Test163  public void testUnevenNumberOfCreatePartitionCalls() throws Exception {164    // create 9 dummy partitions165    Set<PartitionResult> partsNotInMs = createPartsNotInMs(9);166    IMetaStoreClient spyDb = Mockito.spy(db);167    // batch size of 5 and decaying factor of 2168    msck.createPartitionsInBatches(spyDb, repairOutput, partsNotInMs, table, 5, 2, 0);169    // there should be 2 calls to create partitions with batch sizes of 5, 4170    ArgumentCaptor<Boolean> ifNotExistsArg = ArgumentCaptor.forClass(Boolean.class);171    ArgumentCaptor<Boolean> needResultsArg = ArgumentCaptor.forClass(Boolean.class);172    ArgumentCaptor<List<Partition>> argParts = ArgumentCaptor.forClass((Class) List.class);173    Mockito.verify(spyDb, Mockito.times(2)).add_partitions(argParts.capture(), ifNotExistsArg.capture(), needResultsArg.capture());174    // confirm the batch sizes were 5, 4 in the two calls to create partitions175    List<List<Partition>> apds = argParts.getAllValues();176    int retryAttempt = 1;177    Assert.assertEquals(String.format("Unexpected batch size in retry attempt %d ", retryAttempt++),178        5, apds.get(0).size());179    Assert.assertEquals(String.format("Unexpected batch size in retry attempt %d ", retryAttempt++),180        4, apds.get(1).size());181    assertTrue(ifNotExistsArg.getValue());182    assertFalse(needResultsArg.getValue());183  }184  /**185   * Tests the number of times Hive.createPartitions calls are executed with total number of186   * partitions exactly equal to batch size187   *188   * @throws Exception189   */190  @Test191  public void testEqualNumberOfPartitions() throws Exception {192    // create 13 dummy partitions193    Set<PartitionResult> partsNotInMs = createPartsNotInMs(13);194    IMetaStoreClient spyDb = Mockito.spy(db);195    // batch size of 13 and decaying factor of 2196    msck.createPartitionsInBatches(spyDb, repairOutput, partsNotInMs, table, 13, 2, 0);197    // there should be 1 call to create partitions with batch sizes of 13198    ArgumentCaptor<Boolean> ifNotExistsArg = ArgumentCaptor.forClass(Boolean.class);199    ArgumentCaptor<Boolean> needResultsArg = ArgumentCaptor.forClass(Boolean.class);200    ArgumentCaptor<List<Partition>> argParts = ArgumentCaptor.forClass((Class) List.class);201    // there should be 1 call to create partitions with batch sizes of 13202    Mockito.verify(spyDb, Mockito.times(1)).add_partitions(argParts.capture(), ifNotExistsArg.capture(),203      needResultsArg.capture());204    Assert.assertEquals("Unexpected number of batch size", 13,205        argParts.getValue().size());206    assertTrue(ifNotExistsArg.getValue());207    assertFalse(needResultsArg.getValue());208  }209  /**210   * Tests the number of times Hive.createPartitions calls are executed with total number of211   * partitions to is less than batch size212   *213   * @throws Exception214   */215  @Test216  public void testSmallNumberOfPartitions() throws Exception {217    // create 10 dummy partitions218    Set<PartitionResult> partsNotInMs = createPartsNotInMs(10);219    IMetaStoreClient spyDb = Mockito.spy(db);220    // batch size of 20 and decaying factor of 2221    msck.createPartitionsInBatches(spyDb, repairOutput, partsNotInMs, table, 20, 2, 0);222    // there should be 1 call to create partitions with batch sizes of 10223    Mockito.verify(spyDb, Mockito.times(1)).add_partitions(Mockito.anyObject(), Mockito.anyBoolean(),224      Mockito.anyBoolean());225    ArgumentCaptor<Boolean> ifNotExistsArg = ArgumentCaptor.forClass(Boolean.class);226    ArgumentCaptor<Boolean> needResultsArg = ArgumentCaptor.forClass(Boolean.class);227    ArgumentCaptor<List<Partition>> argParts = ArgumentCaptor.forClass((Class) List.class);228    // there should be 1 call to create partitions with batch sizes of 10229    Mockito.verify(spyDb, Mockito.times(1)).add_partitions(argParts.capture(), ifNotExistsArg.capture(),230      needResultsArg.capture());231    Assert.assertEquals("Unexpected number of batch size", 10,232        argParts.getValue().size());233    assertTrue(ifNotExistsArg.getValue());234    assertFalse(needResultsArg.getValue());235  }236  /**237   * Tests the number of calls to createPartitions and the respective batch sizes when first call to238   * createPartitions throws HiveException. The batch size should be reduced by the decayingFactor239   *240   * @throws Exception241   */242  @Test243  public void testBatchingWhenException() throws Exception {244    // create 13 dummy partitions245    Set<PartitionResult> partsNotInMs = createPartsNotInMs(23);246    IMetaStoreClient spyDb = Mockito.spy(db);247    // first call to createPartitions should throw exception248    Mockito.doThrow(HiveException.class).doCallRealMethod().doCallRealMethod().when(spyDb)249      .add_partitions(Mockito.anyObject(), Mockito.anyBoolean(),250        Mockito.anyBoolean());251    // test with a batch size of 30 and decaying factor of 2252    msck.createPartitionsInBatches(spyDb, repairOutput, partsNotInMs, table, 30, 2, 0);253    // confirm the batch sizes were 23, 15, 8 in the three calls to create partitions254    ArgumentCaptor<Boolean> ifNotExistsArg = ArgumentCaptor.forClass(Boolean.class);255    ArgumentCaptor<Boolean> needResultsArg = ArgumentCaptor.forClass(Boolean.class);256    ArgumentCaptor<List<Partition>> argParts = ArgumentCaptor.forClass((Class) List.class);257    // there should be 3 calls to create partitions with batch sizes of 23, 15, 8258    Mockito.verify(spyDb, Mockito.times(3)).add_partitions(argParts.capture(), ifNotExistsArg.capture(),259      needResultsArg.capture());260    List<List<Partition>> apds = argParts.getAllValues();261    int retryAttempt = 1;262    Assert.assertEquals(263        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 23,264        apds.get(0).size());265    Assert.assertEquals(266        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 15,267        apds.get(1).size());268    Assert.assertEquals(269        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 8,270        apds.get(2).size());271    assertTrue(ifNotExistsArg.getValue());272    assertFalse(needResultsArg.getValue());273  }274  /**275   * Tests the retries exhausted case when Hive.createPartitions method call always keep throwing276   * HiveException. The batch sizes should exponentially decreased based on the decaying factor and277   * ultimately give up when it reaches 0278   *279   * @throws Exception280   */281  @Test282  public void testRetriesExhaustedBatchSize() throws Exception {283    Set<PartitionResult> partsNotInMs = createPartsNotInMs(17);284    IMetaStoreClient spyDb = Mockito.spy(db);285    Mockito.doThrow(HiveException.class).when(spyDb)286      .add_partitions(Mockito.anyObject(), Mockito.anyBoolean(), Mockito.anyBoolean());287    // batch size of 5 and decaying factor of 2288    Exception ex = null;289    try {290      msck.createPartitionsInBatches(spyDb, repairOutput, partsNotInMs, table, 30, 2, 0);291    } catch (Exception retryEx) {292      ex = retryEx;293    }294    assertFalse("Exception was expected but was not thrown", ex == null);295    Assert.assertTrue("Unexpected class of exception thrown", ex instanceof RetryUtilities.RetryException);296    // there should be 5 calls to create partitions with batch sizes of 17, 15, 7, 3, 1297    ArgumentCaptor<Boolean> ifNotExistsArg = ArgumentCaptor.forClass(Boolean.class);298    ArgumentCaptor<Boolean> needResultsArg = ArgumentCaptor.forClass(Boolean.class);299    ArgumentCaptor<List<Partition>> argParts = ArgumentCaptor.forClass((Class) List.class);300    // there should be 5 calls to create partitions with batch sizes of 17, 15, 7, 3, 1301    Mockito.verify(spyDb, Mockito.times(5)).add_partitions(argParts.capture(), ifNotExistsArg.capture(),302      needResultsArg.capture());303    List<List<Partition>> apds = argParts.getAllValues();304    int retryAttempt = 1;305    Assert.assertEquals(306        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 17,307        apds.get(0).size());308    Assert.assertEquals(309        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 15,310        apds.get(1).size());311    Assert.assertEquals(312        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 7,313        apds.get(2).size());314    Assert.assertEquals(315        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 3,316        apds.get(3).size());317    Assert.assertEquals(318        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 1,319        apds.get(4).size());320    assertTrue(ifNotExistsArg.getValue());321    assertFalse(needResultsArg.getValue());322  }323  /**324   * Tests the maximum retry attempts provided by configuration325   * @throws Exception326   */327  @Test328  public void testMaxRetriesReached() throws Exception {329    Set<PartitionResult> partsNotInMs = createPartsNotInMs(17);330    IMetaStoreClient spyDb = Mockito.spy(db);331    Mockito.doThrow(HiveException.class).when(spyDb)332      .add_partitions(Mockito.anyObject(), Mockito.anyBoolean(), Mockito.anyBoolean());333    // batch size of 5 and decaying factor of 2334    Exception ex = null;335    try {336      msck.createPartitionsInBatches(spyDb, repairOutput, partsNotInMs, table, 30, 2, 2);337    } catch (Exception retryEx) {338      ex = retryEx;339    }340    assertFalse("Exception was expected but was not thrown", ex == null);341    Assert.assertTrue("Unexpected class of exception thrown", ex instanceof RetryUtilities.RetryException);342    ArgumentCaptor<Boolean> ifNotExistsArg = ArgumentCaptor.forClass(Boolean.class);343    ArgumentCaptor<Boolean> needResultsArg = ArgumentCaptor.forClass(Boolean.class);344    ArgumentCaptor<List<Partition>> argParts = ArgumentCaptor.forClass((Class) List.class);345    Mockito.verify(spyDb, Mockito.times(2)).add_partitions(argParts.capture(), ifNotExistsArg.capture(), needResultsArg.capture());346    List<List<Partition>> apds = argParts.getAllValues();347    int retryAttempt = 1;348    Assert.assertEquals(349        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 17,350        apds.get(0).size());351    Assert.assertEquals(352        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 15,353        apds.get(1).size());354    assertTrue(ifNotExistsArg.getValue());355    assertFalse(needResultsArg.getValue());356  }357  /**358   * Tests when max number of retries is set to 1. In this case the number of retries should359   * be specified360   * @throws Exception361   */362  @Test363  public void testOneMaxRetries() throws Exception {364    Set<PartitionResult> partsNotInMs = createPartsNotInMs(17);365    IMetaStoreClient spyDb = Mockito.spy(db);366    Mockito.doThrow(HiveException.class).when(spyDb)367      .add_partitions(Mockito.anyObject(), Mockito.anyBoolean(), Mockito.anyBoolean());368    // batch size of 5 and decaying factor of 2369    Exception ex = null;370    try {371      msck.createPartitionsInBatches(spyDb, repairOutput, partsNotInMs, table, 30, 2, 1);372    } catch (Exception retryEx) {373      ex = retryEx;374    }375    assertFalse("Exception was expected but was not thrown", ex == null);376    Assert.assertTrue("Unexpected class of exception thrown", ex instanceof RetryUtilities.RetryException);377    // there should be 5 calls to create partitions with batch sizes of 17, 15, 7, 3, 1378    ArgumentCaptor<Boolean> ifNotExistsArg = ArgumentCaptor.forClass(Boolean.class);379    ArgumentCaptor<Boolean> needResultsArg = ArgumentCaptor.forClass(Boolean.class);380    ArgumentCaptor<List<Partition>> argParts = ArgumentCaptor.forClass((Class) List.class);381    // there should be 5 calls to create partitions with batch sizes of 17, 15, 7, 3, 1382    Mockito.verify(spyDb, Mockito.times(1)).add_partitions(argParts.capture(), ifNotExistsArg.capture(),383      needResultsArg.capture());384    List<List<Partition>> apds = argParts.getAllValues();385    int retryAttempt = 1;386    Assert.assertEquals(387        String.format("Unexpected batch size in retry attempt %d ", retryAttempt++), 17,388        apds.get(0).size());389    assertTrue(ifNotExistsArg.getValue());390    assertFalse(needResultsArg.getValue());391  }392}...

Source:CheckpointSpoutTest.java

...52        Values expectedTuple = new Values(-1L, Action.INITSTATE);53        ArgumentCaptor<String> stream = ArgumentCaptor.forClass(String.class);54        ArgumentCaptor<Values> values = ArgumentCaptor.forClass(Values.class);55        ArgumentCaptor<Object> msgId = ArgumentCaptor.forClass(Object.class);56        Mockito.verify(mockOutputCollector).emit(stream.capture(),57                                                 values.capture(),58                                                 msgId.capture());59        assertEquals(CheckpointSpout.CHECKPOINT_STREAM_ID, stream.getValue());60        assertEquals(expectedTuple, values.getValue());61        assertEquals(-1L, msgId.getValue());62        spout.ack(-1L);63        Mockito.verify(mockOutputCollector).emit(stream.capture(),64                                                 values.capture(),65                                                 msgId.capture());66        expectedTuple = new Values(-1L, Action.INITSTATE);67        assertEquals(CheckpointSpout.CHECKPOINT_STREAM_ID, stream.getValue());68        assertEquals(expectedTuple, values.getValue());69        assertEquals(-1L, msgId.getValue());70    }71    @Test72    public void testPrepare() throws Exception {73        spout.open(new HashMap(), mockTopologyContext, mockOutputCollector);74        ArgumentCaptor<String> stream = ArgumentCaptor.forClass(String.class);75        ArgumentCaptor<Values> values = ArgumentCaptor.forClass(Values.class);76        ArgumentCaptor<Object> msgId = ArgumentCaptor.forClass(Object.class);77        spout.nextTuple();78        spout.ack(-1L);79        spout.nextTuple();80        Mockito.verify(mockOutputCollector, Mockito.times(2)).emit(stream.capture(),81                                                 values.capture(),82                                                 msgId.capture());83        Values expectedTuple = new Values(0L, Action.PREPARE);84        assertEquals(CheckpointSpout.CHECKPOINT_STREAM_ID, stream.getValue());85        assertEquals(expectedTuple, values.getValue());86        assertEquals(0L, msgId.getValue());87    }88    @Test89    public void testPrepareWithFail() throws Exception {90        Map<String, Object> stormConf = new HashMap<>();91        KeyValueState<String, CheckPointState> state =92                (KeyValueState<String, CheckPointState>) StateFactory.getState("__state", stormConf, mockTopologyContext);93        CheckPointState txState = new CheckPointState(-1, COMMITTED);94        state.put("__state", txState);95        spout.open(mockTopologyContext, mockOutputCollector, 0, state);96        ArgumentCaptor<String> stream = ArgumentCaptor.forClass(String.class);97        ArgumentCaptor<Values> values = ArgumentCaptor.forClass(Values.class);98        ArgumentCaptor<Object> msgId = ArgumentCaptor.forClass(Object.class);99        spout.nextTuple();100        spout.ack(-1L);101        Utils.sleep(10);102        spout.nextTuple();103        spout.ack(0L);104        Utils.sleep(10);105        spout.nextTuple();106        spout.ack(0L);107        Utils.sleep(10);108        spout.nextTuple();109        spout.fail(1L);110        Utils.sleep(10);111        spout.nextTuple();112        spout.fail(1L);113        Utils.sleep(10);114        spout.nextTuple();115        spout.ack(1L);116        Utils.sleep(10);117        spout.nextTuple();118        spout.ack(0L);119        Utils.sleep(10);120        spout.nextTuple();121        Mockito.verify(mockOutputCollector, Mockito.times(8)).emit(stream.capture(),122                                                                   values.capture(),123                                                                   msgId.capture());124        Values expectedTuple = new Values(1L, Action.PREPARE);125        assertEquals(CheckpointSpout.CHECKPOINT_STREAM_ID, stream.getValue());126        assertEquals(expectedTuple, values.getValue());127        assertEquals(1L, msgId.getValue());128    }129    @Test130    public void testCommit() throws Exception {131        Map<String, Object> stormConf = new HashMap();132        stormConf.put(Config.TOPOLOGY_STATE_CHECKPOINT_INTERVAL, 0);133        spout.open(stormConf, mockTopologyContext, mockOutputCollector);134        ArgumentCaptor<String> stream = ArgumentCaptor.forClass(String.class);135        ArgumentCaptor<Values> values = ArgumentCaptor.forClass(Values.class);136        ArgumentCaptor<Object> msgId = ArgumentCaptor.forClass(Object.class);137        spout.nextTuple();138        spout.ack(-1L);139        spout.nextTuple();140        spout.ack(0L);141        Utils.sleep(10);142        spout.nextTuple();143        spout.fail(0L);144        Utils.sleep(10);145        spout.nextTuple();146        Mockito.verify(mockOutputCollector, Mockito.times(4)).emit(stream.capture(),147                                                                   values.capture(),148                                                                   msgId.capture());149        Values expectedTuple = new Values(0L, Action.COMMIT);150        assertEquals(CheckpointSpout.CHECKPOINT_STREAM_ID, stream.getValue());151        assertEquals(expectedTuple, values.getValue());152        assertEquals(0L, msgId.getValue());153    }154    @Test155    public void testRecoveryRollback() throws Exception {156        Map<String, Object> stormConf = new HashMap();157        KeyValueState<String, CheckPointState> state =158                (KeyValueState<String, CheckPointState>) StateFactory.getState("test-1", stormConf, mockTopologyContext);159        CheckPointState checkPointState = new CheckPointState(100, CheckPointState.State.PREPARING);160        state.put("__state", checkPointState);161        spout.open(mockTopologyContext, mockOutputCollector, 0, state);162        ArgumentCaptor<String> stream = ArgumentCaptor.forClass(String.class);163        ArgumentCaptor<Values> values = ArgumentCaptor.forClass(Values.class);164        ArgumentCaptor<Object> msgId = ArgumentCaptor.forClass(Object.class);165        spout.nextTuple();166        Mockito.verify(mockOutputCollector, Mockito.times(1)).emit(stream.capture(),167                                                                   values.capture(),168                                                                   msgId.capture());169        Values expectedTuple = new Values(100L, Action.ROLLBACK);170        assertEquals(CheckpointSpout.CHECKPOINT_STREAM_ID, stream.getValue());171        assertEquals(expectedTuple, values.getValue());172        assertEquals(100L, msgId.getValue());173    }174    @Test175    public void testRecoveryRollbackAck() throws Exception {176        Map<String, Object> stormConf = new HashMap();177        KeyValueState<String, CheckPointState> state =178                (KeyValueState<String, CheckPointState>) StateFactory.getState("test-1", stormConf, mockTopologyContext);179        CheckPointState checkPointState = new CheckPointState(100, CheckPointState.State.PREPARING);180        state.put("__state", checkPointState);181        spout.open(mockTopologyContext, mockOutputCollector, 0, state);182        ArgumentCaptor<String> stream = ArgumentCaptor.forClass(String.class);183        ArgumentCaptor<Values> values = ArgumentCaptor.forClass(Values.class);184        ArgumentCaptor<Object> msgId = ArgumentCaptor.forClass(Object.class);185        spout.nextTuple();186        spout.ack(100L);187        spout.nextTuple();188        spout.ack(99L);189        spout.nextTuple();190        Mockito.verify(mockOutputCollector, Mockito.times(3)).emit(stream.capture(),191                                                                   values.capture(),192                                                                   msgId.capture());193        Values expectedTuple = new Values(100L, Action.PREPARE);194        assertEquals(CheckpointSpout.CHECKPOINT_STREAM_ID, stream.getValue());195        assertEquals(expectedTuple, values.getValue());196        assertEquals(100L, msgId.getValue());197    }198    @Test199    public void testRecoveryCommit() throws Exception {200        Map<String, Object> stormConf = new HashMap();201        KeyValueState<String, CheckPointState> state =202                (KeyValueState<String, CheckPointState>) StateFactory.getState("test-1", stormConf, mockTopologyContext);203        CheckPointState checkPointState = new CheckPointState(100, CheckPointState.State.COMMITTING);204        state.put("__state", checkPointState);205        spout.open(mockTopologyContext, mockOutputCollector, 0, state);206        ArgumentCaptor<String> stream = ArgumentCaptor.forClass(String.class);207        ArgumentCaptor<Values> values = ArgumentCaptor.forClass(Values.class);208        ArgumentCaptor<Object> msgId = ArgumentCaptor.forClass(Object.class);209        spout.nextTuple();210        Mockito.verify(mockOutputCollector, Mockito.times(1)).emit(stream.capture(),211                                                                   values.capture(),212                                                                   msgId.capture());213        Values expectedTuple = new Values(100L, Action.COMMIT);214        assertEquals(CheckpointSpout.CHECKPOINT_STREAM_ID, stream.getValue());215        assertEquals(expectedTuple, values.getValue());216        assertEquals(100L, msgId.getValue());217    }218}...

Source:CallbackHandlerTest.java

...50    public void testEmergencyListener() {51        mHandler.setEmergencyCallsOnly(true);52        waitForCallbacks();53        ArgumentCaptor<Boolean> captor = ArgumentCaptor.forClass(Boolean.class);54        Mockito.verify(mEmengencyListener).setEmergencyCallsOnly(captor.capture());55        assertTrue(captor.getValue());56    }57    public void testSignalCallback_setWifiIndicators() {58        boolean enabled = true;59        IconState status = new IconState(true, 0, "");60        IconState qs = new IconState(true, 1, "");61        boolean in = true;62        boolean out = true;63        String description = "Test";64        mHandler.setWifiIndicators(enabled, status, qs, in, out, description);65        waitForCallbacks();66        ArgumentCaptor<Boolean> enableArg = ArgumentCaptor.forClass(Boolean.class);67        ArgumentCaptor<IconState> statusArg = ArgumentCaptor.forClass(IconState.class);68        ArgumentCaptor<IconState> qsArg = ArgumentCaptor.forClass(IconState.class);69        ArgumentCaptor<Boolean> inArg = ArgumentCaptor.forClass(Boolean.class);70        ArgumentCaptor<Boolean> outArg = ArgumentCaptor.forClass(Boolean.class);71        ArgumentCaptor<String> descArg = ArgumentCaptor.forClass(String.class);72        Mockito.verify(mSignalCallback).setWifiIndicators(enableArg.capture(),73                statusArg.capture(), qsArg.capture(), inArg.capture(), outArg.capture(),74                descArg.capture());75        assertEquals(enabled, (boolean) enableArg.getValue());76        assertEquals(status, statusArg.getValue());77        assertEquals(qs, qsArg.getValue());78        assertEquals(in, (boolean) inArg.getValue());79        assertEquals(out, (boolean) outArg.getValue());80        assertEquals(description, descArg.getValue());81    }82    public void testSignalCallback_setMobileDataIndicators() {83        IconState status = new IconState(true, 0, "");84        IconState qs = new IconState(true, 1, "");85        boolean in = true;86        boolean out = true;87        String typeDescription = "Test 1";88        String description = "Test 2";89        int type = R.drawable.stat_sys_data_fully_connected_1x;90        int qsType = R.drawable.ic_qs_signal_1x;91        boolean wide = true;92        int subId = 5;93        boolean roaming = true;94        mHandler.setMobileDataIndicators(status, qs, type, qsType, in, out, typeDescription,95                description, wide, subId, roaming);96        waitForCallbacks();97        ArgumentCaptor<IconState> statusArg = ArgumentCaptor.forClass(IconState.class);98        ArgumentCaptor<IconState> qsArg = ArgumentCaptor.forClass(IconState.class);99        ArgumentCaptor<Integer> typeIconArg = ArgumentCaptor.forClass(Integer.class);100        ArgumentCaptor<Integer> qsTypeIconArg = ArgumentCaptor.forClass(Integer.class);101        ArgumentCaptor<Boolean> inArg = ArgumentCaptor.forClass(Boolean.class);102        ArgumentCaptor<Boolean> outArg = ArgumentCaptor.forClass(Boolean.class);103        ArgumentCaptor<String> typeContentArg = ArgumentCaptor.forClass(String.class);104        ArgumentCaptor<String> descArg = ArgumentCaptor.forClass(String.class);105        ArgumentCaptor<Boolean> wideArg = ArgumentCaptor.forClass(Boolean.class);106        ArgumentCaptor<Integer> subIdArg = ArgumentCaptor.forClass(Integer.class);107        Mockito.verify(mSignalCallback).setMobileDataIndicators(statusArg.capture(),108                qsArg.capture(), typeIconArg.capture(), qsTypeIconArg.capture(), inArg.capture(),109                outArg.capture(), typeContentArg.capture(), descArg.capture(), wideArg.capture(),110                subIdArg.capture(), eq(roaming));111        assertEquals(status, statusArg.getValue());112        assertEquals(qs, qsArg.getValue());113        assertEquals(type, (int) typeIconArg.getValue());114        assertEquals(qsType, (int) qsTypeIconArg.getValue());115        assertEquals(in, (boolean) inArg.getValue());116        assertEquals(out, (boolean) outArg.getValue());117        assertEquals(typeDescription, typeContentArg.getValue());118        assertEquals(description, descArg.getValue());119        assertEquals(wide, (boolean) wideArg.getValue());120        assertEquals(subId, (int) subIdArg.getValue());121    }122    @SuppressWarnings("unchecked")123    public void testSignalCallback_setSubs() {124        List<SubscriptionInfo> subs = new ArrayList<>();125        mHandler.setSubs(subs);126        waitForCallbacks();127        ArgumentCaptor<ArrayList> subsArg = ArgumentCaptor.forClass(ArrayList.class);128        Mockito.verify(mSignalCallback).setSubs(subsArg.capture());129        assertTrue(subs == subsArg.getValue());130    }131    public void testSignalCallback_setNoSims() {132        boolean noSims = true;133        mHandler.setNoSims(noSims);134        waitForCallbacks();135        ArgumentCaptor<Boolean> noSimsArg = ArgumentCaptor.forClass(Boolean.class);136        Mockito.verify(mSignalCallback).setNoSims(noSimsArg.capture());137        assertEquals(noSims, (boolean) noSimsArg.getValue());138    }139    public void testSignalCallback_setEthernetIndicators() {140        IconState state = new IconState(true, R.drawable.stat_sys_ethernet, "Test Description");141        mHandler.setEthernetIndicators(state);142        waitForCallbacks();143        ArgumentCaptor<IconState> iconArg = ArgumentCaptor.forClass(IconState.class);144        Mockito.verify(mSignalCallback).setEthernetIndicators(iconArg.capture());145        assertEquals(state, iconArg.getValue());146    }147    public void testSignalCallback_setIsAirplaneMode() {148        IconState state = new IconState(true, R.drawable.stat_sys_airplane_mode, "Test Description");149        mHandler.setIsAirplaneMode(state);150        waitForCallbacks();151        ArgumentCaptor<IconState> iconArg = ArgumentCaptor.forClass(IconState.class);152        Mockito.verify(mSignalCallback).setIsAirplaneMode(iconArg.capture());153        assertEquals(state, iconArg.getValue());154    }155    private void waitForCallbacks() {156        mHandlerThread.quitSafely();157        try {158            mHandlerThread.join();159        } catch (InterruptedException e) {160        }161    }162}...

capture

Using AI Code Generation

1package com.automationrhapsody.mockito;2import static org.junit.Assert.assertEquals;3import static org.mockito.Mockito.mock;4import static org.mockito.Mockito.verify;5import java.util.List;6import org.junit.Test;7import org.mockito.ArgumentCaptor;8public class ArgumentCaptorTest {9    public void testCapture() {10        List<String> mockedList = mock(List.class);11        mockedList.add("one");12        mockedList.add("two");13        mockedList.add("three");14        ArgumentCaptor<String> argument = ArgumentCaptor.forClass(String.class);15        verify(mockedList).add(argument.capture());16        assertEquals("three", argument.getValue());17    }18}191. -> at com.automationrhapsody.mockito.ArgumentCaptorTest.testCapture(ArgumentCaptorTest.java:21)20mockedList.add(21);22-> at com.automationrhapsody.mockito.ArgumentCaptorTest.testCapture(ArgumentCaptorTest.java:21)23	at com.automationrhapsody.mockito.ArgumentCaptorTest.testCapture(ArgumentCaptorTest.java:21)24To resolve this, we need to add another line of code after the verify() method:25assertEquals("three", argument.getValue());

capture

Using AI Code Generation

1package com.ack.j2se.mock;2import org.junit.Test;3import org.mockito.ArgumentCaptor;4import java.util.List;5import static org.junit.Assert.assertEquals;6import static org.mockito.Mockito.mock;7import static org.mockito.Mockito.verify;8public class CaptureTest {9  public void capture() {10    List mockedList = mock( List.class );11    mockedList.add( "one" );12    mockedList.clear();13    ArgumentCaptor argumentCaptor = ArgumentCaptor.forClass( String.class );14    verify( mockedList ).add( argumentCaptor.capture() );15    assertEquals( "one", argumentCaptor.getValue() );16  }17}18package com.ack.j2se.mock;19import org.junit.Test;20import org.mockito.ArgumentCaptor;21import java.util.List;22import static org.junit.Assert.assertEquals;23import static org.mockito.Mockito.mock;24import static org.mockito.Mockito.verify;25public class CaptureTest {26  public void capture() {27    List mockedList = mock( List.class );28    mockedList.add( "one" );29    mockedList.clear();30    ArgumentCaptor argumentCaptor = ArgumentCaptor.forClass( String.class );31    verify( mockedList ).add( argumentCaptor.capture() );32    assertEquals( "one", argumentCaptor.getValue() );33  }34}35package com.ack.j2se.mock;36import org.junit.Test;37import org.mockito.ArgumentCaptor;38import java.util.List;39import static org.junit.Assert.assertEquals;40import static org.mockito.Mockito.mock;41import static org.mockito.Mockito.verify;42public class CaptureTest {43  public void capture() {44    List mockedList = mock( List.class );45    mockedList.add( "one" );46    mockedList.clear();47    ArgumentCaptor argumentCaptor = ArgumentCaptor.forClass( String.class );48    verify( mockedList ).add( argumentCaptor.capture() );49    assertEquals( "one", argumentCaptor.getValue() );50  }51}52package com.ack.j2se.mock;53import org.junit.Test;54import org.mockito.ArgumentCaptor;55import java.util.List;56import static org.junit.Assert.assertEquals;57import static org.mockito.Mockito.mock

capture

Using AI Code Generation

1import java.util.List;2import org.mockito.ArgumentCaptor;3import org.mockito.Captor;4import org.mockito.InjectMocks;5import org.mockito.Mock;6import org.mockito.MockitoAnnotations;7import org.mockito.Spy;8import org.testng.annotations.BeforeMethod;9import org.testng.annotations.Test;10import static org.mockito.Mockito.*;11import static org.testng.Assert.*;12public class MockitoCaptureTest {13    private List mockedList;14    private ArgumentCaptor argCaptor;15    private MyService service;16    public void setUp() {17        MockitoAnnotations.initMocks(this);18    }19    public void testCapture() {20        service.add("one");21        verify(mockedList).add(argCaptor.capture());22        assertEquals(argCaptor.getValue(), "one");23    }24}25import com.googlecode.catchexception.CatchException;26import com.googlecode.catchexception.apis.CatchExceptionHamcrestMatchers;27import com.googlecode.catchexception.apis.CatchExceptionJunitMatchers;28import com.googlecode.catchexception.apis.CatchExceptionMatchers;29import com.googlecode.catchexception.apis.CatchExceptionMockitoMatchers;30import com.googlecode.catchexception.apis.CatchExceptionPowermockMatchers;31import com.googlecode.catchexception.apis.CatchExceptionTestngMatchers;32import com.googlecode.catchexception.apis.CatchExceptionVerify;33import com.googlecode.catchexception.apis.CatchExceptionVerifyNoMoreInteractions;34import com.googlecode.catchexception.apis.CatchExceptionVerifyZeroInteractions;35import com.googlecode.catchexception.apis.DoNothing;36import com.googlecode.catchexception.apis.DoReturn;37import com.googlecode.catchexception.apis.DoThrow;38import com.googlecode.catchexception.apis.MockitoMatchers;39import com.googlecode.catchexception.apis.MockitoMatchersHamcrest;40import com.googlecode.catchexception.apis.MockitoMatchersJunit;41import com.googlecode.catchexception.apis.MockitoMatchersTestng;42import com.googlecode.catchexception.apis.MockitoMatchersVerify;43import com.googlecode.catchexception.apis.MockitoMatchersVerifyNoMoreInteractions;44import com.googlecode.catchexception.apis.MockitoMatchersVerifyZeroInteractions;45import com.googlecode.catchexception.apis.PowermockitoMatchers;46import com.googlecode.catchexception.apis.PowermockitoMatchersHamcrest;47import com.googlecode.catchexception.apis.PowermockitoMatchersJunit;48import com.googlecode.catchexception.apis.PowermockitoMatchersTestng;49import com.googlecode.catchexception.apis.Power

capture

Using AI Code Generation

1package org.example;2import java.util.List;3import org.junit.After;4import org.junit.Before;5import org.junit.Test;6import org.mockito.ArgumentCaptor;7import org.mockito.Mockito;8import static org.junit.Assert.assertEquals;9import static org.mockito.Mockito.*;10public class CaptureTest {11    private List mockedList;12    private ArgumentCaptor<String> argumentCaptor;13    public void setUp() {14        mockedList = mock(List.class);15        argumentCaptor = ArgumentCaptor.forClass(String.class);16    }17    public void tearDown() {18        mockedList = null;19        argumentCaptor = null;20    }21    public void testCapture() {22        mockedList.add("one");23        mockedList.add("two");24        verify(mockedList).add(argumentCaptor.capture());25        assertEquals("two", argumentCaptor.getValue());26    }27}28BUILD SUCCESSFUL (total time: 0 seconds)29Mockito ArgumentCaptor.capture() Method30Mockito ArgumentCaptor.getValue() Method31Mockito ArgumentCaptor.getAllValues() Method32Mockito ArgumentCaptor.forClass() Method33Mockito ArgumentCaptor.capture() Method34Mockito ArgumentCaptor.getValue() Method35Mockito ArgumentCaptor.getAllValues() Method36Mockito ArgumentCaptor.forClass() Method37Mockito ArgumentCaptor.capture() Method38Mockito ArgumentCaptor.getValue() Method39Mockito ArgumentCaptor.getAllValues() Method40Mockito ArgumentCaptor.forClass() Method41Mockito ArgumentCaptor.capture() Method42Mockito ArgumentCaptor.getValue() Method43Mockito ArgumentCaptor.getAllValues() Method44Mockito ArgumentCaptor.forClass() Method45Mockito ArgumentCaptor.capture() Method46Mockito ArgumentCaptor.getValue() Method47Mockito ArgumentCaptor.getAllValues() Method48Mockito ArgumentCaptor.forClass() Method49Mockito ArgumentCaptor.capture() Method50Mockito ArgumentCaptor.getValue() Method51Mockito ArgumentCaptor.getAllValues() Method52Mockito ArgumentCaptor.forClass() Method53Mockito ArgumentCaptor.capture() Method54Mockito ArgumentCaptor.getValue() Method55Mockito ArgumentCaptor.getAllValues() Method56Mockito ArgumentCaptor.forClass() Method57Mockito ArgumentCaptor.capture() Method58Mockito ArgumentCaptor.getValue() Method59Mockito ArgumentCaptor.getAllValues() Method60Mockito ArgumentCaptor.forClass() Method

capture

Using AI Code Generation

1import org.mockito.ArgumentCaptor;2import org.mockito.Mockito;3import org.mockito.stubbing.Answer;4import static org.mockito.Mockito.*;5import java.util.List;6public class MockitoCaptureDemo {7    public static void main(String[] args) {8        List<String> mockedList = mock(List.class);9        ArgumentCaptor<String> argument = ArgumentCaptor.forClass(String.class);10        mockedList.add("one");11        verify(mockedList).add(argument.capture());12        mockedList.add("two");13        verify(mockedList).add(argument.capture());14        Mockito.verifyNoMoreInteractions(mockedList);15    }16}

capture

Using AI Code Generation

1import org.mockito.ArgumentCaptor;2import org.mockito.Mockito;3import org.mockito.stubbing.Answer;4import org.mockito.invocation.InvocationOnMock;5import static org.mockito.Mockito.*;6import java.util.*;7public class CaptureMethod {8 public static void main(String[] args) {9  List mockedList = mock(List.class);10  mockedList.add("one");11  mockedList.clear();12  verify(mockedList).add("one");13  verify(mockedList).clear();14  verify(mockedList, times(2)).add("one");15  verify(mockedList, never()).add("two");16  verify(mockedList, timeout(100)).add("one");17  verify(mockedList, timeout(100).times(2)).add("one");18  verify(mockedList, timeout(100)).add("one");19  verify(mockedList, timeout(100).times(2)).add("one");20  ArgumentCaptor<String> argument = ArgumentCaptor.forClass(String.class);21  verify(mockedList).add(argument.capture());22  assertEquals("one", argument.getValue());23 }24}25mockedList.add("one");26-> at CaptureMethod.main(CaptureMethod.java:30)27import org.mockito.ArgumentCaptor;28import org.mockito.Mockito;29import org.mockito.stubbing.Answer;30import org.mockito.invocation.InvocationOnMock;31import static org.mockito.Mockito.*;32import java.util.*;33public class CaptureMethod {34 public static void main(String[] args) {35  List mockedList = mock(List.class);36  mockedList.add("one");37  mockedList.clear();38  verify(mockedList).add("one");39  verify(mockedList).clear();40  verify(mockedList, times(2)).add("one");41  verify(mockedList, never()).add("two");42  verify(mockedList, timeout(100)).add("one");43  verify(mockedList, timeout(100).times(2)).add

capture

Using AI Code Generation

1package mockitodemo;2import static org.junit.Assert.*;3import org.junit.Test;4import org.mockito.ArgumentCaptor;5import org.mockito.Mockito;6public class Test1 {7   public void testCapture() {8      MyInterface mock = Mockito.mock(MyInterface.class);9      Mockito.when(mock.doSomething("test")).thenReturn(10);10      mock.doSomething("test");11      ArgumentCaptor.forClass(String.class);12      Mockito.verify(mock).doSomething(arg.capture());13      assertEquals("test", arg.getValue());14   }15}16package mockitodemo;17import static org.junit.Assert.*;18import org.junit.Test;19import org.mockito.ArgumentCaptor;20import org.mockito.Mockito;21public class Test2 {22   public void testCapture() {23      MyInterface mock = Mockito.mock(MyInterface.class);24      Mockito.when(mock.doSomething("test")).thenReturn(10);25      mock.doSomething("test");26      ArgumentCaptor.forClass(String.class);27      Mockito.verify(mock).doSomething(arg.capture());28      assertEquals("test", arg.getValue());29   }30}31package mockitodemo;32import static org.junit.Assert.*;33import org.junit.Test;34import org.mockito.ArgumentCaptor;35import org.mockito.Mockito;36public class Test3 {37   public void testCapture() {38      MyInterface mock = Mockito.mock(MyInterface.class);39      Mockito.when(mock.doSomething("test")).thenReturn(10);40      mock.doSomething("test");41      ArgumentCaptor.forClass(String.class);42      Mockito.verify(mock).doSomething(arg.capture());

capture

Using AI Code Generation

1import org.junit.Test;2import org.junit.runner.RunWith;3import org.mockito.ArgumentCaptor;4import org.mockito.Mock;5import org.mockito.runners.MockitoJUnitRunner;6import static org.mockito.Mockito.*;7import static org.junit.Assert.*;8import java.util.List;9import java.util.ArrayList;10@RunWith(MockitoJUnitRunner.class)11public class MockitoTest1 {12    private List mockList;13    public void test1(){14        mockList.add("one");15        mockList.clear();16        ArgumentCaptor captor = ArgumentCaptor.forClass(String.class);17        verify(mockList).add(captor.capture());18        String capturedArgument = (String) captor.getValue();19        assertEquals("one",capturedArgument);20    }21}22import org.junit.Test;23import org.junit.runner.RunWith;24import org.mockito.ArgumentCaptor;25import org.mockito.Mock;26import org.mockito.runners.MockitoJUnitRunner;27import static org.mockito.Mockito.*;28import static org.junit.Assert.*;29import java.util.List;30import java.util.ArrayList;31@RunWith(MockitoJUnitRunner.class)32public class MockitoTest1 {33    private List mockList;34    public void test1(){35        mockList.add("one");36        mockList.clear();37        ArgumentCaptor captor = ArgumentCaptor.forClass(String.class);38        verify(mockList).add(captor.capture());39        String capturedArgument = (String) captor.getValue();40        assertEquals("one",capturedArgument);41    }42}43import org.junit.Test;

capture

Using AI Code Generation

1package com.ack.j2se.mock;2import java.util.List;3import org.junit.Test;4import org.junit.runner.RunWith;5import org.mockito.ArgumentCaptor;6import org.mockito.Mock;7import org.mockito.runners.MockitoJUnitRunner;8import static org.mockito.Mockito.*;9import static org.junit.Assert.*;10@RunWith(MockitoJUnitRunner.class)11public class ArgumentCaptorTest {12  private List<String> mockedList;13  public void testArgumentCaptor() {14    mockedList.add( "one" );15    mockedList.clear();16    ArgumentCaptor<String> argument = ArgumentCaptor.forClass( String.class );17    verify( mockedList ).add( argument.capture() );18    assertEquals( "one", argument.getValue() );19    verify( mockedList ).add( "one" );20    verify( mockedList ).clear();21  }22}23package com.ack.j2se.mock;24import java.util.List;25import org.junit.Test;26import org.junit.runner.RunWith;27import org.mockito.ArgumentCaptor;28import org.mockito.Mock;29import org.mockito.runners.MockitoJUnitRunner;30import static org.mockito.Mockito.*;31import static org.junit.Assert.*;32@RunWith(MockitoJUnitRunner.class)33public class ArgumentCaptorTest {34  private List<String> mockedList;35  public void testArgumentCaptor() {36    mockedList.add( "one" );37    mockedList.clear();38    ArgumentCaptor<String> argument = ArgumentCaptor.forClass( String.class );39    verify( mockedList, times( 1 ) ).add( argument.capture() );40    assertEquals( "one", argument.getValue

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.