use of org.objectweb.proactive.core.node.NodeImpl in project scheduling by ow2-proactive.
the class topologyConcurrencyTest method action.
@Test
public void action() throws Exception {
BasicConfigurator.resetConfiguration();
BasicConfigurator.configure();
Logger.getLogger(TopologyManager.class).setLevel(Level.DEBUG);
PAResourceManagerProperties.RM_TOPOLOGY_PINGER.updateProperty(HostsPinger.class.getName());
PAResourceManagerProperties.RM_TOPOLOGY_ENABLED.updateProperty("true");
PAResourceManagerProperties.RM_TOPOLOGY_DISTANCE_ENABLED.updateProperty("false");
final ProActiveRuntimeImpl runtime = mock(ProActiveRuntimeImpl.class);
when(runtime.getVMInformation()).thenReturn(new DummyVMInfo());
final TopologyManager manager = new TopologyManager();
List<Callable<Boolean>> calls = new ArrayList<>(total);
// first set of tasks, add nodes and remove them
for (int i = 0; i < total; i++) {
final int j = i;
calls.add(new Callable<Boolean>() {
@Override
public Boolean call() throws Exception {
try {
int index = j - (j % collisionSize);
Node node = new NodeImpl(runtime, baseUrl + index);
if ((j % (collisionSize / 2)) == 0) {
manager.addNode(node);
} else {
// more remove than add, to try to reproduce empty list problems
manager.removeNode(node);
}
// launch some selections on the node just added or removed to trigger ConcurrentModficationExceptions if any
NodeSet set = manager.getHandler(TopologyDescriptor.SINGLE_HOST).select(1, Collections.singletonList(node));
System.out.println(set);
} catch (Exception e) {
e.printStackTrace();
return false;
}
return true;
}
});
}
List<Future<Boolean>> futures = s.invokeAll(calls);
for (Future<Boolean> fut : futures) {
Assert.assertTrue(fut.get());
}
calls = new ArrayList<>(nbThreads * nodeFactor);
// second set of task remove all nodes created
for (int i = 0; i < total; i += collisionSize) {
final int j = i;
calls.add(new Callable<Boolean>() {
@Override
public Boolean call() throws Exception {
try {
int index = j;
Node node = new NodeImpl(runtime, baseUrl + index);
manager.removeNode(node);
NodeSet set = manager.getHandler(TopologyDescriptor.SINGLE_HOST).select(1, Collections.singletonList(node));
System.out.println(set);
} catch (Exception e) {
e.printStackTrace();
return false;
}
return true;
}
});
}
futures = s.invokeAll(calls);
for (Future<Boolean> fut : futures) {
Assert.assertTrue(fut.get());
}
System.out.println(manager.getTopology().getHosts());
// finally verify that the nodes on host structure is empty (null)
Assert.assertNull(manager.getNodesOnHost(DummyVMInfo.address));
}
use of org.objectweb.proactive.core.node.NodeImpl in project scheduling by ow2-proactive.
the class InProcessTaskExecutorTest method mockedNodeSet.
private NodeSet mockedNodeSet() {
NodeSet nodes = new NodeSet();
ProActiveRuntime proActiveRuntime = mock(ProActiveRuntime.class);
VMInformation vmInformation = mock(VMInformation.class);
when(vmInformation.getHostName()).thenReturn("dummyhost");
when(proActiveRuntime.getVMInformation()).thenReturn(vmInformation);
nodes.add(new NodeImpl(proActiveRuntime, "dummyhost"));
return nodes;
}
Aggregations