improved test case

stephank · Mar 22, 2011 · 1f795b4 · 1f795b4
1 parent 9589761
commit 1f795b4
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/test/stress_buffer.js b/test/stress_buffer.js
@@ -20,8 +20,8 @@ function sock1Message(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9)
     assert.equal(arg8, "to");
     assert.equal(arg9, "you");
 
-    console.log("got message on sock1: ", arg1 ? arg1.toString() : "", arg2 ? arg2.toString() : "",
-                arg3 ? arg3.toString() : "");
+    //console.log("got message on sock1: ", arg1 ? arg1.toString() : "", arg2 ? arg2.toString() : "",
+    //            arg3 ? arg3.toString() : "");
     //sock1.send('message back');
 }
 
@@ -38,8 +38,8 @@ function sock2Message(arg1, arg2, arg3)
     assert.equal(arg1.toString(), "Sock1");
     assert.equal(arg2.toString(), "message");
 
-    console.log("got message on sock2: ", arg1 ? arg1.toString() : "", arg2 ? arg2.toString() : "",
-                arg3 ? arg3.toString() : "");
+    //console.log("got message on sock2: ", arg1 ? arg1.toString() : "", arg2 ? arg2.toString() : "",
+    //            arg3 ? arg3.toString() : "");
     sock2.send(arg1, arg2, 'this', 'is', 'a', 'long', 'message', 'back', 'to', 'you');
     //console.log("got message on sock2: ", arg1, arg2, arg3);
     //sock2.send(arg1, 'hi', 'back');
@@ -60,20 +60,31 @@ sock2.on('error', sock2Error);
 
 var msg = 0;
 
+var intervalId;
+
 function onTick()
 {
     for (var i = 0;  i < 1000;  ++i) {
         sock1.send('message', "" + msg);
         ++msg;
     }
+
+    console.log("done " + msg + " messages");
+
+    if (msg > 1000000) {
+        clearInterval(intervalId);
+        sock1.close();
+        sock2.close();
+    }
+
 }
 
 function onBindDone()
 {
     sock1.connect(uri);
 
     //sock1.send('hello', 'world');
-    var intervalId = setInterval(onTick, 1);
+    intervalId = setInterval(onTick, 1);
 }
 
 sock2.bind(uri, onBindDone);