Java Capture.newInstance方法代码示例

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void testRestartTaskLeaderRedirect() throws Throwable {
    ConnectorTaskId taskId = new ConnectorTaskId(CONNECTOR_NAME, 0);

    final Capture<Callback<Void>> cb = Capture.newInstance();
    herder.restartTask(EasyMock.eq(taskId), EasyMock.capture(cb));
    expectAndCallbackNotLeaderException(cb);

    EasyMock.expect(RestServer.httpRequest(EasyMock.eq("http://leader:8083/connectors/" + CONNECTOR_NAME + "/tasks/0/restart?forward=true"),
            EasyMock.eq("POST"), EasyMock.isNull(), EasyMock.<TypeReference>anyObject()))
            .andReturn(new RestServer.HttpResponse<>(202, new HashMap<String, List<String>>(), null));

    PowerMock.replayAll();

    connectorsResource.restartTask(CONNECTOR_NAME, 0, null);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void testRestartTaskOwnerRedirect() throws Throwable {
    ConnectorTaskId taskId = new ConnectorTaskId(CONNECTOR_NAME, 0);

    final Capture<Callback<Void>> cb = Capture.newInstance();
    herder.restartTask(EasyMock.eq(taskId), EasyMock.capture(cb));
    String ownerUrl = "http://owner:8083";
    expectAndCallbackException(cb, new NotAssignedException("not owner test", ownerUrl));

    EasyMock.expect(RestServer.httpRequest(EasyMock.eq("http://owner:8083/connectors/" + CONNECTOR_NAME + "/tasks/0/restart?forward=false"),
            EasyMock.eq("POST"), EasyMock.isNull(), EasyMock.<TypeReference>anyObject()))
            .andReturn(new RestServer.HttpResponse<>(202, new HashMap<String, List<String>>(), null));

    PowerMock.replayAll();

    connectorsResource.restartTask(CONNECTOR_NAME, 0, true);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void testDispatchAction_withMiddleWare_shouldTriggerMiddleWareAndChangeState() {
  //GIVEN
  Middleware<Integer> mockMiddleWare = createStrictMock(Middleware.class);
  final Action<Integer> action = createStrictMock(Action.class);
  Store<Integer> store = new Store<>(0, mockMiddleWare);

  Capture<Middleware.Chain<Integer>> capturedChain = Capture.newInstance();
  mockMiddleWare.intercept(capture(capturedChain));
  expectLastCall()
      .andAnswer(
          () -> {
            final Middleware.Chain<Integer> chain =
                (Middleware.Chain<Integer>) getCurrentArguments()[0];
            chain.proceed(chain.action());
            return null;
          });
  expect(action.newState(0)).andReturn(1);
  replay(mockMiddleWare, action);

  //WHEN
  store.dispatch(action);

  //THEN
  verify(mockMiddleWare, action);
  assertThat(capturedChain.getValue().action(), is(action));
  assertThat(store.getState(), is(1));
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void
    testDispatchAction_withMiddleWares_shouldTriggerMiddleWareAndChangeStateUsingNewAction() {
  //GIVEN
  Middleware<Integer> mockMiddleWare = createMock(Middleware.class);
  final Action<Integer> action = createMock(Action.class);
  Store<Integer> store = new Store<>(0, mockMiddleWare);

  Capture<Middleware.Chain<Integer>> capturedChain = Capture.newInstance();
  mockMiddleWare.intercept(capture(capturedChain));
  expectLastCall()
      .andAnswer(
          () -> {
            final Middleware.Chain<Integer> chain =
                (Middleware.Chain<Integer>) getCurrentArguments()[0];
            final Action<Integer> decrementAction = integer -> integer - 1;
            chain.proceed(decrementAction);
            return null;
          });
  replay(mockMiddleWare, action);

  //WHEN
  store.dispatch(action);

  //THEN
  verify(mockMiddleWare, action);
  assertThat(store.getState(), is(-1));
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void testCorruptConfig() {
    Map<String, String> config = new HashMap<>();
    config.put(ConnectorConfig.NAME_CONFIG, CONNECTOR_NAME);
    config.put(ConnectorConfig.CONNECTOR_CLASS_CONFIG, BogusSinkConnector.class.getName());
    Connector connectorMock = PowerMock.createMock(Connector.class);
    String error = "This is an error in your config!";
    List<String> errors = new ArrayList<>(singletonList(error));
    String key = "foo.invalid.key";
    EasyMock.expect(connectorMock.validate(config)).andReturn(
        new Config(
            Arrays.asList(new ConfigValue(key, null, Collections.emptyList(), errors))
        )
    );
    ConfigDef configDef = new ConfigDef();
    configDef.define(key, ConfigDef.Type.STRING, ConfigDef.Importance.HIGH, "");
    EasyMock.expect(worker.getPlugins()).andReturn(plugins).times(3);
    EasyMock.expect(plugins.compareAndSwapLoaders(connectorMock)).andReturn(delegatingLoader);
    EasyMock.expect(worker.getPlugins()).andStubReturn(plugins);
    EasyMock.expect(plugins.newConnector(EasyMock.anyString())).andReturn(connectorMock);
    EasyMock.expect(connectorMock.config()).andStubReturn(configDef);
    EasyMock.expect(Plugins.compareAndSwapLoaders(delegatingLoader)).andReturn(pluginLoader);
    Callback<Herder.Created<ConnectorInfo>> callback = PowerMock.createMock(Callback.class);
    Capture<BadRequestException> capture = Capture.newInstance();
    callback.onCompletion(
        EasyMock.capture(capture), EasyMock.isNull(Herder.Created.class)
    );

    PowerMock.replayAll();

    herder.putConnectorConfig(CONNECTOR_NAME, config, true, callback);
    assertEquals(
        capture.getValue().getMessage(),
        "Connector configuration is invalid and contains the following 1 error(s):\n" +
            error + "\n" +
            "You can also find the above list of errors at the endpoint `/{connectorType}/config/validate`"
    );

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void testGetConnectorTaskConfigs() throws Throwable {
    final Capture<Callback<List<TaskInfo>>> cb = Capture.newInstance();
    herder.taskConfigs(EasyMock.eq(CONNECTOR_NAME), EasyMock.capture(cb));
    expectAndCallbackResult(cb, TASK_INFOS);

    PowerMock.replayAll();

    List<TaskInfo> taskInfos = connectorsResource.getTaskConfigs(CONNECTOR_NAME, FORWARD);
    assertEquals(TASK_INFOS, taskInfos);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void testPutConnectorTaskConfigs() throws Throwable {
    final Capture<Callback<Void>> cb = Capture.newInstance();
    herder.putTaskConfigs(EasyMock.eq(CONNECTOR_NAME), EasyMock.eq(TASK_CONFIGS), EasyMock.capture(cb));
    expectAndCallbackResult(cb, null);

    PowerMock.replayAll();

    connectorsResource.putTaskConfigs(CONNECTOR_NAME, FORWARD, TASK_CONFIGS);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test(expected = AlreadyExistsException.class)
public void testCreateConnectorExists() throws Throwable {
    CreateConnectorRequest body = new CreateConnectorRequest(CONNECTOR_NAME, Collections.singletonMap(ConnectorConfig.NAME_CONFIG, CONNECTOR_NAME));

    final Capture<Callback<Herder.Created<ConnectorInfo>>> cb = Capture.newInstance();
    herder.putConnectorConfig(EasyMock.eq(CONNECTOR_NAME), EasyMock.eq(body.config()), EasyMock.eq(false), EasyMock.capture(cb));
    expectAndCallbackException(cb, new AlreadyExistsException("already exists"));

    PowerMock.replayAll();

    connectorsResource.createConnector(FORWARD, body);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test(expected = NotFoundException.class)
public void testRestartConnectorNotFound() throws Throwable {
    final Capture<Callback<Void>> cb = Capture.newInstance();
    herder.restartConnector(EasyMock.eq(CONNECTOR_NAME), EasyMock.capture(cb));
    expectAndCallbackException(cb, new NotFoundException("not found"));

    PowerMock.replayAll();

    connectorsResource.restartConnector(CONNECTOR_NAME, FORWARD);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void testRestartConnectorLeaderRedirect() throws Throwable {
    final Capture<Callback<Void>> cb = Capture.newInstance();
    herder.restartConnector(EasyMock.eq(CONNECTOR_NAME), EasyMock.capture(cb));
    expectAndCallbackNotLeaderException(cb);

    EasyMock.expect(RestServer.httpRequest(EasyMock.eq("http://leader:8083/connectors/" + CONNECTOR_NAME + "/restart?forward=true"),
            EasyMock.eq("POST"), EasyMock.isNull(), EasyMock.<TypeReference>anyObject()))
            .andReturn(new RestServer.HttpResponse<>(202, new HashMap<String, List<String>>(), null));

    PowerMock.replayAll();

    connectorsResource.restartConnector(CONNECTOR_NAME, null);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void testPutConnectorConfig() throws Throwable {
    final Capture<Callback<Herder.Created<ConnectorInfo>>> cb = Capture.newInstance();
    herder.putConnectorConfig(EasyMock.eq(CONNECTOR_NAME), EasyMock.eq(CONNECTOR_CONFIG), EasyMock.eq(true), EasyMock.capture(cb));
    expectAndCallbackResult(cb, new Herder.Created<>(false, new ConnectorInfo(CONNECTOR_NAME, CONNECTOR_CONFIG, CONNECTOR_TASK_NAMES)));

    PowerMock.replayAll();

    connectorsResource.putConnectorConfig(CONNECTOR_NAME, FORWARD, CONNECTOR_CONFIG);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test(expected = NotFoundException.class)
public void testRestartTaskNotFound() throws Throwable {
    ConnectorTaskId taskId = new ConnectorTaskId(CONNECTOR_NAME, 0);
    final Capture<Callback<Void>> cb = Capture.newInstance();
    herder.restartTask(EasyMock.eq(taskId), EasyMock.capture(cb));
    expectAndCallbackException(cb, new NotFoundException("not found"));

    PowerMock.replayAll();

    connectorsResource.restartTask(CONNECTOR_NAME, 0, FORWARD);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test
public void testGetConnectorConfig() throws Throwable {
    final Capture<Callback<Map<String, String>>> cb = Capture.newInstance();
    herder.connectorConfig(EasyMock.eq(CONNECTOR_NAME), EasyMock.capture(cb));
    expectAndCallbackResult(cb, CONNECTOR_CONFIG);

    PowerMock.replayAll();

    Map<String, String> connConfig = connectorsResource.getConnectorConfig(CONNECTOR_NAME, FORWARD);
    assertEquals(CONNECTOR_CONFIG, connConfig);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Test(expected = NotFoundException.class)
public void testGetConnectorConfigConnectorNotFound() throws Throwable {
    final Capture<Callback<Map<String, String>>> cb = Capture.newInstance();
    herder.connectorConfig(EasyMock.eq(CONNECTOR_NAME), EasyMock.capture(cb));
    expectAndCallbackException(cb, new NotFoundException("not found"));

    PowerMock.replayAll();

    connectorsResource.getConnectorConfig(CONNECTOR_NAME, FORWARD);

    PowerMock.verifyAll();
}
 

import org.easymock.Capture; //导入方法依赖的package包/类
@Before
public void setup() throws Exception {
    super.setup();
    contentType = Capture.newInstance();
}