fix up

Signed-off-by: Sam Yuan <yy19902439@126.com>

e2e/mock/peer.go

@@ -14,7 +14,7 @@ type Peer struct {
 	GrpcServer     *grpc.Server
 	BlkSize, txCnt uint64
 	TxC            chan struct{}
-	CtlCh          chan struct{}
+	CtlCh          chan bool
 }
 
 func (p *Peer) ProcessProposal(context.Context, *peer.SignedProposal) (*peer.ProposalResponse, error) {
@@ -42,8 +42,8 @@ func (p *Peer) DeliverFiltered(srv peer.Deliver_DeliverFilteredServer) error {
 						Number:               p.txCnt / p.BlkSize,
 						FilteredTransactions: make([]*peer.FilteredTransaction, p.BlkSize)}}})
 			}
-		case <-p.CtlCh:
-			if txc != nil {
+		case action := <-p.CtlCh:
+			if action {
 				txc = nil
 			} else {
 				txc = p.TxC
@@ -74,7 +74,7 @@ func NewPeer(TxC chan struct{}, credentials credentials.TransportCredentials) (*
 	if err != nil {
 		return nil, err
 	}
-	CtlCh := make(chan struct{})
+	CtlCh := make(chan bool)
 	instance := &Peer{
 		Listener:   lis,
 		GrpcServer: grpc.NewServer(grpc.Creds(credentials)),
@@ -89,6 +89,10 @@ func NewPeer(TxC chan struct{}, credentials credentials.TransportCredentials) (*
 	return instance, nil
 }
 
-func (p *Peer) Toggle() {
-	p.CtlCh <- struct{}{}
+func (p *Peer) Pause() {
+	p.CtlCh <- true
+}
+
+func (p *Peer) Unpause() {
+	p.CtlCh <- false
 }

pkg/infra/observer_test.go

@@ -152,11 +152,11 @@ var _ = Describe("Observer", func() {
 			}(i)
 		}
 		for i := 0; i < COMMIT_THRESHOLD; i++ {
-			mpeers[i].Toggle()
+			mpeers[i].Pause()
 		}
 		Consistently(finishCh).ShouldNot(Receive())
 		for i := 0; i < COMMIT_THRESHOLD; i++ {
-			mpeers[i].Toggle()
+			mpeers[i].Unpause()
 		}
 		<-finishCh
 	})