/
service_fake.go
73 lines (56 loc) · 1.52 KB
/
service_fake.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package cluster
import (
"context"
"encoding/json"
"github.com/giantswarm/microerror"
"github.com/giantswarm/micrologger/microloggertest"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/types"
runtimeclient "sigs.k8s.io/controller-runtime/pkg/client"
"github.com/giantswarm/kubectl-gs/pkg/data/client"
)
var _ Interface = &FakeService{}
type FakeService struct {
service *Service
storage []runtime.Object
}
func NewFakeService(storage []runtime.Object) *FakeService {
clientConfig := client.Config{
Logger: microloggertest.New(),
}
fakeClient, _ := client.NewFakeClient(clientConfig)
underlyingService := &Service{
client: fakeClient,
}
ms := &FakeService{
service: underlyingService,
storage: storage,
}
return ms
}
func (ms *FakeService) Get(ctx context.Context, options GetOptions) (Resource, error) {
var err error
for _, res := range ms.storage {
err = ms.service.client.K8sClient.CtrlClient().Create(ctx, res)
if err != nil {
return nil, microerror.Mask(err)
}
}
result, err := ms.service.Get(ctx, options)
if err != nil {
return nil, microerror.Mask(err)
}
return result, nil
}
func (ms *FakeService) Patch(ctx context.Context, object runtime.Object, options PatchOptions) error {
var err error
bytes, err := json.Marshal(options.PatchSpecs)
if err != nil {
return microerror.Mask(err)
}
err = ms.service.client.K8sClient.CtrlClient().Patch(ctx, object, runtimeclient.RawPatch(types.JSONPatchType, bytes))
if err != nil {
return microerror.Mask(err)
}
return nil
}