13647: Set X-External-Client header in arvbox.

[arvados.git] / sdk / go / arvadostest / api.go

diff --git a/sdk/go/arvadostest/api.go b/sdk/go/arvadostest/api.go
index a3cacf3f6c5a12911dc9107d25b2a6cb6edc1be1..850bd0639dcaa856c5b6dfa69b09309a1ead1de5 100644 (file)
--- a/sdk/go/arvadostest/api.go
+++ b/sdk/go/arvadostest/api.go
@@ -6,7 +6,10 @@ package arvadostest
 
 import (
        "context"
+       "encoding/json"
        "errors"
+       "reflect"
+       "runtime"
        "sync"
 
        "git.curoverse.com/arvados.git/sdk/go/arvados"
@@ -21,6 +24,10 @@ type APIStub struct {
        mtx   sync.Mutex
 }
 
+func (as *APIStub) ConfigGet(ctx context.Context) (json.RawMessage, error) {
+       as.appendCall(as.ConfigGet, ctx, nil)
+       return nil, as.Error
+}
 func (as *APIStub) CollectionCreate(ctx context.Context, options arvados.CreateOptions) (arvados.Collection, error) {
        as.appendCall(as.CollectionCreate, ctx, options)
        return arvados.Collection{}, as.Error
@@ -121,7 +128,9 @@ func (as *APIStub) Calls(method interface{}) []APIStubCall {
        defer as.mtx.Unlock()
        var calls []APIStubCall
        for _, call := range as.calls {
-               if method == nil || call.Method == method {
+
+               if method == nil || (runtime.FuncForPC(reflect.ValueOf(call.Method).Pointer()).Name() ==
+                       runtime.FuncForPC(reflect.ValueOf(method).Pointer()).Name()) {
                        calls = append(calls, call)
                }
        }