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())
}