@Test(enabled = true)
public void testRegisterTemplate() throws Exception {
Zone zone = Iterables.getFirst(client.getZoneClient().listZones(), null);
assertNotNull(zone);
Iterable<Network> networks =
client
.getNetworkClient()
.listNetworks(ListNetworksOptions.Builder.zoneId(zone.getId()).isDefault(true));
networks =
Iterables.filter(
networks,
new Predicate<Network>() {
@Override
public boolean apply(@Nullable Network network) {
return network != null && network.getState().equals("Implemented");
}
});
assertEquals(Iterables.size(networks), 1);
Network network = Iterables.getOnlyElement(networks, null);
assertNotNull(network);
Set<OSType> osTypes = client.getGuestOSClient().listOSTypes();
OSType osType = Iterables.getFirst(osTypes, null);
// Register a template
RegisterTemplateOptions options = RegisterTemplateOptions.Builder.bits(32).isExtractable(true);
TemplateMetadata templateMetadata =
TemplateMetadata.builder()
.name(prefix + "-registerTemplate")
.osTypeId(osType.getId())
.displayText("jclouds live testRegisterTemplate")
.build();
Set<Template> templates =
client
.getTemplateClient()
.registerTemplate(
templateMetadata, "VHD", "XenServer", IMPORT_VHD_URL, zone.getId(), options);
registeredTemplate = Iterables.getOnlyElement(templates, null);
assertNotNull(registeredTemplate);
// Ensure it is available
final long zoneId = zone.getId();
Predicate<Template> templateReadyPredicate =
new Predicate<Template>() {
@Override
public boolean apply(@Nullable Template template) {
if (template == null) return false;
Template t2 = client.getTemplateClient().getTemplateInZone(template.getId(), zoneId);
Logger.CONSOLE.info("%s", t2.getStatus());
return "Download Complete".equals(t2.getStatus());
}
};
assertTrue(
new RetryablePredicate<Template>(templateReadyPredicate, 60000).apply(registeredTemplate));
// Create a VM that uses this template
vmForRegistration =
VirtualMachineClientLiveTest.createVirtualMachineInNetwork(
network, registeredTemplate.getId(), client, jobComplete, virtualMachineRunning);
assertNotNull(vmForRegistration);
}
@AfterGroups(groups = "live")
protected void tearDown() {
if (vmForCreation != null) {
assertTrue(
jobComplete.apply(
client.getVirtualMachineClient().stopVirtualMachine(vmForCreation.getId())),
vmForCreation.toString());
assertTrue(
jobComplete.apply(
client.getVirtualMachineClient().destroyVirtualMachine(vmForCreation.getId())),
vmForCreation.toString());
assertTrue(virtualMachineDestroyed.apply(vmForCreation));
}
if (vmForRegistration != null) {
assertTrue(
jobComplete.apply(
client.getVirtualMachineClient().stopVirtualMachine(vmForRegistration.getId())),
vmForRegistration.toString());
assertTrue(
jobComplete.apply(
client.getVirtualMachineClient().destroyVirtualMachine(vmForRegistration.getId())),
vmForRegistration.toString());
assert virtualMachineDestroyed.apply(vmForRegistration);
}
if (createdTemplate != null) {
AsyncCreateResponse deleteJob =
client.getTemplateClient().deleteTemplate(createdTemplate.getId());
assertTrue(jobComplete.apply(deleteJob.getJobId()));
}
if (registeredTemplate != null) {
AsyncCreateResponse deleteJob =
client.getTemplateClient().deleteTemplate(registeredTemplate.getId());
assertTrue(jobComplete.apply(deleteJob.getJobId()));
}
super.tearDown();
}
@Test(enabled = true, dependsOnMethods = "testRegisterTemplate")
public void testExtractTemplate() throws Exception {
// Initiate the extraction and wait for it to complete
AsyncCreateResponse response =
client
.getTemplateClient()
.extractTemplate(
registeredTemplate.getId(),
ExtractMode.HTTP_DOWNLOAD,
registeredTemplate.getZoneId());
assertTrue(jobComplete.apply(response.getJobId()), registeredTemplate.toString());
// Get the result
AsyncJob<TemplateExtraction> asyncJob =
client.getAsyncJobClient().getAsyncJob(response.getJobId());
TemplateExtraction extract = asyncJob.getResult();
assertNotNull(extract);
// Check that the URL can be retrieved
String extractUrl = extract.getUrl();
assertNotNull(extractUrl);
URI uri = new URI(URLDecoder.decode(extractUrl, "utf-8"));
assertTrue(context.utils().http().exists(uri), "does not exist: " + uri);
}
@Test
public void testListTemplates() throws Exception {
Set<Template> response = client.getTemplateClient().listTemplates();
assert null != response;
long templateCount = response.size();
assertTrue(templateCount >= 0);
for (Template template : response) {
Template newDetails =
Iterables.getOnlyElement(
client
.getTemplateClient()
.listTemplates(zoneId(template.getZoneId()).id(template.getId())));
Logger.CONSOLE.info("Checking template: " + template);
assertEquals(template, newDetails);
assertEquals(
template,
client.getTemplateClient().getTemplateInZone(template.getId(), template.getZoneId()));
assert template.getId() > 0 : template;
assert template.getName() != null : template;
assert template.getDisplayText() != null : template;
assert template.getCreated() != null : template;
assert template.getFormat() != null && template.getFormat() != Template.Format.UNRECOGNIZED
: template;
assert template.getOSType() != null : template;
assert template.getOSTypeId() > 0 : template;
assert template.getAccount() != null : template;
assert template.getZone() != null : template;
assert template.getZoneId() > 0 : template;
assert (template.getStatus() == null || template.getStatus().equals("Download Complete"))
: template;
assert template.getType() != null && template.getType() != Template.Type.UNRECOGNIZED
: template;
assert template.getHypervisor() != null : template;
assert template.getDomain() != null : template;
assert template.getDomainId() > 0 : template;
assert template.getSize() > 0 : template;
}
}
