fix according to comments

paddle/go/pserver/service.go

@@ -9,8 +9,7 @@ import (
 // ElementType is the type of elements of a Parameter.
 type ElementType int
 
-var ErrUnintialized = errors.New("pserver not initialized")
-var ErrAlreadyIntialized = errors.New("pserver already initialized")
+var ErrAlreadyInitialized = errors.New("pserver already initialized")
 
 // Supported element types
 const (
@@ -56,7 +55,7 @@ func NewService() *Service {
 func (s *Service) BeginInitParams(config []byte, dummy *int) error {
 	select {
 	case <-s.initialized:
-		return ErrAlreadyIntialized
+		return ErrAlreadyInitialized
 	default:
 	}
 
@@ -75,7 +74,7 @@ func (s *Service) BeginInitParams(config []byte, dummy *int) error {
 func (s *Service) InitParam(paramWithConfigs ParameterWithConfig, dummy *int) error {
 	select {
 	case <-s.initialized:
-		return ErrAlreadyIntialized
+		return ErrAlreadyInitialized
 	default:
 	}
 
@@ -94,7 +93,7 @@ func (s *Service) InitParam(paramWithConfigs ParameterWithConfig, dummy *int) er
 func (s *Service) FinishInitParams(dummy0 int, dummy1 *int) error {
 	select {
 	case <-s.initialized:
-		return ErrAlreadyIntialized
+		return ErrAlreadyInitialized
 	default:
 	}
 
@@ -103,11 +102,7 @@ func (s *Service) FinishInitParams(dummy0 int, dummy1 *int) error {
 }
 
 func (s *Service) SendGrads(grads []Gradient, dummy *int) error {
-	select {
-	case <-s.initialized:
-	default:
-		return ErrUnintialized
-	}
+	<-s.initialized
 
 	count := len(grads)
 	if count == 0 {

paddle/go/pserver/service_test.go

@@ -98,21 +98,12 @@ func TestMultipleInit(t *testing.T) {
 	}
 
 	err = s.FinishInitParams(0, &dummy)
-	if err != pserver.ErrAlreadyIntialized {
+	if err != pserver.ErrAlreadyInitialized {
 		t.FailNow()
 	}
 
 	err = s.BeginInitParams(nil, &dummy)
-	if err != pserver.ErrAlreadyIntialized {
-		t.FailNow()
-	}
-}
-
-func TestUninitialized(t *testing.T) {
-	s := pserver.NewService()
-	var dummy int
-	err := s.SendGrads(nil, &dummy)
-	if err != pserver.ErrUnintialized {
+	if err != pserver.ErrAlreadyInitialized {
 		t.FailNow()
 	}
 }
@@ -140,6 +131,16 @@ func TestBlockUntilInitialized(t *testing.T) {
 		wg.Done()
 	}()
 
+	wg.Add(1)
+	go func() {
+		var dummy int
+		err := s.SendGrads(nil, &dummy)
+		if err != nil {
+			t.FailNow()
+		}
+		wg.Done()
+	}()
+
 	var dummy int
 	err := s.BeginInitParams(nil, &dummy)
 	if err != nil {