Actually use the args to SASLConnectReconnect; shutdown nicely.

ingenthr · ingenthr · commit 3483b2aa1db7 · 2010-04-12T15:59:50.000-07:00
Change-Id: I7517a355707ed9440974d5b5b947fb13a8ba7142
diff --git a/src/test/manual/net/spy/memcached/test/SASLConnectReconnect.java b/src/test/manual/net/spy/memcached/test/SASLConnectReconnect.java
@@ -3,6 +3,7 @@
 import java.io.IOException;
 import java.net.InetSocketAddress;
 import java.util.List;
+import java.util.concurrent.TimeUnit;
 import java.util.logging.ConsoleHandler;
 import java.util.logging.Handler;
 import java.util.logging.Level;
@@ -95,7 +96,7 @@ public static void main(String[] args) throws InterruptedException {
 	m.verifySetAndGet();
 	System.err.println("Pass one done.");
 	Thread.sleep(60000);
-	m.verifySetAndGet2();
+	m.verifySetAndGet2(Integer.parseInt(args[3]));
 	System.err.println("Pass two done.");
 
     }
@@ -118,10 +119,9 @@ public void verifySetAndGet() {
     /**
      * verify set and get go to the right place
      */
-    public void verifySetAndGet2() {
+    public void verifySetAndGet2(int iterations) {
 	try {
-	    int iterations = 50000;
-	    for (int i = 0; i < iterations; i++) {
+	    for (int i = 0; i <= iterations; i++) {
 		mc.set("me" + i, 0, "me" + i);
 	    }
 
@@ -137,6 +137,7 @@ public void verifySetAndGet2() {
 		    System.err.println("Operation timeed out, continuing.");
 		}
 	    }
+	mc.shutdown(1, TimeUnit.SECONDS);
 	} catch (Exception ex) {
 	    System.err.println("Bailing out " + ex.toString() + "\n");
 	    ex.printStackTrace();
