From ef8d0759d544c36866d628170be8dbb9c0ede4eb Mon Sep 17 00:00:00 2001
From: Travis <travis@pilosa.com>
Date: Tue, 12 Jan 2021 23:04:49 -0600
Subject: [PATCH] add retry to pg test ServerTLS()

---
 pg/pgtest/server.go | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/pg/pgtest/server.go b/pg/pgtest/server.go
index e3a48e244..52cfdf82f 100644
--- a/pg/pgtest/server.go
+++ b/pg/pgtest/server.go
@@ -16,6 +16,7 @@ package pgtest
 
 import (
 	"context"
+	"fmt"
 	"net"
 	"testing"
 
@@ -64,7 +65,19 @@ func ServeTLS(addr string, server *pg.Server) (net.Addr, ShutdownFunc, error) {
 		return nil, nil, errors.Wrap(err, "server TLS setup failed")
 	}
 
-	return ServeTCP(addr, server)
+	var tries int = 5
+	var netAddr net.Addr
+	var shutdown ShutdownFunc
+
+	for i := 0; i < tries; i++ {
+		if i > 0 {
+			fmt.Printf("--- try serving TLS again: %d\n", i)
+		}
+		if netAddr, shutdown, err = ServeTCP(addr, server); err == nil {
+			break
+		}
+	}
+	return netAddr, shutdown, err
 }
 
 // ConnectFunc is a function to connect to a server.