use of com.baidu.hugegraph.computer.k8s.operator.common.AbstractController 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);
}
Aggregations