Adjusted tests to new behaviour on init and promotion

fluxcd · May 13, 2021 · 9ee487f · 9ee487f
1 parent 4385de3
commit 9ee487f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pkg/router/ingress_test.go b/pkg/router/ingress_test.go
@@ -42,13 +42,15 @@ func TestIngressRouter_Reconcile(t *testing.T) {
  require.NoError(t, err)
 
  canaryAn := "custom.ingress.kubernetes.io/canary"
+ canaryWeightAn := "custom.ingress.kubernetes.io/canary-weight"
 
  canaryName := fmt.Sprintf("%s-canary", mocks.ingressCanary.Spec.IngressRef.Name)
  inCanary, err := router.kubeClient.NetworkingV1beta1().Ingresses("default").Get(context.TODO(), canaryName, metav1.GetOptions{})
  require.NoError(t, err)
 
  // test initialisation
- assert.Equal(t, "false", inCanary.Annotations[canaryAn])
+ assert.Equal(t, "true", inCanary.Annotations[canaryAn])
+ assert.Equal(t, "0", inCanary.Annotations[canaryWeightAn])
 }
 
 func TestIngressRouter_GetSetRoutes(t *testing.T) {
@@ -94,7 +96,8 @@ func TestIngressRouter_GetSetRoutes(t *testing.T) {
  require.NoError(t, err)
 
  // test promotion
- assert.Equal(t, "false", inCanary.Annotations[canaryAn])
+ assert.Equal(t, "true", inCanary.Annotations[canaryAn])
+ assert.Equal(t, "0", inCanary.Annotations[canaryWeightAn])
 }
 
 func TestIngressRouter_ABTest(t *testing.T) {