github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-06 18:54:06 +00:00
Code Issues Projects Releases Wiki Activity

Expose WithError in FakeResourceFinder for tests

Browse Source
This commit is contained in:
Maciej Szulik 2024-07-08 13:18:37 +02:00
parent 7ec344dcef
commit f35d8a4595
No known key found for this signature in database
GPG Key ID: F15E55D276FA84C4
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
staging/src/k8s.io/cli-runtime/pkg/genericclioptions/builder_flags_fake.go
View File

@ -21,29 +21,40 @@ import (
) )
// NewSimpleFakeResourceFinder builds a super simple ResourceFinder that just iterates over the objects you provided // NewSimpleFakeResourceFinder builds a super simple ResourceFinder that just iterates over the objects you provided
func NewSimpleFakeResourceFinder(infos ...*resource.Info) ResourceFinder { func NewSimpleFakeResourceFinder(infos ...*resource.Info) *FakeResourceFinder {
return &fakeResourceFinder{ return &FakeResourceFinder{
Infos: infos, Infos: infos,
} }
} }
type fakeResourceFinder struct { func (f *FakeResourceFinder) WithError(err error) *FakeResourceFinder {
f.err = err
return f
}
type FakeResourceFinder struct {
Infos []*resource.Info Infos []*resource.Info
err error
} }
// Do implements the interface // Do implements the interface
func (f *fakeResourceFinder) Do() resource.Visitor { func (f *FakeResourceFinder) Do() resource.Visitor {
return &fakeResourceResult{ return &fakeResourceResult{
Infos: f.Infos, Infos: f.Infos,
err: f.err,
} }
} }
type fakeResourceResult struct { type fakeResourceResult struct {
Infos []*resource.Info Infos []*resource.Info
err error
} }
// Visit just iterates over info // Visit just iterates over info
func (r *fakeResourceResult) Visit(fn resource.VisitorFunc) error { func (r *fakeResourceResult) Visit(fn resource.VisitorFunc) error {
if r.err != nil {
return r.err
}
for _, info := range r.Infos { for _, info := range r.Infos {
err := fn(info, nil) err := fn(info, nil)
if err != nil { if err != nil {