Skip to content

Commit

Permalink
fix unit test
Browse files Browse the repository at this point in the history
  • Loading branch information
petersutter committed Mar 12, 2024
1 parent 4399a8b commit b2b7e6e
Showing 1 changed file with 30 additions and 32 deletions.
62 changes: 30 additions & 32 deletions internal/fake/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import (
"reflect"

gardencore "github.com/gardener/gardener/pkg/apis/core"
gardencorev1beta1 "github.com/gardener/gardener/pkg/apis/core/v1beta1"
"k8s.io/apimachinery/pkg/api/meta"
"k8s.io/apimachinery/pkg/fields"
"k8s.io/apimachinery/pkg/runtime"
Expand Down Expand Up @@ -55,38 +56,35 @@ func (w *clientWrapper) List(ctx context.Context, list client.ObjectList, opts .
return nil
}

apiVersion, kind := list.GetObjectKind().GroupVersionKind().ToAPIVersionAndKind()
if apiVersion == "core.gardener.cloud/v1beta1" {
switch kind {
case "ShootList":
filterItems(list, fieldSelector, func(vItem reflect.Value) fields.Set {
fieldSet := fields.Set{}
vName := vItem.FieldByName("Name")
fieldSet["metadata.name"] = vName.String()
vSpec := vItem.FieldByName("Spec")
vSeedName := vSpec.FieldByName("SeedName")

if !vSeedName.IsNil() {
fieldSet[gardencore.ShootSeedName] = reflect.Indirect(vSeedName).String()
}

return fieldSet
})
case "ProjectList":
filterItems(list, fieldSelector, func(vItem reflect.Value) fields.Set {
fieldSet := fields.Set{}
vName := vItem.FieldByName("Name")
fieldSet["metadata.name"] = vName.String()
vSpec := vItem.FieldByName("Spec")
vNamespace := vSpec.FieldByName("Namespace")

if !vNamespace.IsNil() {
fieldSet[gardencore.ProjectNamespace] = reflect.Indirect(vNamespace).String()
}

return fieldSet
})
}
switch list := list.(type) {
case *gardencorev1beta1.ShootList:
filterItems(list, fieldSelector, func(vItem reflect.Value) fields.Set {
fieldSet := fields.Set{}
vName := vItem.FieldByName("Name")
fieldSet["metadata.name"] = vName.String()
vSpec := vItem.FieldByName("Spec")
vSeedName := vSpec.FieldByName("SeedName")

if !vSeedName.IsNil() {
fieldSet[gardencore.ShootSeedName] = reflect.Indirect(vSeedName).String()
}

return fieldSet
})
case *gardencorev1beta1.ProjectList:
filterItems(list, fieldSelector, func(vItem reflect.Value) fields.Set {
fieldSet := fields.Set{}
vName := vItem.FieldByName("Name")
fieldSet["metadata.name"] = vName.String()
vSpec := vItem.FieldByName("Spec")
vNamespace := vSpec.FieldByName("Namespace")

if !vNamespace.IsNil() {
fieldSet[gardencore.ProjectNamespace] = reflect.Indirect(vNamespace).String()
}

return fieldSet
})
}

return nil
Expand Down

0 comments on commit b2b7e6e

Please sign in to comment.