mirror of
https://github.com/fluxcd/flagger.git
synced 2026-04-15 06:57:34 +00:00
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package router
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
func TestIngressRouter_Reconcile(t *testing.T) {
|
|
mocks := newFixture(nil)
|
|
router := &IngressRouter{
|
|
logger: mocks.logger,
|
|
kubeClient: mocks.kubeClient,
|
|
annotationsPrefix: "custom.ingress.kubernetes.io",
|
|
}
|
|
|
|
err := router.Reconcile(mocks.ingressCanary)
|
|
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.ExtensionsV1beta1().Ingresses("default").Get(canaryName, metav1.GetOptions{})
|
|
require.NoError(t, err)
|
|
|
|
// test initialisation
|
|
assert.Equal(t, "false", inCanary.Annotations[canaryAn])
|
|
assert.Equal(t, "0", inCanary.Annotations[canaryWeightAn])
|
|
}
|
|
|
|
func TestIngressRouter_GetSetRoutes(t *testing.T) {
|
|
mocks := newFixture(nil)
|
|
router := &IngressRouter{
|
|
logger: mocks.logger,
|
|
kubeClient: mocks.kubeClient,
|
|
annotationsPrefix: "prefix1.nginx.ingress.kubernetes.io",
|
|
}
|
|
|
|
err := router.Reconcile(mocks.ingressCanary)
|
|
require.NoError(t, err)
|
|
|
|
p, c, m, err := router.GetRoutes(mocks.ingressCanary)
|
|
require.NoError(t, err)
|
|
|
|
p = 50
|
|
c = 50
|
|
m = false
|
|
|
|
err = router.SetRoutes(mocks.ingressCanary, p, c, m)
|
|
require.NoError(t, err)
|
|
|
|
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{})
|
|
require.NoError(t, err)
|
|
|
|
// test rollout
|
|
assert.Equal(t, "true", inCanary.Annotations[canaryAn])
|
|
assert.Equal(t, "50", inCanary.Annotations[canaryWeightAn])
|
|
|
|
p = 100
|
|
c = 0
|
|
m = false
|
|
|
|
err = router.SetRoutes(mocks.ingressCanary, p, c, m)
|
|
require.NoError(t, err)
|
|
|
|
inCanary, err = router.kubeClient.ExtensionsV1beta1().Ingresses("default").Get(canaryName, metav1.GetOptions{})
|
|
require.NoError(t, err)
|
|
|
|
// test promotion
|
|
assert.Equal(t, "false", inCanary.Annotations[canaryAn])
|
|
assert.Equal(t, "0", inCanary.Annotations[canaryWeightAn])
|
|
}
|