Merge branch 'master' into 15577-ownership-transfer
[arvados.git] / 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) APIClientAuthorizationCurrent(ctx context.Context, options arvados.GetOptions) (arvados.APIClientAuthorization, error) {
125         as.appendCall(as.APIClientAuthorizationCurrent, ctx, options)
126         return arvados.APIClientAuthorization{}, as.Error
127 }
128
129 func (as *APIStub) appendCall(method interface{}, ctx context.Context, options interface{}) {
130         as.mtx.Lock()
131         defer as.mtx.Unlock()
132         as.calls = append(as.calls, APIStubCall{method, ctx, options})
133 }
134
135 func (as *APIStub) Calls(method interface{}) []APIStubCall {
136         as.mtx.Lock()
137         defer as.mtx.Unlock()
138         var calls []APIStubCall
139         for _, call := range as.calls {
140
141                 if method == nil || (runtime.FuncForPC(reflect.ValueOf(call.Method).Pointer()).Name() ==
142                         runtime.FuncForPC(reflect.ValueOf(method).Pointer()).Name()) {
143                         calls = append(calls, call)
144                 }
145         }
146         return calls
147 }
148
149 type APIStubCall struct {
150         Method  interface{}
151         Context context.Context
152         Options interface{}
153 }