Examples with StateListener com.cloud.utils.fsm.StateListener used on opensource projects

Example 1 with StateListener

use of com.cloud.utils.fsm.StateListener in project cloudstack by apache.

the class ConfigDriveNetworkElementTest method testExpunge.

@Test
@SuppressWarnings("unchecked")
public void testExpunge() throws NoTransitionException, NoSuchFieldException, IllegalAccessException {
    final StateMachine2<VirtualMachine.State, VirtualMachine.Event, VirtualMachine> stateMachine = VirtualMachine.State.getStateMachine();
    final Field listenersField = StateMachine2.class.getDeclaredField("_listeners");
    listenersField.setAccessible(true);
    List<StateListener<VirtualMachine.State, VirtualMachine.Event, VirtualMachine>> listeners = (List<StateListener<VirtualMachine.State, VirtualMachine.Event, VirtualMachine>>) listenersField.get(stateMachine);
    listeners.clear();
    _configDrivesNetworkElement.start();
    when(virtualMachine.getState()).thenReturn(VirtualMachine.State.Stopped);
    when(_vmInstanceDao.updateState(VirtualMachine.State.Stopped, VirtualMachine.Event.ExpungeOperation, VirtualMachine.State.Expunging, virtualMachine, null)).thenReturn(true);
    final HandleConfigDriveIsoAnswer answer = mock(HandleConfigDriveIsoAnswer.class);
    when(agentManager.easySend(anyLong(), any(HandleConfigDriveIsoCommand.class))).thenReturn(answer);
    when(answer.getResult()).thenReturn(true);
    stateMachine.transitTo(virtualMachine, VirtualMachine.Event.ExpungeOperation, null, _vmInstanceDao);
    ArgumentCaptor<HandleConfigDriveIsoCommand> commandCaptor = ArgumentCaptor.forClass(HandleConfigDriveIsoCommand.class);
    verify(agentManager, times(1)).easySend(anyLong(), commandCaptor.capture());
    HandleConfigDriveIsoCommand deleteCommand = commandCaptor.getValue();
    assertThat(deleteCommand.isCreate(), is(false));
}