reflect code review
This commit is contained in:
parent
6a65d121de
commit
d5ba24767f
|
|
@ -28,9 +28,7 @@ const (
|
||||||
)
|
)
|
||||||
|
|
||||||
var logger = logrus.New().WithField("test", "cluster")
|
var logger = logrus.New().WithField("test", "cluster")
|
||||||
|
var eventRecorder = record.NewFakeRecorder(1)
|
||||||
// bufferSize might have to be increased when unit test cover functions emitting events
|
|
||||||
var eventRecorder = record.NewFakeRecorder(5)
|
|
||||||
|
|
||||||
var cl = New(
|
var cl = New(
|
||||||
Config{
|
Config{
|
||||||
|
|
|
||||||
|
|
@ -228,19 +228,15 @@ func (c *Cluster) generateResourceRequirements(
|
||||||
defaultResources acidv1.Resources,
|
defaultResources acidv1.Resources,
|
||||||
containerName string) (*v1.ResourceRequirements, error) {
|
containerName string) (*v1.ResourceRequirements, error) {
|
||||||
var err error
|
var err error
|
||||||
|
specRequests := acidv1.ResourceDescription{}
|
||||||
|
specLimits := acidv1.ResourceDescription{}
|
||||||
|
result := v1.ResourceRequirements{}
|
||||||
|
|
||||||
var specRequests, specLimits acidv1.ResourceDescription
|
if resources != nil {
|
||||||
|
|
||||||
if resources == nil {
|
|
||||||
specRequests = acidv1.ResourceDescription{}
|
|
||||||
specLimits = acidv1.ResourceDescription{}
|
|
||||||
} else {
|
|
||||||
specRequests = resources.ResourceRequests
|
specRequests = resources.ResourceRequests
|
||||||
specLimits = resources.ResourceLimits
|
specLimits = resources.ResourceLimits
|
||||||
}
|
}
|
||||||
|
|
||||||
result := v1.ResourceRequirements{}
|
|
||||||
|
|
||||||
result.Requests, err = fillResourceList(specRequests, defaultResources.ResourceRequests)
|
result.Requests, err = fillResourceList(specRequests, defaultResources.ResourceRequests)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("could not fill resource requests: %v", err)
|
return nil, fmt.Errorf("could not fill resource requests: %v", err)
|
||||||
|
|
|
||||||
|
|
@ -30,6 +30,7 @@ import (
|
||||||
"k8s.io/apimachinery/pkg/util/intstr"
|
"k8s.io/apimachinery/pkg/util/intstr"
|
||||||
"k8s.io/client-go/kubernetes/fake"
|
"k8s.io/client-go/kubernetes/fake"
|
||||||
v1core "k8s.io/client-go/kubernetes/typed/core/v1"
|
v1core "k8s.io/client-go/kubernetes/typed/core/v1"
|
||||||
|
"k8s.io/client-go/tools/record"
|
||||||
)
|
)
|
||||||
|
|
||||||
func newFakeK8sTestClient() (k8sutil.KubernetesClient, *fake.Clientset) {
|
func newFakeK8sTestClient() (k8sutil.KubernetesClient, *fake.Clientset) {
|
||||||
|
|
@ -1676,6 +1677,10 @@ func TestGenerateResourceRequirements(t *testing.T) {
|
||||||
roleLabel := "spilo-role"
|
roleLabel := "spilo-role"
|
||||||
sidecarName := "postgres-exporter"
|
sidecarName := "postgres-exporter"
|
||||||
|
|
||||||
|
// two test cases will call enforceMinResourceLimits which emits 2 events per call
|
||||||
|
// hence bufferSize of 4 is required
|
||||||
|
newEventRecorder := record.NewFakeRecorder(4)
|
||||||
|
|
||||||
configResources := config.Resources{
|
configResources := config.Resources{
|
||||||
ClusterLabels: map[string]string{"application": "spilo"},
|
ClusterLabels: map[string]string{"application": "spilo"},
|
||||||
ClusterNameLabel: clusterNameLabel,
|
ClusterNameLabel: clusterNameLabel,
|
||||||
|
|
@ -1954,7 +1959,7 @@ func TestGenerateResourceRequirements(t *testing.T) {
|
||||||
var cluster = New(
|
var cluster = New(
|
||||||
Config{
|
Config{
|
||||||
OpConfig: tt.config,
|
OpConfig: tt.config,
|
||||||
}, client, tt.pgSpec, logger, eventRecorder)
|
}, client, tt.pgSpec, logger, newEventRecorder)
|
||||||
|
|
||||||
cluster.Name = clusterName
|
cluster.Name = clusterName
|
||||||
cluster.Namespace = namespace
|
cluster.Namespace = namespace
|
||||||
|
|
|
||||||
|
|
@ -597,9 +597,12 @@ func trimCronjobName(name string) string {
|
||||||
|
|
||||||
func parseResourceRequirements(resourcesRequirement v1.ResourceRequirements) (acidv1.Resources, error) {
|
func parseResourceRequirements(resourcesRequirement v1.ResourceRequirements) (acidv1.Resources, error) {
|
||||||
var resources acidv1.Resources
|
var resources acidv1.Resources
|
||||||
resourcesJSON, _ := json.Marshal(resourcesRequirement)
|
resourcesJSON, err := json.Marshal(resourcesRequirement)
|
||||||
if err := json.Unmarshal(resourcesJSON, &resources); err != nil {
|
if err != nil {
|
||||||
return acidv1.Resources{}, fmt.Errorf("could not convert K8s resources requirements into acidv1.Resources struct")
|
return acidv1.Resources{}, fmt.Errorf("could not marshal K8s resources requirements")
|
||||||
|
}
|
||||||
|
if err = json.Unmarshal(resourcesJSON, &resources); err != nil {
|
||||||
|
return acidv1.Resources{}, fmt.Errorf("could not unmarshal K8s resources requirements into acidv1.Resources struct")
|
||||||
}
|
}
|
||||||
return resources, nil
|
return resources, nil
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue