Search in sources :

Example 6 with ApplicationApi

use of com.yahoo.vespa.orchestrator.model.ApplicationApi in project vespa by vespa-engine.

the class HostedVespaPolicyTest method testAcquirePermissionToRemove.

@Test
public void testAcquirePermissionToRemove() throws OrchestrationException {
    final HostedVespaClusterPolicy clusterPolicy = mock(HostedVespaClusterPolicy.class);
    final HostedVespaPolicy policy = new HostedVespaPolicy(clusterPolicy, clientFactory);
    final ApplicationApi applicationApi = mock(ApplicationApi.class);
    when(applicationApi.applicationId()).thenReturn(ApplicationId.fromSerializedForm("tenant:app:default"));
    ClusterApi clusterApi1 = mock(ClusterApi.class);
    ClusterApi clusterApi2 = mock(ClusterApi.class);
    ClusterApi clusterApi3 = mock(ClusterApi.class);
    List<ClusterApi> clusterApis = Arrays.asList(clusterApi1, clusterApi2, clusterApi3);
    when(applicationApi.getClusters()).thenReturn(clusterApis);
    StorageNode storageNode1 = mock(StorageNode.class);
    HostName hostName1 = new HostName("storage-1");
    when(storageNode1.hostName()).thenReturn(hostName1);
    HostName hostName2 = new HostName("host-2");
    StorageNode storageNode3 = mock(StorageNode.class);
    HostName hostName3 = new HostName("storage-3");
    when(storageNode1.hostName()).thenReturn(hostName3);
    List<StorageNode> upStorageNodes = Arrays.asList(storageNode1, storageNode3);
    when(applicationApi.getStorageNodesInGroupInClusterOrder()).thenReturn(upStorageNodes);
    List<HostName> noRemarksHostNames = Arrays.asList(hostName1, hostName2, hostName3);
    when(applicationApi.getNodesInGroupWithStatus(HostStatus.NO_REMARKS)).thenReturn(noRemarksHostNames);
    InOrder order = inOrder(applicationApi, clusterPolicy, storageNode1, storageNode3);
    policy.acquirePermissionToRemove(applicationApi);
    order.verify(applicationApi).getClusters();
    order.verify(clusterPolicy).verifyGroupGoingDownPermanentlyIsFine(clusterApi1);
    order.verify(clusterPolicy).verifyGroupGoingDownPermanentlyIsFine(clusterApi2);
    order.verify(clusterPolicy).verifyGroupGoingDownPermanentlyIsFine(clusterApi3);
    order.verify(applicationApi).getStorageNodesInGroupInClusterOrder();
    order.verify(storageNode1).setNodeState(ClusterControllerNodeState.DOWN);
    order.verify(storageNode3).setNodeState(ClusterControllerNodeState.DOWN);
    order.verify(applicationApi).getNodesInGroupWithStatus(HostStatus.NO_REMARKS);
    order.verify(applicationApi).setHostState(hostName1, HostStatus.ALLOWED_TO_BE_DOWN);
    order.verify(applicationApi).setHostState(hostName2, HostStatus.ALLOWED_TO_BE_DOWN);
    order.verify(applicationApi).setHostState(hostName3, HostStatus.ALLOWED_TO_BE_DOWN);
    order.verifyNoMoreInteractions();
}
Also used : ClusterApi(com.yahoo.vespa.orchestrator.model.ClusterApi) InOrder(org.mockito.InOrder) ApplicationApi(com.yahoo.vespa.orchestrator.model.ApplicationApi) StorageNode(com.yahoo.vespa.orchestrator.model.StorageNode) HostName(com.yahoo.vespa.applicationmodel.HostName) Test(org.junit.Test)

Aggregations

ApplicationApi (com.yahoo.vespa.orchestrator.model.ApplicationApi)6 ApplicationApiImpl (com.yahoo.vespa.orchestrator.model.ApplicationApiImpl)4 NodeGroup (com.yahoo.vespa.orchestrator.model.NodeGroup)3 HostName (com.yahoo.vespa.applicationmodel.HostName)2 ClusterApi (com.yahoo.vespa.orchestrator.model.ClusterApi)2 StorageNode (com.yahoo.vespa.orchestrator.model.StorageNode)2 MutableStatusRegistry (com.yahoo.vespa.orchestrator.status.MutableStatusRegistry)2 Test (org.junit.Test)2 InOrder (org.mockito.InOrder)2 ApplicationInstance (com.yahoo.vespa.applicationmodel.ApplicationInstance)1 ApplicationInstanceReference (com.yahoo.vespa.applicationmodel.ApplicationInstanceReference)1 ApplicationInstanceStatus (com.yahoo.vespa.orchestrator.status.ApplicationInstanceStatus)1