/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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.apache.whirr.service.jclouds.integration;
import com.google.common.collect.Sets;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.apache.whirr.Cluster;
import org.apache.whirr.ClusterSpec;
import org.apache.whirr.service.ComputeCache;
import org.apache.whirr.service.jclouds.FirewallSettings;
import org.jclouds.compute.ComputeServiceContext;
import org.jclouds.domain.Credentials;
import org.jclouds.ec2.EC2Client;
import org.junit.BeforeClass;
import org.junit.Test;
import java.io.IOException;
import java.util.Set;
public class FirewallSettingsTest {
private static final String REGION = "us-east-1";
private static ClusterSpec spec;
private static Set<Cluster.Instance> instances;
private static ComputeServiceContext context;
private static ClusterSpec getTestClusterSpec() throws Exception {
return ClusterSpec.withTemporaryKeys(
new PropertiesConfiguration("whirr-core-test.properties"));
}
@BeforeClass
public static void setUpClass() throws Exception {
spec = getTestClusterSpec();
context = ComputeCache.INSTANCE.apply(spec);
/* create a dummy instance for testing */
instances = Sets.newHashSet(new Cluster.Instance(
new Credentials("dummy", "dummy"),
Sets.newHashSet("dummy-role"),
"50.0.0.1",
"10.0.0.1",
REGION + "/i-dummy",
null
));
}
@Test
public void testFirewallAuthorizationIsIdempotent() throws IOException {
if (context.getProviderSpecificContext().getApi() instanceof EC2Client) {
EC2Client ec2Client = EC2Client.class.cast(
context.getProviderSpecificContext().getApi());
String groupName = "jclouds#" + spec.getClusterName() + "#" + REGION;
ec2Client.getSecurityGroupServices()
.createSecurityGroupInRegion(REGION, groupName, "group description");
try {
FirewallSettings.authorizeIngress(context, instances, spec, 23344);
/* The second call should not throw an exception. */
FirewallSettings.authorizeIngress(context, instances, spec, 23344);
} finally {
ec2Client.getSecurityGroupServices()
.deleteSecurityGroupInRegion(REGION, groupName);
}
}
}
}