increase test cov

Signed-off-by: zachaller <[email protected]>
argoproj · Jul 21, 2023 · 64e2ad9 · 64e2ad9
1 parent 2842baa
commit 64e2ad9
Showing 1 changed file with 40 additions and 0 deletions.
diff --git a/utils/annotations/annotations_test.go b/utils/annotations/annotations_test.go
@@ -444,3 +444,43 @@ func TestReplicasAnnotationsNeedUpdate(t *testing.T) {
 		})
 	}
 }
+
+func TestGetRevisionAnnotation(t *testing.T) {
+	rev, found := GetRevisionAnnotation(&v1alpha1.Rollout{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "foo",
+			Namespace: metav1.NamespaceDefault,
+			Annotations: map[string]string{
+				RevisionAnnotation: "1",
+			},
+		},
+	})
+	assert.True(t, found)
+	assert.Equal(t, int32(1), rev)
+
+	rev, found = GetRevisionAnnotation(&v1alpha1.Rollout{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:        "foo",
+			Namespace:   metav1.NamespaceDefault,
+			Annotations: map[string]string{},
+		},
+	})
+	assert.False(t, found)
+	assert.Equal(t, int32(0), rev)
+
+	rev, found = GetRevisionAnnotation(nil)
+	assert.False(t, found)
+	assert.Equal(t, int32(0), rev)
+
+	rev, found = GetRevisionAnnotation(&v1alpha1.Rollout{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "foo",
+			Namespace: metav1.NamespaceDefault,
+			Annotations: map[string]string{
+				RevisionAnnotation: "abc",
+			},
+		},
+	})
+	assert.False(t, found)
+	assert.Equal(t, int32(0), rev)
+}