Fix check for nil or non-nil error

dell · Dec 1, 2024 · 905b90f · 905b90f
1 parent 22b5607
commit 905b90f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/service/service_unit_test.go b/service/service_unit_test.go
@@ -575,7 +575,11 @@ func TestGetZoneKeyLabelFromSecret(t *testing.T) {
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			label, err := getZoneKeyLabelFromSecret(tt.arrays)
-			assert.Equal(t, err, tt.expectedErr)
+			if tt.expectedErr == nil {
+				assert.Nil(t, err)
+			} else {
+				assert.NotNil(t, err)
+			}
 			assert.Equal(t, label, tt.expectedLabel)
 		})
 	}