func createEventClient(eventAddress string) *adapter { var obcEHClient *consumer.EventsClient done := make(chan *pb.Event_Block) adapter := &adapter{notfy: done} obcEHClient = consumer.NewEventsClient(eventAddress, adapter) if err := obcEHClient.Start(); err != nil { fmt.Printf("could not start chat %s\n", err) obcEHClient.Stop() return nil } return adapter }
func createEventClient(eventAddress string, listenToRejections bool) *adapter { var obcEHClient *consumer.EventsClient done := make(chan *pb.Event_Block) reject := make(chan *pb.Event_Rejection) adapter := &adapter{notfy: done, rejected: reject, listenToRejections: listenToRejections} obcEHClient = consumer.NewEventsClient(eventAddress, adapter) if err := obcEHClient.Start(); err != nil { fmt.Printf("could not start chat %s\n", err) obcEHClient.Stop() return nil } return adapter }
func TestMain(m *testing.M) { SetupTestConfig() var opts []grpc.ServerOption if viper.GetBool("peer.tls.enabled") { creds, err := credentials.NewServerTLSFromFile(viper.GetString("peer.tls.cert.file"), viper.GetString("peer.tls.key.file")) if err != nil { grpclog.Fatalf("Failed to generate credentials %v", err) } opts = []grpc.ServerOption{grpc.Creds(creds)} } grpcServer := grpc.NewServer(opts...) //use a different address than what we usually use for "peer" //we override the peerAddress set in chaincode_support.go peerAddress = "0.0.0.0:60303" lis, err := net.Listen("tcp", peerAddress) if err != nil { fmt.Printf("Error starting events listener %s....not doing tests", err) return } // Register EventHub server // use a buffer of 100 and blocking timeout ehServer := producer.NewEventsServer(100, 0) ehpb.RegisterEventsServer(grpcServer, ehServer) fmt.Printf("Starting events server\n") go grpcServer.Serve(lis) var regTimeout = 5 * time.Second done := make(chan struct{}) adapter = &Adapter{notfy: done} obcEHClient, _ = consumer.NewEventsClient(peerAddress, regTimeout, adapter) if err = obcEHClient.Start(); err != nil { fmt.Printf("could not start chat %s\n", err) obcEHClient.Stop() return } time.Sleep(2 * time.Second) os.Exit(m.Run()) }