Merge pull request #9 from it-chain/feature/avengers

fix listening network manager consume logic

mock/network_manager.go

@@ -134,15 +134,19 @@ func (n *NetworkManager) Start() {
 	for id, process := range n.ProcessMap {
 		go func(id string, process *Process) {
 			logger.Infof(nil, "process %s is running", process.Id)
-			logger.Infof(nil, "channel %s is %o", process.Id, process.GrpcCommandReceiver)
-			select {
-			case message := <-process.GrpcCommandReceiver:
-				for _, handler := range process.GrpcCommandHandlers {
-					handler(message)
+			end := true
+			for end {
+				select {
+				case message := <-process.GrpcCommandReceiver:
+					logger.Infof(nil, "receive message from : %s message: %v", id, message)
+					for _, handler := range process.GrpcCommandHandlers {
+						handler(message)
+					}
+
+				case <-time.After(4 * time.Second):
+					logger.Info(nil, "failed to consume, timed out!")
+					end = false
 				}
-
-			case <-time.After(4 * time.Second):
-				logger.Info(nil, "failed to consume, timed out!")
 			}
 		}(id, process)
 	}

mock/network_manager_test.go

@@ -158,11 +158,12 @@ func TestNetworkManager_Start(t *testing.T) {
 		RecipientList: []string{"2", "3"},
 	}
 
+	net.Publish("1", "message.deliver", command)
 	net.Publish("1", "message.deliver", command)
 
 	time.Sleep(2 * time.Second)
 
-	assert.Equal(t, mem(), 3)
+	assert.Equal(t, mem(), 5)
 }
 
 func SetNetworkManager(closerMemory func() int) *mock.NetworkManager {