Merge pull request #179 from andreisavu/vm-expunged-predicate

Defined new predicate to check if a VM is expunged
This commit is contained in:
Adrian Cole 2011-11-29 07:46:02 -08:00
commit 5762f0f851
2 changed files with 116 additions and 0 deletions

View File

@ -0,0 +1,60 @@
/**
* Licensed to jclouds, Inc. (jclouds) under one or more
* contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. jclouds licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.jclouds.cloudstack.predicates;
import com.google.common.base.Predicate;
import com.google.inject.Inject;
import org.jclouds.cloudstack.CloudStackClient;
import org.jclouds.cloudstack.domain.VirtualMachine;
import org.jclouds.cloudstack.domain.VirtualMachine.State;
import org.jclouds.logging.Logger;
import javax.annotation.Resource;
import javax.inject.Singleton;
import static com.google.common.base.Preconditions.checkNotNull;
/**
*
* Tests to see if a virtualMachine is expunged from the system
*
* @author Andrei Savu
*/
@Singleton
public class VirtualMachineExpunged implements Predicate<VirtualMachine> {
private final CloudStackClient client;
@Resource
protected Logger logger = Logger.NULL;
@Inject
public VirtualMachineExpunged(CloudStackClient client) {
this.client = client;
}
public boolean apply(VirtualMachine virtualMachine) {
logger.trace("looking for state on virtualMachine %s", checkNotNull(virtualMachine, "virtualMachine"));
return refresh(virtualMachine) == null;
}
private VirtualMachine refresh(VirtualMachine virtualMachine) {
return client.getVirtualMachineClient().getVirtualMachine(virtualMachine.getId());
}
}

View File

@ -0,0 +1,56 @@
/**
* Licensed to jclouds, Inc. (jclouds) under one or more
* contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. jclouds licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.jclouds.cloudstack.predicates;
import org.jclouds.cloudstack.CloudStackClient;
import org.jclouds.cloudstack.domain.VirtualMachine;
import org.jclouds.cloudstack.features.VirtualMachineClient;
import org.testng.annotations.Test;
import static org.easymock.EasyMock.createMock;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.replay;
import static org.easymock.EasyMock.verify;
import static org.testng.Assert.assertTrue;
/**
* @author Andrei Savu
*/
@Test(groups = "unit", testName = "VirtualMachineExpungedTest")
public class VirtualMachineExpungedTest {
@Test
public void testWaitForVirtualMachineToBeExpunged() {
CloudStackClient client = createMock(CloudStackClient.class);
VirtualMachineClient virtualMachineClient = createMock(VirtualMachineClient.class);
long virtualMachineId = 229;
VirtualMachine virtualMachine = VirtualMachine.builder().id(virtualMachineId).build();
expect(client.getVirtualMachineClient()).andReturn(virtualMachineClient);
expect(virtualMachineClient.getVirtualMachine(virtualMachineId)).andReturn(null);
replay(client, virtualMachineClient);
assertTrue(new VirtualMachineExpunged(client).apply(virtualMachine));
verify(client, virtualMachineClient);
}
}