X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/8b43f32b2c11d45f951bf4ff1bffab03d391ff41..4e1e7f762ff1acd13b18efed5974b32833a467e2:/sdk/go/arvadostest/proxy.go

diff --git a/sdk/go/arvadostest/proxy.go b/sdk/go/arvadostest/proxy.go
index 015061ad50..9940ddd3d9 100644
--- a/sdk/go/arvadostest/proxy.go
+++ b/sdk/go/arvadostest/proxy.go
@@ -13,7 +13,7 @@ import (
 	"net/url"
 	"time"
 
-	"git.curoverse.com/arvados.git/sdk/go/arvados"
+	"git.arvados.org/arvados.git/sdk/go/arvados"
 	"gopkg.in/check.v1"
 )
 
@@ -26,6 +26,10 @@ type Proxy struct {
 
 	// A dump of each request that has been proxied.
 	RequestDumps [][]byte
+
+	// If non-nil, func will be called on each incoming request
+	// before proxying it.
+	Director func(*http.Request)
 }
 
 // NewProxy returns a new Proxy that saves a dump of each reqeust
@@ -63,6 +67,9 @@ func NewProxy(c *check.C, svc arvados.Service) *Proxy {
 		URL:    u,
 	}
 	rp.Director = func(r *http.Request) {
+		if proxy.Director != nil {
+			proxy.Director(r)
+		}
 		dump, _ := httputil.DumpRequest(r, true)
 		proxy.RequestDumps = append(proxy.RequestDumps, dump)
 		r.URL.Scheme = target.Scheme