Parcourir la source

Use the ConstantPartitioner in the functional test

Evan Huus il y a 11 ans
Parent
commit
acd62d54ab
1 fichiers modifiés avec 3 ajouts et 1 suppressions
  1. 3 1
      functional_test.go

+ 3 - 1
functional_test.go

@@ -60,7 +60,9 @@ func TestProducingMessages(t *testing.T) {
 	}
 	}
 	defer consumer.Close()
 	defer consumer.Close()
 
 
-	producer, err := NewProducer(client, nil)
+	producerConfig := NewProducerConfig()
+	producerConfig.Partitioner = &ConstantPartitioner{Constant: 0}
+	producer, err := NewProducer(client, producerConfig)
 	if err != nil {
 	if err != nil {
 		t.Fatal(err)
 		t.Fatal(err)
 	}
 	}