| /** |
| * 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.aws.ec2.services; |
| |
| import static com.google.common.collect.Iterables.getOnlyElement; |
| import static com.google.common.collect.Lists.newArrayList; |
| import static com.google.common.collect.Sets.newTreeSet; |
| import static org.jclouds.compute.ComputeTestUtils.buildScript; |
| import static org.testng.Assert.assertEquals; |
| import static org.testng.Assert.assertNotNull; |
| |
| import java.io.FileNotFoundException; |
| import java.io.IOException; |
| import java.util.Properties; |
| import java.util.Set; |
| import java.util.SortedSet; |
| import java.util.concurrent.TimeUnit; |
| |
| import org.jclouds.aws.domain.Region; |
| import org.jclouds.aws.ec2.AWSEC2Client; |
| import org.jclouds.aws.ec2.domain.PlacementGroup; |
| import org.jclouds.aws.ec2.domain.PlacementGroup.State; |
| import org.jclouds.aws.ec2.predicates.PlacementGroupAvailable; |
| import org.jclouds.aws.ec2.predicates.PlacementGroupDeleted; |
| import org.jclouds.compute.BaseVersionedServiceLiveTest; |
| import org.jclouds.compute.ComputeServiceContext; |
| import org.jclouds.compute.ComputeServiceContextFactory; |
| import org.jclouds.compute.RunNodesException; |
| import org.jclouds.compute.domain.NodeMetadata; |
| import org.jclouds.compute.domain.OsFamily; |
| import org.jclouds.compute.domain.Template; |
| import org.jclouds.compute.predicates.NodePredicates; |
| import org.jclouds.ec2.domain.InstanceType; |
| import org.jclouds.logging.log4j.config.Log4JLoggingModule; |
| import org.jclouds.predicates.RetryablePredicate; |
| import org.jclouds.scriptbuilder.domain.Statements; |
| import org.jclouds.scriptbuilder.statements.login.AdminAccess; |
| import org.jclouds.sshj.config.SshjSshClientModule; |
| import org.testng.annotations.AfterTest; |
| import org.testng.annotations.BeforeGroups; |
| import org.testng.annotations.Test; |
| |
| import com.google.common.base.Throwables; |
| import com.google.common.collect.ImmutableSet; |
| import com.google.inject.Module; |
| |
| /** |
| * Tests behavior of {@code PlacementGroupClient} |
| * |
| * @author Adrian Cole |
| */ |
| @Test(groups = "live", singleThreaded = true, testName = "PlacementGroupClientLiveTest") |
| public class PlacementGroupClientLiveTest extends BaseVersionedServiceLiveTest { |
| public PlacementGroupClientLiveTest() { |
| provider = "aws-ec2"; |
| } |
| |
| private AWSEC2Client client; |
| private ComputeServiceContext context; |
| private RetryablePredicate<PlacementGroup> availableTester; |
| private RetryablePredicate<PlacementGroup> deletedTester; |
| private PlacementGroup group; |
| |
| @BeforeGroups(groups = { "live" }) |
| public void setupClient() throws FileNotFoundException, IOException { |
| setupCredentials(); |
| Properties overrides = setupProperties(); |
| context = new ComputeServiceContextFactory().createContext(provider, |
| ImmutableSet.<Module> of(new Log4JLoggingModule(), new SshjSshClientModule()), overrides); |
| |
| client = AWSEC2Client.class.cast(context.getProviderSpecificContext().getApi()); |
| |
| availableTester = new RetryablePredicate<PlacementGroup>(new PlacementGroupAvailable(client), 60, 1, |
| TimeUnit.SECONDS); |
| |
| deletedTester = new RetryablePredicate<PlacementGroup>(new PlacementGroupDeleted(client), 60, 1, TimeUnit.SECONDS); |
| } |
| |
| @Test |
| void testDescribe() { |
| for (String region : newArrayList(Region.US_EAST_1)) { |
| SortedSet<PlacementGroup> allResults = newTreeSet(client.getPlacementGroupServices() |
| .describePlacementGroupsInRegion(region)); |
| assertNotNull(allResults); |
| if (allResults.size() >= 1) { |
| PlacementGroup group = allResults.last(); |
| SortedSet<PlacementGroup> result = newTreeSet(client.getPlacementGroupServices() |
| .describePlacementGroupsInRegion(region, group.getName())); |
| assertNotNull(result); |
| PlacementGroup compare = result.last(); |
| assertEquals(compare, group); |
| } |
| } |
| |
| for (String region : client.getAvailabilityZoneAndRegionServices().describeRegions().keySet()) { |
| if (!region.equals(Region.US_EAST_1)) |
| try { |
| client.getPlacementGroupServices().describePlacementGroupsInRegion(region); |
| assert false : "should be unsupported"; |
| } catch (UnsupportedOperationException e) { |
| } |
| } |
| } |
| |
| @Test |
| void testCreatePlacementGroup() { |
| String groupName = PREFIX + "1"; |
| client.getPlacementGroupServices().deletePlacementGroupInRegion(null, groupName); |
| client.getPlacementGroupServices().createPlacementGroupInRegion(null, groupName); |
| |
| verifyPlacementGroup(groupName); |
| } |
| |
| private void verifyPlacementGroup(String groupName) { |
| assert availableTester.apply(new PlacementGroup(Region.US_EAST_1, groupName, "cluster", State.PENDING)) : group; |
| Set<PlacementGroup> oneResult = client.getPlacementGroupServices().describePlacementGroupsInRegion(null, |
| groupName); |
| assertNotNull(oneResult); |
| assertEquals(oneResult.size(), 1); |
| group = oneResult.iterator().next(); |
| assertEquals(group.getName(), groupName); |
| assertEquals(group.getStrategy(), "cluster"); |
| assert availableTester.apply(group) : group; |
| } |
| |
| public void testStartCCInstance() throws Exception { |
| |
| Template template = context.getComputeService().templateBuilder().fastest().osFamily(OsFamily.AMZN_LINUX).build(); |
| assert template != null : "The returned template was null, but it should have a value."; |
| assertEquals(template.getHardware().getProviderId(), InstanceType.CC2_8XLARGE); |
| assertEquals(template.getImage().getUserMetadata().get("rootDeviceType"), "ebs"); |
| assertEquals(template.getImage().getUserMetadata().get("virtualizationType"), "hvm"); |
| assertEquals(template.getImage().getUserMetadata().get("hypervisor"), "xen"); |
| |
| template.getOptions().runScript( |
| Statements.newStatementList(AdminAccess.standard(), |
| buildScript(template.getImage().getOperatingSystem()))); |
| |
| String group = PREFIX + "cccluster"; |
| context.getComputeService().destroyNodesMatching(NodePredicates.inGroup(group)); |
| // TODO make this not lookup an explicit region |
| client.getPlacementGroupServices().deletePlacementGroupInRegion(null, "jclouds#" + group + "#us-east-1"); |
| |
| try { |
| Set<? extends NodeMetadata> nodes = context.getComputeService().createNodesInGroup(group, 1, template); |
| NodeMetadata node = getOnlyElement(nodes); |
| |
| getOnlyElement(getOnlyElement(client.getInstanceServices().describeInstancesInRegion(null, |
| node.getProviderId()))); |
| |
| } catch (RunNodesException e) { |
| System.err.println(e.getNodeErrors().keySet()); |
| Throwables.propagate(e); |
| } finally { |
| context.getComputeService().destroyNodesMatching(NodePredicates.inGroup(group)); |
| } |
| } |
| |
| public static final String PREFIX = System.getProperty("user.name") + "ec2"; |
| |
| @AfterTest |
| public void shutdown() { |
| if (group != null) { |
| client.getPlacementGroupServices().deletePlacementGroupInRegion(group.getRegion(), group.getName()); |
| assert deletedTester.apply(group) : group; |
| } |
| context.close(); |
| } |
| } |