test fixes and some refactoring

Signed-off-by: Viktor Kramarenko <[email protected]>
deckhouse · Aug 26, 2024 · 9e2ef85 · 9e2ef85
1 parent c648ce9
commit 9e2ef85
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/images/agent/src/pkg/controller/block_device_test.go b/images/agent/src/pkg/controller/block_device_test.go
@@ -340,6 +340,10 @@ func TestBlockDeviceCtrl(t *testing.T) {
 				MachineID:             "testMACHINE",
 			},
 		}
+		blockDevice.Labels = map[string]string{
+			"some-custom-label1": "v",
+			"some-custom-label2": "v",
+		}
 
 		expectedLabels := map[string]string{
 			"kubernetes.io/metadata.name":                     blockDevice.ObjectMeta.Name,
@@ -358,6 +362,8 @@ func TestBlockDeviceCtrl(t *testing.T) {
 			BlockDeviceLabelPrefix + "/rota":                  strconv.FormatBool(blockDevice.Status.Rota),
 			BlockDeviceLabelPrefix + "/hotplug":               strconv.FormatBool(blockDevice.Status.HotPlug),
 			BlockDeviceLabelPrefix + "/machineid":             blockDevice.Status.MachineID,
+			"some-custom-label1":                              "v",
+			"some-custom-label2":                              "v",
 		}
 
 		assert.Equal(t, expectedLabels, ConfigureBlockDeviceLabels(blockDevice))