Slightly improve average calculation by using length of lats. Use eph…

…emeral port for tests

reporter.go

@@ -83,7 +83,9 @@ type ResultDetail struct {
 }
 
 func newReporter(results chan *callResult, options *Options) *Reporter {
+
 	cap := min(options.N, maxResult)
+
 	return &Reporter{
 		options:        options,
 		results:        results,
@@ -123,7 +125,7 @@ func (r *Reporter) Run() {
 
 // Finalize all the gathered data into a final report
 func (r *Reporter) Finalize(total time.Duration) *Report {
-	average := r.avgTotal / float64(r.totalCount)
+	average := r.avgTotal / float64(len(r.lats))
 	avgDuration := time.Duration(average * float64(time.Second))
 	rps := float64(r.totalCount) / total.Seconds()
 

requester_test.go

@@ -2,6 +2,7 @@ package ghz
 
 import (
 	"net"
+	"strconv"
 	"sync"
 	"testing"
 	"time"
@@ -15,11 +16,11 @@ import (
 	"google.golang.org/grpc/credentials"
 )
 
-const port = ":50051"
-const localhost = "localhost:50051"
+var port string
+var localhost string
 
 func startServer(secure bool) (*helloworld.Greeter, *grpc.Server, error) {
-	lis, err := net.Listen("tcp", port)
+	lis, err := net.Listen("tcp", ":0")
 	if err != nil {
 		return nil, nil, err
 	}
@@ -39,9 +40,14 @@ func startServer(secure bool) (*helloworld.Greeter, *grpc.Server, error) {
 	gs := helloworld.NewGreeter()
 	helloworld.RegisterGreeterServer(s, gs)
 	// reflection.Register(s)
+
+	port = strconv.Itoa(lis.Addr().(*net.TCPAddr).Port)
+	localhost = "localhost:" + port
+
 	go func() {
 		s.Serve(lis)
 	}()
+
 	return gs, s, err
 }
 

stats_handler_test.go

@@ -10,8 +10,6 @@ import (
 	"google.golang.org/grpc"
 )
 
-const address = "localhost:50051"
-
 func TestStatsHandler(t *testing.T) {
 	_, s, err := startServer(false)
 
@@ -32,7 +30,7 @@ func TestStatsHandler(t *testing.T) {
 		done <- true
 	}()
 
-	conn, err := grpc.Dial(address, grpc.WithInsecure(), grpc.WithStatsHandler(&statsHandler{rChan}))
+	conn, err := grpc.Dial(localhost, grpc.WithInsecure(), grpc.WithStatsHandler(&statsHandler{rChan}))
 
 	if err != nil {
 		assert.FailNow(t, err.Error())