Fix csi_client_test.go metrics nil pointer dereference

This commit is contained in:
Mengjiao Liu 2021-04-09 10:15:56 +08:00
parent a651804427
commit 8725c96014

View File

@ -750,13 +750,18 @@ func TestVolumeHealthEnable(t *testing.T) {
assert.Nil(t, err) assert.Nil(t, err)
} }
if tc.volumeConditionSet { if metrics == nil {
assert.NotNil(t, metrics.Abnormal) t.Errorf("csi.NodeGetVolumeStats returned nil metrics for volume %s", tc.volumeData.VolumeID)
assert.NotNil(t, metrics.Message)
} else { } else {
assert.Nil(t, metrics.Abnormal) if tc.volumeConditionSet {
assert.Nil(t, metrics.Message) assert.NotNil(t, metrics.Abnormal)
assert.NotNil(t, metrics.Message)
} else {
assert.Nil(t, metrics.Abnormal)
assert.Nil(t, metrics.Message)
}
} }
}) })
} }
} }
@ -792,8 +797,12 @@ func TestVolumeHealthDisable(t *testing.T) {
assert.Nil(t, err) assert.Nil(t, err)
} }
assert.Nil(t, metrics.Abnormal) if metrics == nil {
assert.Nil(t, metrics.Message) t.Errorf("csi.NodeGetVolumeStats returned nil metrics for volume %s", tc.volumeData.VolumeID)
} else {
assert.Nil(t, metrics.Abnormal)
assert.Nil(t, metrics.Message)
}
}) })
} }
} }