Raise client timeout to 30s (#69)

* rpc: create a client with a 30 second timeout

* cmd: pass the transport instead of http client to rpc.New

cmd/waas-auth/main.go

@@ -40,17 +40,14 @@ func main() {
 		}
 	}
 
-	// HTTP client to use for all outgoing connections out of the enclave
-	client := &http.Client{
-		Timeout: 10 * time.Second,
-		Transport: transport.Chain(
-			baseTransport,
-			transport.SetHeader("User-Agent", "waas-authenticator/"+waasauthenticator.VERSION),
-			traceid.Transport,
-		),
-	}
+	// HTTP transport chain to use for all outgoing connections out of the enclave
+	transportChain := transport.Chain(
+		baseTransport,
+		transport.SetHeader("User-Agent", "waas-authenticator/"+waasauthenticator.VERSION),
+		traceid.Transport,
+	)
 
-	s, err := rpc.New(cfg, client)
+	s, err := rpc.New(cfg, transportChain)
 	if err != nil {
 		panic(err)
 	}

rpc/auth_test.go

@@ -341,14 +341,14 @@ func TestStytchAuth(t *testing.T) {
 		stytchServer, tok := issueAccessTokenAndRunStytchJwksServer(t, "project-123", tokBuilderFn)
 		defer stytchServer.Close()
 
-		svc := initRPCWithClient(t, &http.Client{Transport: testTransport{
+		svc := initRPCWithClient(t, testTransport{
 			RoundTripper: http.DefaultTransport,
 			modifyRequest: func(req *http.Request) {
 				if strings.Contains(req.URL.String(), "stytch.com") {
 					req.URL.Host = stytchServer.Listener.Addr().String()
 				}
 			},
-		}})
+		})
 		tenant, _ := newTenantWithAuthConfig(t, svc.Enclave, proto.AuthConfig{
 			Stytch: proto.AuthStytchConfig{
 				Enabled:   true,
@@ -442,14 +442,14 @@ func TestPlayFabAuth(t *testing.T) {
 			playfabAPI := httptest.NewServer(http.HandlerFunc(testCase.playfabHandler))
 			defer playfabAPI.Close()
 
-			svc := initRPCWithClient(t, &http.Client{Transport: testTransport{
+			svc := initRPCWithClient(t, testTransport{
 				RoundTripper: http.DefaultTransport,
 				modifyRequest: func(req *http.Request) {
 					if strings.Contains(req.URL.String(), "playfabapi.com") {
 						req.URL.Host = playfabAPI.Listener.Addr().String()
 					}
 				},
-			}})
+			})
 			tenant, _ := newTenantWithAuthConfig(t, svc.Enclave, proto.AuthConfig{
 				Playfab: proto.AuthPlayfabConfig{
 					Enabled: true,

rpc/helpers_test.go

@@ -51,21 +51,21 @@ func initRPC(t *testing.T, options ...func(*config.Config)) *rpc.RPC {
 		opt(cfg)
 	}
 
-	svc, err := rpc.New(cfg, &http.Client{Transport: &testTransport{RoundTripper: http.DefaultTransport}})
+	svc, err := rpc.New(cfg, &testTransport{RoundTripper: http.DefaultTransport})
 	if err != nil {
 		t.Fatal(err)
 	}
 	svc.Wallets = newWalletServiceMock(nil)
 	return svc
 }
 
-func initRPCWithClient(t *testing.T, client *http.Client, options ...func(*config.Config)) *rpc.RPC {
+func initRPCWithClient(t *testing.T, transport http.RoundTripper, options ...func(*config.Config)) *rpc.RPC {
 	cfg := initConfig(t, awsEndpoint)
 	for _, opt := range options {
 		opt(cfg)
 	}
 
-	svc, err := rpc.New(cfg, client)
+	svc, err := rpc.New(cfg, transport)
 	if err != nil {
 		t.Fatal(err)
 	}

rpc/rpc.go

@@ -75,9 +75,10 @@ type RPC struct {
 	running      int32
 }
 
-func New(cfg *config.Config, client *http.Client) (*RPC, error) {
-	if client == nil {
-		client = &http.Client{Timeout: 10 * time.Second}
+func New(cfg *config.Config, transport http.RoundTripper) (*RPC, error) {
+	client := &http.Client{
+		Timeout:   30 * time.Second,
+		Transport: transport,
 	}
 	wrappedClient := tracing.WrapClient(client)