sdk/go/arvadostest/api.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: Apache-2.0
   4
   5 package arvadostest
   6
   7 import (
   8         "context"
   9         "encoding/json"
  10         "errors"
  11         "net/url"
  12         "reflect"
  13         "runtime"
  14         "sync"
  15
  16         "git.curoverse.com/arvados.git/sdk/go/arvados"
  17 )
  18
  19 var ErrStubUnimplemented = errors.New("stub unimplemented")
  20
  21 type APIStub struct {
  22         // The error to return from every stubbed API method.
  23         Error error
  24         calls []APIStubCall
  25         mtx   sync.Mutex
  26 }
  27
  28 // BaseURL implements federation.backend
  29 func (as *APIStub) BaseURL() url.URL {
  30         return url.URL{Scheme: "https", Host: "apistub.example.com"}
  31 }
  32 func (as *APIStub) ConfigGet(ctx context.Context) (json.RawMessage, error) {
  33         as.appendCall(as.ConfigGet, ctx, nil)
  34         return nil, as.Error
  35 }
  36 func (as *APIStub) Login(ctx context.Context, options arvados.LoginOptions) (arvados.LoginResponse, error) {
  37         as.appendCall(as.Login, ctx, options)
  38         return arvados.LoginResponse{}, as.Error
  39 }
  40 func (as *APIStub) CollectionCreate(ctx context.Context, options arvados.CreateOptions) (arvados.Collection, error) {
  41         as.appendCall(as.CollectionCreate, ctx, options)
  42         return arvados.Collection{}, as.Error
  43 }
  44 func (as *APIStub) CollectionUpdate(ctx context.Context, options arvados.UpdateOptions) (arvados.Collection, error) {
  45         as.appendCall(as.CollectionUpdate, ctx, options)
  46         return arvados.Collection{}, as.Error
  47 }
  48 func (as *APIStub) CollectionGet(ctx context.Context, options arvados.GetOptions) (arvados.Collection, error) {
  49         as.appendCall(as.CollectionGet, ctx, options)
  50         return arvados.Collection{}, as.Error
  51 }
  52 func (as *APIStub) CollectionList(ctx context.Context, options arvados.ListOptions) (arvados.CollectionList, error) {
  53         as.appendCall(as.CollectionList, ctx, options)
  54         return arvados.CollectionList{}, as.Error
  55 }
  56 func (as *APIStub) CollectionProvenance(ctx context.Context, options arvados.GetOptions) (map[string]interface{}, error) {
  57         as.appendCall(as.CollectionProvenance, ctx, options)
  58         return nil, as.Error
  59 }
  60 func (as *APIStub) CollectionUsedBy(ctx context.Context, options arvados.GetOptions) (map[string]interface{}, error) {
  61         as.appendCall(as.CollectionUsedBy, ctx, options)
  62         return nil, as.Error
  63 }
  64 func (as *APIStub) CollectionDelete(ctx context.Context, options arvados.DeleteOptions) (arvados.Collection, error) {
  65         as.appendCall(as.CollectionDelete, ctx, options)
  66         return arvados.Collection{}, as.Error
  67 }
  68 func (as *APIStub) CollectionTrash(ctx context.Context, options arvados.DeleteOptions) (arvados.Collection, error) {
  69         as.appendCall(as.CollectionTrash, ctx, options)
  70         return arvados.Collection{}, as.Error
  71 }
  72 func (as *APIStub) CollectionUntrash(ctx context.Context, options arvados.UntrashOptions) (arvados.Collection, error) {
  73         as.appendCall(as.CollectionUntrash, ctx, options)
  74         return arvados.Collection{}, as.Error
  75 }
  76 func (as *APIStub) ContainerCreate(ctx context.Context, options arvados.CreateOptions) (arvados.Container, error) {
  77         as.appendCall(as.ContainerCreate, ctx, options)
  78         return arvados.Container{}, as.Error
  79 }
  80 func (as *APIStub) ContainerUpdate(ctx context.Context, options arvados.UpdateOptions) (arvados.Container, error) {
  81         as.appendCall(as.ContainerUpdate, ctx, options)
  82         return arvados.Container{}, as.Error
  83 }
  84 func (as *APIStub) ContainerGet(ctx context.Context, options arvados.GetOptions) (arvados.Container, error) {
  85         as.appendCall(as.ContainerGet, ctx, options)
  86         return arvados.Container{}, as.Error
  87 }
  88 func (as *APIStub) ContainerList(ctx context.Context, options arvados.ListOptions) (arvados.ContainerList, error) {
  89         as.appendCall(as.ContainerList, ctx, options)
  90         return arvados.ContainerList{}, as.Error
  91 }
  92 func (as *APIStub) ContainerDelete(ctx context.Context, options arvados.DeleteOptions) (arvados.Container, error) {
  93         as.appendCall(as.ContainerDelete, ctx, options)
  94         return arvados.Container{}, as.Error
  95 }
  96 func (as *APIStub) ContainerLock(ctx context.Context, options arvados.GetOptions) (arvados.Container, error) {
  97         as.appendCall(as.ContainerLock, ctx, options)
  98         return arvados.Container{}, as.Error
  99 }
 100 func (as *APIStub) ContainerUnlock(ctx context.Context, options arvados.GetOptions) (arvados.Container, error) {
 101         as.appendCall(as.ContainerUnlock, ctx, options)
 102         return arvados.Container{}, as.Error
 103 }
 104 func (as *APIStub) SpecimenCreate(ctx context.Context, options arvados.CreateOptions) (arvados.Specimen, error) {
 105         as.appendCall(as.SpecimenCreate, ctx, options)
 106         return arvados.Specimen{}, as.Error
 107 }
 108 func (as *APIStub) SpecimenUpdate(ctx context.Context, options arvados.UpdateOptions) (arvados.Specimen, error) {
 109         as.appendCall(as.SpecimenUpdate, ctx, options)
 110         return arvados.Specimen{}, as.Error
 111 }
 112 func (as *APIStub) SpecimenGet(ctx context.Context, options arvados.GetOptions) (arvados.Specimen, error) {
 113         as.appendCall(as.SpecimenGet, ctx, options)
 114         return arvados.Specimen{}, as.Error
 115 }
 116 func (as *APIStub) SpecimenList(ctx context.Context, options arvados.ListOptions) (arvados.SpecimenList, error) {
 117         as.appendCall(as.SpecimenList, ctx, options)
 118         return arvados.SpecimenList{}, as.Error
 119 }
 120 func (as *APIStub) SpecimenDelete(ctx context.Context, options arvados.DeleteOptions) (arvados.Specimen, error) {
 121         as.appendCall(as.SpecimenDelete, ctx, options)
 122         return arvados.Specimen{}, as.Error
 123 }
 124 func (as *APIStub) UserCreate(ctx context.Context, options arvados.CreateOptions) (arvados.User, error) {
 125         as.appendCall(as.UserCreate, ctx, options)
 126         return arvados.User{}, as.Error
 127 }
 128 func (as *APIStub) UserUpdate(ctx context.Context, options arvados.UpdateOptions) (arvados.User, error) {
 129         as.appendCall(as.UserUpdate, ctx, options)
 130         return arvados.User{}, as.Error
 131 }
 132 func (as *APIStub) UserUpdateUUID(ctx context.Context, options arvados.UpdateUUIDOptions) (arvados.User, error) {
 133         as.appendCall(as.UserUpdateUUID, ctx, options)
 134         return arvados.User{}, as.Error
 135 }
 136 func (as *APIStub) UserActivate(ctx context.Context, options arvados.UserActivateOptions) (arvados.User, error) {
 137         as.appendCall(as.UserActivate, ctx, options)
 138         return arvados.User{}, as.Error
 139 }
 140 func (as *APIStub) UserSetup(ctx context.Context, options arvados.UserSetupOptions) (map[string]interface{}, error) {
 141         as.appendCall(as.UserSetup, ctx, options)
 142         return nil, as.Error
 143 }
 144 func (as *APIStub) UserUnsetup(ctx context.Context, options arvados.GetOptions) (arvados.User, error) {
 145         as.appendCall(as.UserUnsetup, ctx, options)
 146         return arvados.User{}, as.Error
 147 }
 148 func (as *APIStub) UserGet(ctx context.Context, options arvados.GetOptions) (arvados.User, error) {
 149         as.appendCall(as.UserGet, ctx, options)
 150         return arvados.User{}, as.Error
 151 }
 152 func (as *APIStub) UserGetCurrent(ctx context.Context, options arvados.GetOptions) (arvados.User, error) {
 153         as.appendCall(as.UserGetCurrent, ctx, options)
 154         return arvados.User{}, as.Error
 155 }
 156 func (as *APIStub) UserGetSystem(ctx context.Context, options arvados.GetOptions) (arvados.User, error) {
 157         as.appendCall(as.UserGetSystem, ctx, options)
 158         return arvados.User{}, as.Error
 159 }
 160 func (as *APIStub) UserList(ctx context.Context, options arvados.ListOptions) (arvados.UserList, error) {
 161         as.appendCall(as.UserList, ctx, options)
 162         return arvados.UserList{}, as.Error
 163 }
 164 func (as *APIStub) UserDelete(ctx context.Context, options arvados.DeleteOptions) (arvados.User, error) {
 165         as.appendCall(as.UserDelete, ctx, options)
 166         return arvados.User{}, as.Error
 167 }
 168 func (as *APIStub) UserMerge(ctx context.Context, options arvados.UserMergeOptions) (arvados.User, error) {
 169         as.appendCall(as.UserMerge, ctx, options)
 170         return arvados.User{}, as.Error
 171 }
 172 func (as *APIStub) APIClientAuthorizationCurrent(ctx context.Context, options arvados.GetOptions) (arvados.APIClientAuthorization, error) {
 173         as.appendCall(as.APIClientAuthorizationCurrent, ctx, options)
 174         return arvados.APIClientAuthorization{}, as.Error
 175 }
 176
 177 func (as *APIStub) appendCall(method interface{}, ctx context.Context, options interface{}) {
 178         as.mtx.Lock()
 179         defer as.mtx.Unlock()
 180         as.calls = append(as.calls, APIStubCall{method, ctx, options})
 181 }
 182
 183 func (as *APIStub) Calls(method interface{}) []APIStubCall {
 184         as.mtx.Lock()
 185         defer as.mtx.Unlock()
 186         var calls []APIStubCall
 187         for _, call := range as.calls {
 188
 189                 if method == nil || (runtime.FuncForPC(reflect.ValueOf(call.Method).Pointer()).Name() ==
 190                         runtime.FuncForPC(reflect.ValueOf(method).Pointer()).Name()) {
 191                         calls = append(calls, call)
 192                 }
 193         }
 194         return calls
 195 }
 196
 197 type APIStubCall struct {
 198         Method  interface{}
 199         Context context.Context
 200         Options interface{}
 201 }