From 9d5d1ab3e91f2497afdd2bbb714090882d58f679 Mon Sep 17 00:00:00 2001
From: Deseao <alexhaynes93@gmail.com>
Date: Thu, 10 May 2018 12:07:57 -0500
Subject: [PATCH] feat(publisher): add SetClient func

Add SetClient func to allow dsl users to control aspects of requests to the
Broker API, such as timeouts, headers, and Transport.
---
 dsl/publish.go      | 12 ++++++++++--
 dsl/publish_test.go |  9 +++++++++
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/dsl/publish.go b/dsl/publish.go
index 7385ff643..5c994501e 100644
--- a/dsl/publish.go
+++ b/dsl/publish.go
@@ -32,6 +32,7 @@ type PactName struct {
 // Publisher is the API to send Pact files to a Pact Broker.
 type Publisher struct {
 	request types.PublishRequest
+	client  *http.Client
 }
 
 // validate the publish requests.
@@ -71,7 +72,9 @@ func (p *Publisher) validate() error {
 
 // call sends a message to the Pact Broker.
 func (p *Publisher) call(method string, url string, content []byte) error {
-	client := &http.Client{}
+	if p.client == nil {
+		p.client = &http.Client{}
+	}
 	var req *http.Request
 	var err error
 	req, err = http.NewRequest(method, url, bytes.NewReader(content))
@@ -85,7 +88,7 @@ func (p *Publisher) call(method string, url string, content []byte) error {
 		req.SetBasicAuth(p.request.BrokerUsername, p.request.BrokerPassword)
 	}
 
-	res, err := client.Do(req)
+	res, err := p.client.Do(req)
 	if err != nil {
 		return err
 	}
@@ -191,3 +194,8 @@ func (p *Publisher) tagRequest(consumerName string, request types.PublishRequest
 
 	return nil
 }
+
+// SetClient allows dsl users to configure the http.Client used when publishing Pacts
+func (p *Publisher) SetClient(client *http.Client) {
+	p.client = client
+}
diff --git a/dsl/publish_test.go b/dsl/publish_test.go
index 836b7b212..aff061263 100644
--- a/dsl/publish_test.go
+++ b/dsl/publish_test.go
@@ -513,3 +513,12 @@ func TestPublish_tagRequestFail(t *testing.T) {
 		t.Fatalf("Expected error but got none")
 	}
 }
+
+func TestPublish_SetClient(t *testing.T) {
+	p := &Publisher{}
+	client := &http.Client{}
+	p.SetClient(client)
+	if p.client != client {
+		t.Fatalf("SetClient Failed To Set Client On Publisher")
+	}
+}