fixing concurrent dispatcher test (#641)

Co-authored-by: Gabriel Freites <gfreites@vmware.com>
knative-extensions · Mar 11, 2022 · 402f765 · 402f765
1 parent 18d5420
commit 402f765
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/test/e2e/config/brokertrigger/brokertrigger.go b/test/e2e/config/brokertrigger/brokertrigger.go
@@ -30,14 +30,19 @@ func init() {
 }
 
 type Topology struct {
-	MessageCount int
-	Triggers     []duckv1.KReference
+	MessageCount, PrefetchCount int
+	Triggers                    []duckv1.KReference
 }
 
 func Install(topology Topology) feature.StepFn {
+	if topology.PrefetchCount == 0 {
+		topology.PrefetchCount = 1
+	}
+
 	args := map[string]interface{}{
-		"messageCount": topology.MessageCount,
-		"triggers":     topology.Triggers,
+		"messageCount":  topology.MessageCount,
+		"triggers":      topology.Triggers,
+		"prefetchCount": topology.PrefetchCount,
 	}
 	return func(ctx context.Context, t feature.T) {
 		if _, err := manifest.InstallLocalYaml(ctx, args); err != nil {

diff --git a/test/e2e/config/brokertrigger/trigger.yaml b/test/e2e/config/brokertrigger/trigger.yaml
@@ -5,6 +5,8 @@ kind: Trigger
 metadata:
   name: {{ printf "%s-%d" $kref.Name $index }}
   namespace: {{ $.namespace }}
+  annotations:
+     rabbitmq.eventing.knative.dev/prefetchCount: '{{ $.prefetchCount }}'
 spec:
   broker: testbroker
   subscriber:

diff --git a/test/e2e/dispatcher_test.go b/test/e2e/dispatcher_test.go
@@ -41,7 +41,8 @@ func ConcurrentDispatchTest() *feature.Feature {
 
 	f.Setup("install recorder", eventshub.Install("recorder", eventshub.StartReceiver, eventshub.ResponseWaitTime(time.Second)))
 	f.Setup("install test resources", brokertrigger.Install(brokertrigger.Topology{
-		MessageCount: 2,
+		MessageCount:  2,
+		PrefetchCount: 10,
 		Triggers: []duckv1.KReference{{
 			Kind: "Service",
 			Name: "recorder",