fix the ut

Signed-off-by: myan <[email protected]>
stolostron · Aug 21, 2024 · 4c6bc28 · 4c6bc28
1 parent 2f2bd54
commit 4c6bc28
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/transport/assembler_test.go b/pkg/transport/assembler_test.go
@@ -23,10 +23,12 @@ func TestAssembler(t *testing.T) {
 		},
 	}
 
+	transportConfig.IsManager = true
 	genericProducer, err := producer.NewGenericProducer(transportConfig)
 	assert.Nil(t, err)
 	genericProducer.SetDataLimit(5)
 
+	transportConfig.IsManager = false
 	genericConsumer, err := consumer.NewGenericConsumer(transportConfig)
 	assert.Nil(t, err)
 	go func() {