Skip to content

Commit 3203d8c

Browse files
committed
refactor for api changes
Signed-off-by: Per G. da Silva <[email protected]>
1 parent 5b339bc commit 3203d8c

File tree

2 files changed

+25
-7
lines changed

2 files changed

+25
-7
lines changed

pkg/controller/registry/reconciler/reconciler.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,8 +181,8 @@ func Pod(source *v1alpha1.CatalogSource, name string, image string, saName strin
181181
}
182182

183183
// Override priority class name
184-
if grpcPodConfig.PriorityClassName != "" {
185-
pod.Spec.PriorityClassName = grpcPodConfig.PriorityClassName
184+
if grpcPodConfig.PriorityClassName != nil {
185+
pod.Spec.PriorityClassName = *grpcPodConfig.PriorityClassName
186186
}
187187

188188
// Override tolerations

pkg/controller/registry/reconciler/reconciler_test.go

Lines changed: 23 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ func TestPodSchedulingOverrides(t *testing.T) {
132132
expectedPriorityClassName string
133133
}{
134134
{
135-
title: "No overrides",
135+
title: "no overrides",
136136
catalogSource: &v1alpha1.CatalogSource{
137137
ObjectMeta: metav1.ObjectMeta{
138138
Name: "test",
@@ -147,7 +147,7 @@ func TestPodSchedulingOverrides(t *testing.T) {
147147
expectedPriorityClassName: defaultPriorityClassName,
148148
expectedNodeSelectors: defaultNodeSelectors,
149149
}, {
150-
title: "Override node selectors",
150+
title: "override node selectors",
151151
catalogSource: &v1alpha1.CatalogSource{
152152
ObjectMeta: metav1.ObjectMeta{
153153
Name: "test",
@@ -165,7 +165,7 @@ func TestPodSchedulingOverrides(t *testing.T) {
165165
expectedPriorityClassName: defaultPriorityClassName,
166166
expectedNodeSelectors: overriddenNodeSelectors,
167167
}, {
168-
title: "Override priority class name",
168+
title: "override priority class name",
169169
catalogSource: &v1alpha1.CatalogSource{
170170
ObjectMeta: metav1.ObjectMeta{
171171
Name: "test",
@@ -175,13 +175,31 @@ func TestPodSchedulingOverrides(t *testing.T) {
175175
SourceType: v1alpha1.SourceTypeGrpc,
176176
Image: "repo/image:tag",
177177
GrpcPodConfig: &v1alpha1.GrpcPodConfig{
178-
PriorityClassName: overriddenPriorityClassName,
178+
PriorityClassName: &overriddenPriorityClassName,
179179
},
180180
},
181181
},
182182
expectedTolerations: nil,
183183
expectedPriorityClassName: overriddenPriorityClassName,
184184
expectedNodeSelectors: defaultNodeSelectors,
185+
}, {
186+
title: "doesn't override priority class name when its nil",
187+
catalogSource: &v1alpha1.CatalogSource{
188+
ObjectMeta: metav1.ObjectMeta{
189+
Name: "test",
190+
Namespace: "testns",
191+
},
192+
Spec: v1alpha1.CatalogSourceSpec{
193+
SourceType: v1alpha1.SourceTypeGrpc,
194+
Image: "repo/image:tag",
195+
GrpcPodConfig: &v1alpha1.GrpcPodConfig{
196+
PriorityClassName: nil,
197+
},
198+
},
199+
},
200+
expectedTolerations: nil,
201+
expectedPriorityClassName: defaultPriorityClassName,
202+
expectedNodeSelectors: defaultNodeSelectors,
185203
}, {
186204
title: "Override node tolerations",
187205
catalogSource: &v1alpha1.CatalogSource{
@@ -212,7 +230,7 @@ func TestPodSchedulingOverrides(t *testing.T) {
212230
Image: "repo/image:tag",
213231
GrpcPodConfig: &v1alpha1.GrpcPodConfig{
214232
NodeSelector: overriddenNodeSelectors,
215-
PriorityClassName: overriddenPriorityClassName,
233+
PriorityClassName: &overriddenPriorityClassName,
216234
Tolerations: overriddenTolerations,
217235
},
218236
},

0 commit comments

Comments
 (0)