From 5bd59d50cdd11efe3bd30c1caac22d02289f464b Mon Sep 17 00:00:00 2001
From: Adrian Cole <adrian@jclouds.org>
Date: Thu, 10 May 2012 19:29:19 -0700
Subject: [PATCH] fixed test configuration

---
 .../jclouds/aws/ec2/services/AWSKeyPairClientLiveTest.java | 5 +++++
 .../aws/ec2/services/PlacementGroupClientLiveTest.java     | 7 +++++++
 2 files changed, 12 insertions(+)

diff --git a/providers/aws-ec2/src/test/java/org/jclouds/aws/ec2/services/AWSKeyPairClientLiveTest.java b/providers/aws-ec2/src/test/java/org/jclouds/aws/ec2/services/AWSKeyPairClientLiveTest.java
index eb4f4f8ddf..15a4030046 100644
--- a/providers/aws-ec2/src/test/java/org/jclouds/aws/ec2/services/AWSKeyPairClientLiveTest.java
+++ b/providers/aws-ec2/src/test/java/org/jclouds/aws/ec2/services/AWSKeyPairClientLiveTest.java
@@ -48,6 +48,7 @@ import org.jclouds.compute.options.TemplateOptions;
 import org.jclouds.domain.LoginCredentials;
 import org.jclouds.ec2.domain.KeyPair;
 import org.jclouds.logging.log4j.config.Log4JLoggingModule;
+import org.jclouds.sshj.config.SshjSshClientModule;
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.Test;
 
@@ -202,4 +203,8 @@ public class AWSKeyPairClientLiveTest extends BaseComputeServiceContextLiveTest
       return getOnlyElement(getOnlyElement(instanceClient.describeInstancesInRegion(null, id)));
    }
    
+   @Override
+   protected Module getSshModule() {
+      return new SshjSshClientModule();
+   }
 }
diff --git a/providers/aws-ec2/src/test/java/org/jclouds/aws/ec2/services/PlacementGroupClientLiveTest.java b/providers/aws-ec2/src/test/java/org/jclouds/aws/ec2/services/PlacementGroupClientLiveTest.java
index facb340cf0..74b2be2578 100644
--- a/providers/aws-ec2/src/test/java/org/jclouds/aws/ec2/services/PlacementGroupClientLiveTest.java
+++ b/providers/aws-ec2/src/test/java/org/jclouds/aws/ec2/services/PlacementGroupClientLiveTest.java
@@ -46,11 +46,13 @@ import org.jclouds.predicates.RetryablePredicate;
 import org.jclouds.scriptbuilder.domain.Statements;
 import org.jclouds.scriptbuilder.statements.java.InstallJDK;
 import org.jclouds.scriptbuilder.statements.login.AdminAccess;
+import org.jclouds.sshj.config.SshjSshClientModule;
 import org.testng.annotations.AfterTest;
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.Test;
 
 import com.google.common.base.Throwables;
+import com.google.inject.Module;
 
 /**
  * Tests behavior of {@code PlacementGroupClient}
@@ -168,4 +170,9 @@ public class PlacementGroupClientLiveTest extends BaseComputeServiceContextLiveT
          assert deletedTester.apply(group) : group;
       }
    }
+   
+   @Override
+   protected Module getSshModule() {
+      return new SshjSshClientModule();
+   }
 }