@@ -31,15 +31,16 @@ import (
3131)
3232
3333const (
34- nmcName = "nmc"
35- nsFirst = "example-ns-1"
36- nsSecond = "example-ns-2"
37- nameFirst = "example-name-1"
38- nameSecond = "example-name-2"
39- imageFirst = "example-image-1"
40- imageSecond = "example-image-2"
41- kmodName = "kmm.node.kubernetes.io/test-ns.test-module.ready"
42- labelNotToRemove = "example.node.kubernetes.io/label-not-to-be-removed"
34+ nmcName = "nmc"
35+ nsFirst = "example-ns-1"
36+ nsSecond = "example-ns-2"
37+ nameFirst = "example-name-1"
38+ nameSecond = "example-name-2"
39+ imageFirst = "example-image-1"
40+ imageSecond = "example-image-2"
41+ kmodReadyLabel = "kmm.node.kubernetes.io/test-ns.test-module.ready"
42+ kmodVersionReadyLabel = "kmm.node.kubernetes.io/test-ns.test-module.version.ready"
43+ labelNotToRemove = "example.node.kubernetes.io/label-not-to-be-removed"
4344)
4445
4546var _ = Describe ("NodeModulesConfigReconciler_Reconcile" , func () {
@@ -138,7 +139,7 @@ var _ = Describe("NodeModulesConfigReconciler_Reconcile", func() {
138139 node := v1.Node {
139140 ObjectMeta : metav1.ObjectMeta {
140141 Labels : map [string ]string {
141- kmodName : "" ,
142+ kmodReadyLabel : "" ,
142143 labelNotToRemove : "" ,
143144 },
144145 },
@@ -165,9 +166,10 @@ var _ = Describe("NodeModulesConfigReconciler_Reconcile", func() {
165166 ),
166167 wh .EXPECT ().SyncStatus (ctx , nmc , & node ),
167168 nm .EXPECT ().IsNodeSchedulable (& node , nil ).Return (false ),
168- nm .EXPECT ().UpdateLabels (ctx , & node , nil , map [string ]string {kmodName : "" }).DoAndReturn (
169+ nm .EXPECT ().UpdateLabels (ctx , & node , nil , map [string ]string {kmodReadyLabel : "" , kmodVersionReadyLabel : "" }).DoAndReturn (
169170 func (_ context.Context , obj ctrlclient.Object , _ , _ map [string ]string ) error {
170- delete (node .ObjectMeta .Labels , kmodName )
171+ delete (node .ObjectMeta .Labels , kmodReadyLabel )
172+ delete (node .ObjectMeta .Labels , kmodVersionReadyLabel )
171173 return nil
172174 },
173175 ),
@@ -194,8 +196,9 @@ var _ = Describe("NodeModulesConfigReconciler_Reconcile", func() {
194196 node := v1.Node {
195197 ObjectMeta : metav1.ObjectMeta {
196198 Labels : map [string ]string {
197- kmodName : "" ,
198- labelNotToRemove : "" ,
199+ kmodReadyLabel : "" ,
200+ kmodVersionReadyLabel : "2" ,
201+ labelNotToRemove : "" ,
199202 },
200203 },
201204 }
@@ -221,7 +224,7 @@ var _ = Describe("NodeModulesConfigReconciler_Reconcile", func() {
221224 ),
222225 wh .EXPECT ().SyncStatus (ctx , nmc , & node ),
223226 nm .EXPECT ().IsNodeSchedulable (& node , nil ).Return (false ),
224- nm .EXPECT ().UpdateLabels (ctx , & node , nil , map [string ]string {kmodName : "" }).DoAndReturn (
227+ nm .EXPECT ().UpdateLabels (ctx , & node , nil , map [string ]string {kmodReadyLabel : "" , kmodVersionReadyLabel : "" }).DoAndReturn (
225228 func (_ context.Context , obj ctrlclient.Object , _ , _ map [string ]string ) error {
226229 return fmt .Errorf ("some error" )
227230 },
0 commit comments