mirror of
https://github.com/fluxcd/flagger.git
synced 2026-04-15 06:57:34 +00:00
2e079ba7a1
The mirror option will be used to tell routers to configure traffic mirroring. Implement mirror for GetRoutes and SetRoutes for Istio. For other routers, GetRoutes always returns mirror == false, and SetRoutes ignores mirror. After this change there is no behavior change because no code sets mirror true (yet). Enhanced TestIstioRouter_SetRoutes and TestIstioRouter_GetRoutes.
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package router
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
func TestIngressRouter_Reconcile(t *testing.T) {
|
|
mocks := setupfakeClients()
|
|
router := &IngressRouter{
|
|
logger: mocks.logger,
|
|
kubeClient: mocks.kubeClient,
|
|
annotationsPrefix: "custom.ingress.kubernetes.io",
|
|
}
|
|
|
|
err := router.Reconcile(mocks.ingressCanary)
|
|
if err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
|
|
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.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,
|
|
annotationsPrefix: "prefix1.nginx.ingress.kubernetes.io",
|
|
}
|
|
|
|
err := router.Reconcile(mocks.ingressCanary)
|
|
if err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
|
|
p, c, m, err := router.GetRoutes(mocks.ingressCanary)
|
|
if err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
|
|
p = 50
|
|
c = 50
|
|
m = false
|
|
|
|
err = router.SetRoutes(mocks.ingressCanary, p, c, m)
|
|
if err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
|
|
canaryAn := "prefix1.nginx.ingress.kubernetes.io/canary"
|
|
canaryWeightAn := "prefix1.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
|
|
m = false
|
|
|
|
err = router.SetRoutes(mocks.ingressCanary, p, c, m)
|
|
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])
|
|
}
|
|
}
|