Merge pull request #234 from chrismoos/proxy_option

Add option to override HTTP Proxy on Attacker.

lib/attack.go

@@ -6,6 +6,7 @@ import (
 	"io/ioutil"
 	"net"
 	"net/http"
+	"net/url"
 	"sync"
 	"time"
 
@@ -105,6 +106,15 @@ func Redirects(n int) func(*Attacker) {
 	}
 }
 
+// Proxy returns a functional option which sets the `Proxy` field on
+// the http.Client's Transport
+func Proxy(proxy func(*http.Request) (*url.URL, error)) func(*Attacker) {
+	return func(a *Attacker) {
+		tr := a.client.Transport.(*http.Transport)
+		tr.Proxy = proxy
+	}
+}
+
 // Timeout returns a functional option which sets the maximum amount of time
 // an Attacker will wait for a request to be responded to.
 func Timeout(d time.Duration) func(*Attacker) {