From d871f0e6ee706ff4dd35b8ec2cf8dc5bfe5227ad Mon Sep 17 00:00:00 2001 From: f10atin9 Date: Wed, 9 Mar 2022 14:59:55 +0800 Subject: [PATCH] Test: update ListVolumeSnapshotClassTest Signed-off-by: f10atin9 --- .../volumesnapshotclass_test.go | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/pkg/models/resources/v1alpha3/volumesnapshotclass/volumesnapshotclass_test.go b/pkg/models/resources/v1alpha3/volumesnapshotclass/volumesnapshotclass_test.go index 26383a113..fe66492ab 100644 --- a/pkg/models/resources/v1alpha3/volumesnapshotclass/volumesnapshotclass_test.go +++ b/pkg/models/resources/v1alpha3/volumesnapshotclass/volumesnapshotclass_test.go @@ -72,6 +72,18 @@ func TestListVolumeSnapshotClass(t *testing.T) { snapshotClass3.CreationTimestamp = v1.NewTime(snapshotClass1.CreationTimestamp.Add(time.Hour * 3)) snapshotClass3.DeletionPolicy = snapshotv1.VolumeSnapshotContentRetain + sc1Expected := snapshotClass1.DeepCopy() + sc1Expected.Annotations = make(map[string]string) + sc1Expected.Annotations["kubesphere.io/snapshot-count"] = "0" + + sc2Expected := snapshotClass2.DeepCopy() + sc2Expected.Annotations = make(map[string]string) + sc2Expected.Annotations["kubesphere.io/snapshot-count"] = "0" + + sc3Expected := snapshotClass3.DeepCopy() + sc3Expected.Annotations = make(map[string]string) + sc3Expected.Annotations["kubesphere.io/snapshot-count"] = "0" + volumeSnapshotClasses := []interface{}{snapshotClass1, snapshotClass2, snapshotClass3} for _, s := range volumeSnapshotClasses { @@ -86,7 +98,7 @@ func TestListVolumeSnapshotClass(t *testing.T) { snapshotClassList, err := getter.List("", query1) Expect(err).To(BeNil()) Expect(snapshotClassList.TotalItems).To(Equal(1)) - Expect(snapshotClassList.Items[0]).To(Equal(snapshotClass1)) + Expect(snapshotClassList.Items[0]).To(Equal(sc1Expected)) }) It("match driver", func() { @@ -95,7 +107,7 @@ func TestListVolumeSnapshotClass(t *testing.T) { snapshotClassList, err := getter.List("", query1) Expect(err).To(BeNil()) Expect(snapshotClassList.TotalItems).To(Equal(1)) - Expect(snapshotClassList.Items[0]).To(Equal(snapshotClass2)) + Expect(snapshotClassList.Items[0]).To(Equal(sc2Expected)) }) It("match deletionPolicy", func() { @@ -104,7 +116,7 @@ func TestListVolumeSnapshotClass(t *testing.T) { snapshotClassList, err := getter.List("", query1) Expect(err).To(BeNil()) Expect(snapshotClassList.TotalItems).To(Equal(1)) - Expect(snapshotClassList.Items[0]).To(Equal(snapshotClass3)) + Expect(snapshotClassList.Items[0]).To(Equal(sc3Expected)) }) })