You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@jclouds.apache.org by an...@apache.org on 2014/08/14 18:49:30 UTC

git commit: JCLOUDS-661: Adding a test for single-port firewall rules

Repository: jclouds
Updated Branches:
  refs/heads/master 7d7036259 -> f3778b6cb


JCLOUDS-661: Adding a test for single-port firewall rules


Project: http://git-wip-us.apache.org/repos/asf/jclouds/repo
Commit: http://git-wip-us.apache.org/repos/asf/jclouds/commit/f3778b6c
Tree: http://git-wip-us.apache.org/repos/asf/jclouds/tree/f3778b6c
Diff: http://git-wip-us.apache.org/repos/asf/jclouds/diff/f3778b6c

Branch: refs/heads/master
Commit: f3778b6cbded6c01acfe7a4fa1379afcede3f524
Parents: 7d70362
Author: Stuart Hendren <st...@tenode.com>
Authored: Tue Aug 12 16:34:36 2014 +0100
Committer: Andrew Phillips <an...@apache.org>
Committed: Thu Aug 14 12:49:06 2014 -0400

----------------------------------------------------------------------
 .../BaseSecurityGroupExtensionLiveTest.java     | 57 ++++++++++++--------
 1 file changed, 35 insertions(+), 22 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/jclouds/blob/f3778b6c/compute/src/test/java/org/jclouds/compute/extensions/internal/BaseSecurityGroupExtensionLiveTest.java
----------------------------------------------------------------------
diff --git a/compute/src/test/java/org/jclouds/compute/extensions/internal/BaseSecurityGroupExtensionLiveTest.java b/compute/src/test/java/org/jclouds/compute/extensions/internal/BaseSecurityGroupExtensionLiveTest.java
index 5c461eb..258b8cb 100644
--- a/compute/src/test/java/org/jclouds/compute/extensions/internal/BaseSecurityGroupExtensionLiveTest.java
+++ b/compute/src/test/java/org/jclouds/compute/extensions/internal/BaseSecurityGroupExtensionLiveTest.java
@@ -135,21 +135,18 @@ public abstract class BaseSecurityGroupExtensionLiveTest extends BaseComputeServ
 
       SecurityGroup group = optGroup.get();
 
-      IpPermission.Builder builder = IpPermission.builder();
-
-      builder.ipProtocol(IpProtocol.TCP);
-      builder.fromPort(10);
-      builder.toPort(20);
-      builder.cidrBlock("0.0.0.0/0");
+      IpPermission portRangeIpPermission = createPortRangePermission();
+      IpPermission singlePortIpPermission = createSinglePortPermission();
 
-      IpPermission perm = builder.build();
+      Set<IpPermission> expectedPermissions = ImmutableSet.of(portRangeIpPermission, singlePortIpPermission);
 
-      SecurityGroup newGroup = securityGroupExtension.get().addIpPermission(perm, group);
+      SecurityGroup onePermissionAdded = securityGroupExtension.get().addIpPermission(portRangeIpPermission, group);
+      SecurityGroup twoPermissionsAdded = securityGroupExtension.get().addIpPermission(singlePortIpPermission,
+            onePermissionAdded);
 
-      assertEquals(Iterables.getOnlyElement(newGroup.getIpPermissions()), perm,
-              "Expecting IpPermission " + perm + " but group was " + newGroup);
+      assertEquals(twoPermissionsAdded.getIpPermissions(), expectedPermissions);
    }
-   
+
    @Test(groups = { "integration", "live" }, singleThreaded = true, dependsOnMethods = "testAddIpPermission")
    public void testRemoveIpPermission() {
       skipIfSecurityGroupsNotSupported();
@@ -165,19 +162,14 @@ public abstract class BaseSecurityGroupExtensionLiveTest extends BaseComputeServ
 
       SecurityGroup group = optGroup.get();
 
-      IpPermission.Builder builder = IpPermission.builder();
-
-      builder.ipProtocol(IpProtocol.TCP);
-      builder.fromPort(10);
-      builder.toPort(20);
-      builder.cidrBlock("0.0.0.0/0");
-
-      IpPermission perm = builder.build();
+      IpPermission portRangeIpPermission = createPortRangePermission();
+      IpPermission singlePortIpPermission = createSinglePortPermission();
 
-      SecurityGroup newGroup = securityGroupExtension.get().removeIpPermission(perm, group);
+      SecurityGroup newGroup = securityGroupExtension.get().removeIpPermission(portRangeIpPermission, group);
+      SecurityGroup emptyGroup = securityGroupExtension.get().removeIpPermission(singlePortIpPermission, newGroup);
 
-      assertEquals(Iterables.size(newGroup.getIpPermissions()), 0,
-              "Group should have no permissions, but has " + Iterables.size(newGroup.getIpPermissions()));
+      assertEquals(Iterables.size(emptyGroup.getIpPermissions()), 0, "Group should have no permissions, but has "
+            + Iterables.size(emptyGroup.getIpPermissions()));
    }
 
    @Test(groups = { "integration", "live" }, singleThreaded = true, dependsOnMethods = "testRemoveIpPermission")
@@ -355,6 +347,27 @@ public abstract class BaseSecurityGroupExtensionLiveTest extends BaseComputeServ
       });
    }
 
+   private IpPermission createPortRangePermission() {
+      IpPermission.Builder builder = IpPermission.builder();
+
+      builder.ipProtocol(IpProtocol.TCP);
+      builder.fromPort(10);
+      builder.toPort(20);
+      builder.cidrBlock("0.0.0.0/0");
+
+      return builder.build();
+   }
+
+   private IpPermission createSinglePortPermission() {
+      IpPermission.Builder builder = IpPermission.builder();
+
+      builder.ipProtocol(IpProtocol.UDP);
+      builder.fromPort(41);
+      builder.toPort(41);
+      builder.cidrBlock("1.1.1.1/32");
+
+      return builder.build();
+   }
 
    private void cleanup() {
       if (securityGroupsSupported) {