github/flagger
1
0
Fork 0
mirror of https://github.com/fluxcd/flagger.git synced 2026-04-15 06:57:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add unit tests for ingress A/B Testing

Browse Source
This commit is contained in:
stefanprodan
2020-04-02 01:35:15 +03:00
parent 38ef4ef4d8
commit b8e9f57e1e
1 changed files with 89 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
pkg/router/ingress_test.go
View File

@@ -6,8 +6,11 @@ import (
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
flaggerv1 "github.com/weaveworks/flagger/pkg/apis/flagger/v1beta1"
istiov1alpha1 "github.com/weaveworks/flagger/pkg/apis/istio/common/v1alpha1"
istiov1alpha3 "github.com/weaveworks/flagger/pkg/apis/istio/v1alpha3"
)
func TestIngressRouter_Reconcile(t *testing.T) {
@@ -79,3 +82,88 @@ func TestIngressRouter_GetSetRoutes(t *testing.T) {
assert.Equal(t, "false", inCanary.Annotations[canaryAn])
assert.Equal(t, "0", inCanary.Annotations[canaryWeightAn])
}
func TestIngressRouter_ABTest(t *testing.T) {
mocks := newFixture(nil)
router := &IngressRouter{
logger: mocks.logger,
kubeClient: mocks.kubeClient,
annotationsPrefix: "nginx.ingress.kubernetes.io",
}
tables := []struct {
makeCanary func() *flaggerv1.Canary
annotation string
}{
// Header exact match
{
makeCanary: func() *flaggerv1.Canary {
mocks.ingressCanary.Spec.Analysis.Iterations = 1
mocks.ingressCanary.Spec.Analysis.Match = []istiov1alpha3.HTTPMatchRequest{
{
Headers: map[string]istiov1alpha1.StringMatch{
"x-user-type": {
Exact: "test",
},
},
},
}
return mocks.ingressCanary
},
annotation: router.GetAnnotationWithPrefix("canary-by-header-value"),
},
// Header regex match
{
makeCanary: func() *flaggerv1.Canary {
mocks.ingressCanary.Spec.Analysis.Iterations = 1
mocks.ingressCanary.Spec.Analysis.Match = []istiov1alpha3.HTTPMatchRequest{
{
Headers: map[string]istiov1alpha1.StringMatch{
"x-user-type": {
Regex: "test",
},
},
},
}
return mocks.ingressCanary
},
annotation: router.GetAnnotationWithPrefix("canary-by-header-pattern"),
},
// Cookie exact match
{
makeCanary: func() *flaggerv1.Canary {
mocks.ingressCanary.Spec.Analysis.Iterations = 1
mocks.ingressCanary.Spec.Analysis.Match = []istiov1alpha3.HTTPMatchRequest{
{
Headers: map[string]istiov1alpha1.StringMatch{
"cookie": {
Exact: "test",
},
},
},
}
return mocks.ingressCanary
},
annotation: router.GetAnnotationWithPrefix("canary-by-cookie"),
},
}
for _, table := range tables {
err := router.Reconcile(table.makeCanary())
require.NoError(t, err)
err = router.SetRoutes(table.makeCanary(), 50, 50, false)
require.NoError(t, err)
canaryAn := router.GetAnnotationWithPrefix("canary")
canaryName := fmt.Sprintf("%s-canary", table.makeCanary().Spec.IngressRef.Name)
inCanary, err := router.kubeClient.NetworkingV1beta1().Ingresses("default").Get(canaryName, metav1.GetOptions{})
require.NoError(t, err)
// test initialisation
assert.Equal(t, "true", inCanary.Annotations[canaryAn])
assert.Equal(t, "test", inCanary.Annotations[table.annotation])
}
}