Examples with JobObserver com.baidu.hugegraph.computer.driver.JobObserver used on opensource projects

Example 11 with JobObserver

use of com.baidu.hugegraph.computer.driver.JobObserver in project hugegraph-computer by hugegraph.

the class MiniKubeTest method testGetResourceListWithLabels.

@Test
public void testGetResourceListWithLabels() {
    Map<String, String> params = new HashMap<>();
    params.put(KubeSpecOptions.WORKER_INSTANCES.name(), "1");
    params.put(ComputerOptions.TRANSPORT_SERVER_PORT.name(), "0");
    params.put(ComputerOptions.RPC_SERVER_PORT.name(), "0");
    String jobId = this.driver.submitJob(ALGORITHM_NAME, params);
    JobObserver jobObserver = Mockito.mock(JobObserver.class);
    CompletableFuture<Void> future = this.driver.waitJobAsync(jobId, params, jobObserver);
    DefaultJobState jobState = new DefaultJobState();
    jobState.jobStatus(JobStatus.INITIALIZING);
    Mockito.verify(jobObserver, Mockito.timeout(15000L).atLeast(1)).onJobStateChanged(Mockito.eq(jobState));
    List<AbstractController<?>> controllers = Whitebox.getInternalState(this.entrypoint, "controllers");
    AbstractController<?> abstractController = controllers.get(0);
    List<Pod> pods = Whitebox.invoke(AbstractController.class, new Class[] { String.class, Class.class, Map.class }, "getResourceListWithLabels", abstractController, this.namespace, Pod.class, new HashMap<String, String>());
    Assert.assertNotEquals(0, pods.size());
    future.cancel(true);
}