X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/15c688d1c4b41232536d0a275dd5c0fdb8879a00..e838828374ceed5ef6da260939251e86f72b6f27:/sdk/go/arvadostest/stub.go

diff --git a/sdk/go/arvadostest/stub.go b/sdk/go/arvadostest/stub.go
index 2b6e40c950..80735f86eb 100644
--- a/sdk/go/arvadostest/stub.go
+++ b/sdk/go/arvadostest/stub.go
@@ -42,14 +42,20 @@ func (stub *ServerStub) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
 }
 
 // SetServiceURL overrides the given service config/discovery with the
-// given internalURL.
+// given internalURLs.
+//
+// ExternalURL is set to the last internalURL, which only aims to
+// address the case where there is only one.
 //
 // SetServiceURL panics on errors.
-func SetServiceURL(service *arvados.Service, internalURL string) {
-	u, err := url.Parse(internalURL)
-	if err != nil {
-		panic(err)
+func SetServiceURL(service *arvados.Service, internalURLs ...string) {
+	service.InternalURLs = map[arvados.URL]arvados.ServiceInstance{}
+	for _, u := range internalURLs {
+		u, err := url.Parse(u)
+		if err != nil {
+			panic(err)
+		}
+		service.InternalURLs[arvados.URL(*u)] = arvados.ServiceInstance{}
+		service.ExternalURL = arvados.URL(*u)
 	}
-	service.InternalURLs = map[arvados.URL]arvados.ServiceInstance{arvados.URL(*u): {}}
-	service.ExternalURL = arvados.URL(*u)
 }