Search in sources :

Example 6 with UpscaleEvent

use of com.sequenceiq.freeipa.flow.freeipa.upscale.event.UpscaleEvent in project cloudbreak by hortonworks.

the class UpgradeFlowEventChainFactoryTest method testFlowChainCreation.

@Test
public void testFlowChainCreation() {
    ImageSettingsRequest imageSettingsRequest = new ImageSettingsRequest();
    UpgradeEvent event = new UpgradeEvent("selector", STACK_ID, Sets.newHashSet("repl1", "repl2"), "pgw", OPERATION_ID, imageSettingsRequest, false);
    FlowTriggerEventQueue eventQueue = underTest.createFlowTriggerEventQueue(event);
    assertEquals("UpgradeFlowEventChainFactory", eventQueue.getFlowChainName());
    Queue<Selectable> queue = eventQueue.getQueue();
    assertEquals(10, queue.size());
    SaltUpdateTriggerEvent saltUpdateTriggerEvent = (SaltUpdateTriggerEvent) queue.poll();
    assertEquals(OPERATION_ID, saltUpdateTriggerEvent.getOperationId());
    assertEquals(STACK_ID, saltUpdateTriggerEvent.getResourceId());
    assertEquals(SaltUpdateEvent.SALT_UPDATE_EVENT.event(), saltUpdateTriggerEvent.selector());
    assertFalse(saltUpdateTriggerEvent.isFinalChain());
    ImageChangeEvent imageChangeEvent = (ImageChangeEvent) queue.poll();
    assertEquals(OPERATION_ID, imageChangeEvent.getOperationId());
    assertEquals(STACK_ID, imageChangeEvent.getResourceId());
    assertEquals(IMAGE_CHANGE_EVENT.event(), imageChangeEvent.selector());
    assertEquals(imageSettingsRequest, imageChangeEvent.getRequest());
    UpscaleEvent upscaleEvent1 = (UpscaleEvent) queue.poll();
    assertEquals(OPERATION_ID, upscaleEvent1.getOperationId());
    assertEquals(STACK_ID, upscaleEvent1.getResourceId());
    assertTrue(upscaleEvent1.isChained());
    assertFalse(upscaleEvent1.isFinalChain());
    assertFalse(upscaleEvent1.isRepair());
    assertEquals(UpscaleFlowEvent.UPSCALE_EVENT.event(), upscaleEvent1.selector());
    assertEquals(4, upscaleEvent1.getInstanceCountByGroup());
    DownscaleEvent downscaleEvent1 = (DownscaleEvent) queue.poll();
    assertEquals(OPERATION_ID, downscaleEvent1.getOperationId());
    assertEquals(STACK_ID, downscaleEvent1.getResourceId());
    assertTrue(downscaleEvent1.isChained());
    assertFalse(downscaleEvent1.isFinalChain());
    assertFalse(downscaleEvent1.isRepair());
    assertEquals(DownscaleFlowEvent.DOWNSCALE_EVENT.event(), downscaleEvent1.selector());
    assertEquals(3, downscaleEvent1.getInstanceCountByGroup());
    assertEquals(1, downscaleEvent1.getInstanceIds().size());
    String firstInstanceToDownscale = downscaleEvent1.getInstanceIds().get(0);
    assertTrue(firstInstanceToDownscale.startsWith("repl"));
    UpscaleEvent upscaleEvent2 = (UpscaleEvent) queue.poll();
    assertEquals(OPERATION_ID, upscaleEvent2.getOperationId());
    assertEquals(STACK_ID, upscaleEvent2.getResourceId());
    assertTrue(upscaleEvent2.isChained());
    assertFalse(upscaleEvent2.isFinalChain());
    assertFalse(upscaleEvent2.isRepair());
    assertEquals(UpscaleFlowEvent.UPSCALE_EVENT.event(), upscaleEvent2.selector());
    assertEquals(4, upscaleEvent2.getInstanceCountByGroup());
    DownscaleEvent downscaleEvent2 = (DownscaleEvent) queue.poll();
    assertEquals(OPERATION_ID, downscaleEvent2.getOperationId());
    assertEquals(STACK_ID, downscaleEvent2.getResourceId());
    assertTrue(downscaleEvent2.isChained());
    assertFalse(downscaleEvent2.isFinalChain());
    assertFalse(downscaleEvent2.isRepair());
    assertEquals(DownscaleFlowEvent.DOWNSCALE_EVENT.event(), downscaleEvent2.selector());
    assertEquals(3, downscaleEvent2.getInstanceCountByGroup());
    assertEquals(1, downscaleEvent2.getInstanceIds().size());
    String secondInstanceToDownscale = downscaleEvent2.getInstanceIds().get(0);
    assertTrue(secondInstanceToDownscale.startsWith("repl"));
    assertNotEquals(firstInstanceToDownscale, secondInstanceToDownscale);
    UpscaleEvent upscaleEvent3 = (UpscaleEvent) queue.poll();
    assertEquals(OPERATION_ID, upscaleEvent3.getOperationId());
    assertEquals(STACK_ID, upscaleEvent3.getResourceId());
    assertTrue(upscaleEvent3.isChained());
    assertFalse(upscaleEvent3.isFinalChain());
    assertFalse(upscaleEvent3.isRepair());
    assertEquals(UpscaleFlowEvent.UPSCALE_EVENT.event(), upscaleEvent3.selector());
    assertEquals(4, upscaleEvent3.getInstanceCountByGroup());
    ChangePrimaryGatewayEvent changePrimaryGatewayEvent = (ChangePrimaryGatewayEvent) queue.poll();
    assertEquals(OPERATION_ID, changePrimaryGatewayEvent.getOperationId());
    assertEquals(STACK_ID, changePrimaryGatewayEvent.getResourceId());
    assertFalse(changePrimaryGatewayEvent.isFinalChain());
    assertEquals(ChangePrimaryGatewayFlowEvent.CHANGE_PRIMARY_GATEWAY_EVENT.event(), changePrimaryGatewayEvent.selector());
    assertEquals(3, changePrimaryGatewayEvent.getRepairInstaceIds().size());
    assertTrue(List.of("repl1", "repl2", "pgw").containsAll(changePrimaryGatewayEvent.getRepairInstaceIds()));
    DownscaleEvent downscaleEvent3 = (DownscaleEvent) queue.poll();
    assertEquals(OPERATION_ID, downscaleEvent3.getOperationId());
    assertEquals(STACK_ID, downscaleEvent3.getResourceId());
    assertTrue(downscaleEvent3.isChained());
    assertFalse(downscaleEvent3.isFinalChain());
    assertFalse(downscaleEvent3.isRepair());
    assertEquals(DownscaleFlowEvent.DOWNSCALE_EVENT.event(), downscaleEvent3.selector());
    assertEquals(3, downscaleEvent3.getInstanceCountByGroup());
    assertEquals(1, downscaleEvent3.getInstanceIds().size());
    assertEquals("pgw", downscaleEvent3.getInstanceIds().get(0));
    SaltUpdateTriggerEvent saltUpdateTriggerEvent2 = (SaltUpdateTriggerEvent) queue.poll();
    assertEquals(OPERATION_ID, saltUpdateTriggerEvent2.getOperationId());
    assertEquals(STACK_ID, saltUpdateTriggerEvent2.getResourceId());
    assertEquals(SaltUpdateEvent.SALT_UPDATE_EVENT.event(), saltUpdateTriggerEvent2.selector());
    assertTrue(saltUpdateTriggerEvent2.isChained());
    assertTrue(saltUpdateTriggerEvent2.isFinalChain());
}
Also used : UpgradeEvent(com.sequenceiq.freeipa.flow.freeipa.upgrade.UpgradeEvent) ImageSettingsRequest(com.sequenceiq.freeipa.api.v1.freeipa.stack.model.common.image.ImageSettingsRequest) FlowTriggerEventQueue(com.sequenceiq.flow.core.chain.config.FlowTriggerEventQueue) DownscaleEvent(com.sequenceiq.freeipa.flow.freeipa.downscale.event.DownscaleEvent) Selectable(com.sequenceiq.cloudbreak.common.event.Selectable) ImageChangeEvent(com.sequenceiq.freeipa.flow.stack.image.change.event.ImageChangeEvent) UpscaleEvent(com.sequenceiq.freeipa.flow.freeipa.upscale.event.UpscaleEvent) SaltUpdateTriggerEvent(com.sequenceiq.freeipa.flow.freeipa.salt.update.SaltUpdateTriggerEvent) ChangePrimaryGatewayEvent(com.sequenceiq.freeipa.flow.freeipa.repair.changeprimarygw.event.ChangePrimaryGatewayEvent) Test(org.junit.jupiter.api.Test)

Aggregations

UpscaleEvent (com.sequenceiq.freeipa.flow.freeipa.upscale.event.UpscaleEvent)6 Selectable (com.sequenceiq.cloudbreak.common.event.Selectable)4 DownscaleEvent (com.sequenceiq.freeipa.flow.freeipa.downscale.event.DownscaleEvent)4 ChangePrimaryGatewayEvent (com.sequenceiq.freeipa.flow.freeipa.repair.changeprimarygw.event.ChangePrimaryGatewayEvent)3 ArrayList (java.util.ArrayList)3 FlowTriggerEventQueue (com.sequenceiq.flow.core.chain.config.FlowTriggerEventQueue)2 CloudStack (com.sequenceiq.cloudbreak.cloud.model.CloudStack)1 BadRequestException (com.sequenceiq.cloudbreak.common.exception.BadRequestException)1 FlowIdentifier (com.sequenceiq.flow.api.model.FlowIdentifier)1 ImageSettingsRequest (com.sequenceiq.freeipa.api.v1.freeipa.stack.model.common.image.ImageSettingsRequest)1 UpscaleResponse (com.sequenceiq.freeipa.api.v1.freeipa.stack.model.scale.UpscaleResponse)1 Operation (com.sequenceiq.freeipa.entity.Operation)1 Stack (com.sequenceiq.freeipa.entity.Stack)1 SaltUpdateTriggerEvent (com.sequenceiq.freeipa.flow.freeipa.salt.update.SaltUpdateTriggerEvent)1 UpgradeEvent (com.sequenceiq.freeipa.flow.freeipa.upgrade.UpgradeEvent)1 StackContext (com.sequenceiq.freeipa.flow.stack.StackContext)1 StackEvent (com.sequenceiq.freeipa.flow.stack.StackEvent)1 ImageChangeEvent (com.sequenceiq.freeipa.flow.stack.image.change.event.ImageChangeEvent)1 HashSet (java.util.HashSet)1 Map (java.util.Map)1