mirror of https://github.com/apache/jclouds.git
issue 384: TakeSnapshotIfNotAlreadyAttached and Test added as preparation step for IMachineToIpAddress feature
This commit is contained in:
Andrea Turli
parent
b396a80a84
commit
6726b04645
|
|
@ -0,0 +1,78 @@
|
||||||
|
/**
|
||||||
|
* 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.virtualbox.functions;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
|
import javax.annotation.Resource;
|
||||||
|
import javax.inject.Named;
|
||||||
|
|
||||||
|
import org.jclouds.compute.reference.ComputeServiceConstants;
|
||||||
|
import org.jclouds.logging.Logger;
|
||||||
|
import org.virtualbox_4_1.IMachine;
|
||||||
|
import org.virtualbox_4_1.IProgress;
|
||||||
|
import org.virtualbox_4_1.ISession;
|
||||||
|
import org.virtualbox_4_1.ISnapshot;
|
||||||
|
import org.virtualbox_4_1.VirtualBoxManager;
|
||||||
|
|
||||||
|
import com.google.common.base.Function;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Andrea Turli
|
||||||
|
*/
|
||||||
|
public class TakeSnapshotIfNotAlreadyAttached implements Function<IMachine, ISnapshot> {
|
||||||
|
|
||||||
|
|
||||||
|
@Resource
|
||||||
|
@Named(ComputeServiceConstants.COMPUTE_LOGGER)
|
||||||
|
protected Logger logger = Logger.NULL;
|
||||||
|
|
||||||
|
private VirtualBoxManager manager;
|
||||||
|
private String snapshotName;
|
||||||
|
private String snapshotDesc;
|
||||||
|
|
||||||
|
|
||||||
|
public TakeSnapshotIfNotAlreadyAttached(VirtualBoxManager manager, String snapshotName, String snapshotDesc) {
|
||||||
|
this.manager = manager;
|
||||||
|
this.snapshotName = snapshotName;
|
||||||
|
this.snapshotDesc = snapshotDesc;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ISnapshot apply(@Nullable IMachine machine) {
|
||||||
|
// Snapshot a machine
|
||||||
|
ISession session = null;
|
||||||
|
if(machine.getCurrentSnapshot() == null ) {
|
||||||
|
try {
|
||||||
|
session = manager.openMachineSession(machine);
|
||||||
|
IProgress progress = session.getConsole().takeSnapshot(snapshotName, snapshotDesc);
|
||||||
|
if (progress.getCompleted())
|
||||||
|
logger.debug("Clone done with snapshot name: %s and descripton: %s", snapshotName, snapshotDesc);
|
||||||
|
} catch (Exception e) {
|
||||||
|
// TODO Auto-generated catch block
|
||||||
|
e.printStackTrace();
|
||||||
|
} finally {
|
||||||
|
session.unlockMachine();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return machine.getCurrentSnapshot();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,103 @@
|
||||||
|
/**
|
||||||
|
* 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.virtualbox.functions;
|
||||||
|
|
||||||
|
import static org.easymock.EasyMock.expect;
|
||||||
|
import static org.easymock.classextension.EasyMock.createMock;
|
||||||
|
import static org.easymock.classextension.EasyMock.createNiceMock;
|
||||||
|
import static org.easymock.classextension.EasyMock.replay;
|
||||||
|
import static org.easymock.classextension.EasyMock.verify;
|
||||||
|
|
||||||
|
import org.testng.annotations.Test;
|
||||||
|
import org.virtualbox_4_1.IConsole;
|
||||||
|
import org.virtualbox_4_1.IMachine;
|
||||||
|
import org.virtualbox_4_1.IProgress;
|
||||||
|
import org.virtualbox_4_1.ISession;
|
||||||
|
import org.virtualbox_4_1.ISnapshot;
|
||||||
|
import org.virtualbox_4_1.IVirtualBox;
|
||||||
|
import org.virtualbox_4_1.VirtualBoxManager;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Andrea Turli
|
||||||
|
*/
|
||||||
|
@Test(groups = "unit", testName = "TakeSnapshotIfNotAlreadyAttachedTest")
|
||||||
|
public class TakeSnapshotIfNotAlreadyAttachedTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testTakeSnapshotIfNotAlreadyAttached() throws Exception {
|
||||||
|
|
||||||
|
String snapshotName = "snap";
|
||||||
|
String snapshotDesc = "snapDesc";
|
||||||
|
|
||||||
|
VirtualBoxManager manager = createNiceMock(VirtualBoxManager.class);
|
||||||
|
IMachine machine = createMock(IMachine.class);
|
||||||
|
IVirtualBox vBox = createMock(IVirtualBox.class);
|
||||||
|
ISession session = createMock(ISession.class);
|
||||||
|
IConsole console = createNiceMock(IConsole.class);
|
||||||
|
IProgress progress = createNiceMock(IProgress.class);
|
||||||
|
ISnapshot snapshot = createNiceMock(ISnapshot.class);
|
||||||
|
expect(machine.getCurrentSnapshot()).andReturn(snapshot).anyTimes();
|
||||||
|
|
||||||
|
expect(manager.openMachineSession(machine)).andReturn(session);
|
||||||
|
|
||||||
|
expect(session.getConsole()).andReturn(console);
|
||||||
|
expect(console.takeSnapshot(snapshotName, snapshotDesc)).andReturn(
|
||||||
|
progress);
|
||||||
|
expect(progress.getCompleted()).andReturn(true);
|
||||||
|
|
||||||
|
session.unlockMachine();
|
||||||
|
replay(manager, machine, vBox, session, console, progress);
|
||||||
|
|
||||||
|
new TakeSnapshotIfNotAlreadyAttached(manager, snapshotName, snapshotDesc)
|
||||||
|
.apply(machine);
|
||||||
|
|
||||||
|
verify(machine);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDoNothingIfAlreadyTakenSnapshot() throws Exception {
|
||||||
|
String snapshotName = "snap";
|
||||||
|
String snapshotDesc = "snapDesc";
|
||||||
|
|
||||||
|
VirtualBoxManager manager = createNiceMock(VirtualBoxManager.class);
|
||||||
|
IMachine machine = createMock(IMachine.class);
|
||||||
|
IVirtualBox vBox = createMock(IVirtualBox.class);
|
||||||
|
ISession session = createMock(ISession.class);
|
||||||
|
IConsole console = createNiceMock(IConsole.class);
|
||||||
|
|
||||||
|
IProgress progress = createNiceMock(IProgress.class);
|
||||||
|
expect(progress.getCompleted()).andReturn(true);
|
||||||
|
expect(machine.getCurrentSnapshot()).andReturn(null).anyTimes();
|
||||||
|
expect(manager.openMachineSession(machine)).andReturn(session);
|
||||||
|
|
||||||
|
expect(session.getConsole()).andReturn(console);
|
||||||
|
expect(console.takeSnapshot(snapshotName, snapshotDesc)).andReturn(
|
||||||
|
progress);
|
||||||
|
|
||||||
|
session.unlockMachine();
|
||||||
|
replay(manager, machine, vBox, session, console, progress);
|
||||||
|
|
||||||
|
new TakeSnapshotIfNotAlreadyAttached(manager, snapshotName, snapshotDesc)
|
||||||
|
.apply(machine);
|
||||||
|
|
||||||
|
verify(machine);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue