@Before public void setup() { when(piIdBuilder.getPIdForEc2AvailabilityZone(uri)).thenReturn(instancePId); when(piIdBuilder.getGlobalAvailabilityZoneCodeFromEc2Id(instanceId)).thenReturn(globalAvzCode); when(piIdBuilder.getPId(SecurityGroup.getUrl(userId, securityGroupName))) .thenReturn(securityGroupPId); when(piIdBuilder.getPiQueuePId(PiQueue.INSTANCE_NETWORK_MANAGER_TEARDOWN)) .thenReturn(instanceNetworkManagerTeardownQueueId); when(securityGroupPId.forGlobalAvailablityZoneCode(globalAvzCode)).thenReturn(securityGroupPId); when(instanceNetworkManagerTeardownQueueId.forLocalScope( PiQueue.TERMINATE_INSTANCE.getNodeScope())) .thenReturn(instanceNetworkManagerTeardownQueueId); when(messageContextFactory.newMessageContext()).thenReturn(messageContext); when(dhtClientFactory.createReader()).thenReturn(dhtReader); when(resultInstance.getInstanceId()).thenReturn(instanceId); when(resultInstance.getNodeId()).thenReturn(instanceNodeId); when(resultInstance.getUserId()).thenReturn(userId); when(resultInstance.getSecurityGroupName()).thenReturn(securityGroupName); }
private void setupDht() { when(dhtClientFactory.createReader()).thenReturn(dhtReader); setupInstancesInDht(); }