fix pserver test

Author: helinwang

go/pserver/client_test.go

@@ -31,7 +31,7 @@ func init() {
 port[i] = p
 
 go func(l net.Listener) {
-s, err := pserver.NewService("", time.Second*5)
+s, err := pserver.NewService("", 1, time.Second*5)
 if err != nil {
 panic(err)
 }

go/pserver/service_test.go

@@ -10,7 +10,7 @@ import (
 )
 
 func TestFull(t *testing.T) {
-s, err := pserver.NewService("", time.Second*5)
+s, err := pserver.NewService("", 1, time.Second*5)
 if err != nil {
 t.Error(err)
 }
@@ -75,7 +75,7 @@ func TestFull(t *testing.T) {
 }
 
 func TestMultipleInit(t *testing.T) {
-s, err := pserver.NewService("", time.Second*5)
+s, err := pserver.NewService("", 1, time.Second*5)
 if err != nil {
 t.Error(err)
 }
@@ -91,15 +91,15 @@ func TestMultipleInit(t *testing.T) {
 }
 
 func TestUninitialized(t *testing.T) {
-s, err := pserver.NewService("", time.Second*5)
+s, err := pserver.NewService("", 1, time.Second*5)
 err = s.SendGrad(pserver.Gradient{}, nil)
 if err.Error() != pserver.Uninitialized {
 t.FailNow()
 }
 }
 
 func TestBlockUntilInitialized(t *testing.T) {
-s, err := pserver.NewService("", time.Second*5)
+s, err := pserver.NewService("", 1, time.Second*5)
 if err != nil {
 t.Error(err)
 }