瀏覽代碼

Use the ConstantPartitioner in the functional test

Evan Huus 11 年之前
父節點
當前提交
acd62d54ab
共有 1 個文件被更改,包括 3 次插入1 次删除
  1. 3 1
      functional_test.go

+ 3 - 1
functional_test.go

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