use of org.ow2.proactive.resourcemanager.authentication.Client in project scheduling by ow2-proactive.
the class RMCoreTest method configureRMNode.
private void configureRMNode(MockedRMNodeParameters param) {
RMNode rmNode = param.getRmNode();
Node mockedNode = Mockito.mock(Node.class);
NodeInformation mockedNodeInformation = Mockito.mock(NodeInformation.class);
when(mockedNode.getNodeInformation()).thenReturn(mockedNodeInformation);
when(rmNode.getNode()).thenReturn(mockedNode);
when(rmNode.getNodeName()).thenReturn(param.getUrl());
when(rmNode.getNodeURL()).thenReturn(param.getUrl());
when(rmNode.isDown()).thenReturn(param.isDown());
when(rmNode.isFree()).thenReturn(param.isFree());
when(rmNode.isLocked()).thenReturn(param.isLocked());
when(mockedNodeInformation.getURL()).thenReturn(param.getUrl());
when(mockedNodeInformation.getName()).thenReturn(param.getUrl());
when(rmNode.getNodeSource()).thenReturn(param.getNodeSource());
when(rmNode.getNodeSourceName()).thenReturn(param.getNodeSourceName());
when(rmNode.getAdminPermission()).thenReturn(null);
when(rmNode.getProvider()).thenReturn(new Client());
Client client = Mockito.mock(Client.class);
when(rmNode.getOwner()).thenReturn(client);
when(client.getName()).thenReturn("test");
when(rmNode.getNodeURL()).thenReturn(param.getUrl());
try {
when(mockedNode.getThreadDump()).thenReturn("threadDumpExample");
} catch (ProActiveException e) {
e.printStackTrace();
}
}
use of org.ow2.proactive.resourcemanager.authentication.Client in project scheduling by ow2-proactive.
the class RMCoreTest method testGetNodeByUrlIncludingDeployingNodesUnknownNodeUrl.
@Test
public void testGetNodeByUrlIncludingDeployingNodesUnknownNodeUrl() {
RMDeployingNode rmNode = new RMDeployingNode("node", mockedNodeSource, "command", new Client());
doReturn(rmNode).when(mockedNodeSource).getNodeInDeployingOrLostNodes(rmNode.getNodeURL());
RMNode rmNodeFound = rmCore.getNodeByUrlIncludingDeployingNodes(rmNode.getNodeURL());
assertThat(rmNodeFound).isSameAs(rmNode);
}
use of org.ow2.proactive.resourcemanager.authentication.Client in project scheduling by ow2-proactive.
the class NodeSourceTest method createNodeSource.
private NodeSource createNodeSource(InfrastructureManager infrastructureManager, NodeSourcePolicy nodeSourcePolicy, Client client) {
NodeSourceData nodeSourceData = new NodeSourceData();
nodeSourceData.setProvider(client);
nodeSourceData.setNodesRecoverable(NODES_NOT_RECOVERABLE);
return new NodeSource("registrationURL", "name", infrastructureManager, nodeSourcePolicy, mock(RMCore.class), mock(RMMonitoringImpl.class), nodeSourceData.toNodeSourceDescriptor());
}
use of org.ow2.proactive.resourcemanager.authentication.Client in project scheduling by ow2-proactive.
the class NodeSourceTest method testSetNodeAvailableKnownNode.
@Test
public void testSetNodeAvailableKnownNode() throws RMException {
Node node = createNode(PROACTIVE_PROGRAMMING_NODE_URL);
RMNode rmNode = new RMNodeImpl(node, nodeSource, client, mock(Permission.class));
nodeSource.internalAddNode(node);
nodeSource.detectedPingedDownNode(node.getNodeInformation().getName(), node.getNodeInformation().getURL());
assertThat(nodeSource.getDownNodes()).hasSize(1);
boolean result = nodeSource.setNodeAvailable(rmNode);
assertThat(result).isTrue();
assertThat(nodeSource.getDownNodes()).hasSize(0);
}
use of org.ow2.proactive.resourcemanager.authentication.Client in project scheduling by ow2-proactive.
the class NodeSourceTest method testSetNodeAvailableUnknownNode.
@Test
public void testSetNodeAvailableUnknownNode() throws RMException {
Node node = createNode(PROACTIVE_PROGRAMMING_NODE_URL);
RMNode rmNode = new RMNodeImpl(node, nodeSource, client, mock(Permission.class));
nodeSource.internalAddNode(node);
assertThat(nodeSource.getDownNodes()).hasSize(0);
boolean result = nodeSource.setNodeAvailable(rmNode);
assertThat(result).isFalse();
assertThat(nodeSource.getDownNodes()).hasSize(0);
}
Aggregations