Improve server test.

msgpack-rpc · Apr 3, 2011 · 8a3d2ff · 8a3d2ff
1 parent c01b1bb
commit 8a3d2ff
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/go/rpc/server_test.go b/go/rpc/server_test.go
@@ -33,15 +33,16 @@ func TestRun(t *testing.T) {
         return
     }
     client := NewSession(conn, true)
-    retval, xerr := client.Send("echo", "test")
-    if xerr != nil {
-        t.Error(xerr.String())
-        return
-    }
-    _retval, ok := retval.(*reflect.StringValue)
-    if !ok {
-        return
+    for _, v := range []string { "world", "test", "hey" } {
+        retval, xerr := client.Send("echo", v)
+        if xerr != nil {
+            t.Error(xerr.String())
+            return
+        }
+        _retval, ok := retval.(*reflect.StringValue)
+        if !ok {
+            return
+        }
+        if _retval.Get() != "Hello, " + v { t.Error("retval != \"Hello, " + v + "\"") }
     }
-    if _retval.Get() != "Hello, test" { t.Error("retval != \"Hello, test\"") }
-
 }