flagger/pkg/router/ingress_test.go

package router

import (
	"fmt"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"testing"
)

func TestIngressRouter_Reconcile(t *testing.T) {
	mocks := setupfakeClients()
	router := &IngressRouter{
		logger:     mocks.logger,
		kubeClient: mocks.kubeClient,
	}

	err := router.Reconcile(mocks.ingressCanary)
	if err != nil {
		t.Fatal(err.Error())
	}

	canaryAn := "nginx.ingress.kubernetes.io/canary"
	canaryWeightAn := "nginx.ingress.kubernetes.io/canary-weight"

	canaryName := fmt.Sprintf("%s-canary", mocks.ingressCanary.Spec.IngressRef.Name)
	inCanary, err := router.kubeClient.ExtensionsV1beta1().Ingresses("default").Get(canaryName, metav1.GetOptions{})
	if err != nil {
		t.Fatal(err.Error())
	}

	if _, ok := inCanary.Annotations[canaryAn]; !ok {
		t.Errorf("Canary annotation missing")
	}

	// test initialisation
	if inCanary.Annotations[canaryAn] != "false" {
		t.Errorf("Got canary annotation %v wanted false", inCanary.Annotations[canaryAn])
	}

	if inCanary.Annotations[canaryWeightAn] != "0" {
		t.Errorf("Got canary weight annotation %v wanted 0", inCanary.Annotations[canaryWeightAn])
	}
}

func TestIngressRouter_GetSetRoutes(t *testing.T) {
	mocks := setupfakeClients()
	router := &IngressRouter{
		logger:     mocks.logger,
		kubeClient: mocks.kubeClient,
	}

	err := router.Reconcile(mocks.ingressCanary)
	if err != nil {
		t.Fatal(err.Error())
	}

	p, c, err := router.GetRoutes(mocks.ingressCanary)
	if err != nil {
		t.Fatal(err.Error())
	}

	p = 50
	c = 50

	err = router.SetRoutes(mocks.ingressCanary, p, c)
	if err != nil {
		t.Fatal(err.Error())
	}

	canaryAn := "nginx.ingress.kubernetes.io/canary"
	canaryWeightAn := "nginx.ingress.kubernetes.io/canary-weight"

	canaryName := fmt.Sprintf("%s-canary", mocks.ingressCanary.Spec.IngressRef.Name)
	inCanary, err := router.kubeClient.ExtensionsV1beta1().Ingresses("default").Get(canaryName, metav1.GetOptions{})
	if err != nil {
		t.Fatal(err.Error())
	}

	if _, ok := inCanary.Annotations[canaryAn]; !ok {
		t.Errorf("Canary annotation missing")
	}

	// test rollout
	if inCanary.Annotations[canaryAn] != "true" {
		t.Errorf("Got canary annotation %v wanted true", inCanary.Annotations[canaryAn])
	}

	if inCanary.Annotations[canaryWeightAn] != "50" {
		t.Errorf("Got canary weight annotation %v wanted 50", inCanary.Annotations[canaryWeightAn])
	}

	p = 100
	c = 0

	err = router.SetRoutes(mocks.ingressCanary, p, c)
	if err != nil {
		t.Fatal(err.Error())
	}

	inCanary, err = router.kubeClient.ExtensionsV1beta1().Ingresses("default").Get(canaryName, metav1.GetOptions{})
	if err != nil {
		t.Fatal(err.Error())
	}

	// test promotion
	if inCanary.Annotations[canaryAn] != "false" {
		t.Errorf("Got canary annotation %v wanted false", inCanary.Annotations[canaryAn])
	}

	if inCanary.Annotations[canaryWeightAn] != "0" {
		t.Errorf("Got canary weight annotation %v wanted 0", inCanary.Annotations[canaryWeightAn])
	}
}