|
|
@@ -41,11 +41,11 @@ func TestSimpleProducer(t *testing.T) {
|
|
|
|
|
|
metadataResponse := new(MetadataResponse)
|
|
|
metadataResponse.AddBroker(leader.Addr(), leader.BrokerID())
|
|
|
- metadataResponse.AddTopicPartition("my_topic", 0, 2, nil, nil, NoError)
|
|
|
+ metadataResponse.AddTopicPartition("my_topic", 0, 2, nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataResponse)
|
|
|
|
|
|
prodSuccess := new(ProduceResponse)
|
|
|
- prodSuccess.AddTopicPartition("my_topic", 0, NoError)
|
|
|
+ prodSuccess.AddTopicPartition("my_topic", 0, ErrNoError)
|
|
|
for i := 0; i < 10; i++ {
|
|
|
leader.Returns(prodSuccess)
|
|
|
}
|
|
|
@@ -79,11 +79,11 @@ func TestConcurrentSimpleProducer(t *testing.T) {
|
|
|
|
|
|
metadataResponse := new(MetadataResponse)
|
|
|
metadataResponse.AddBroker(leader.Addr(), leader.BrokerID())
|
|
|
- metadataResponse.AddTopicPartition("my_topic", 0, 2, nil, nil, NoError)
|
|
|
+ metadataResponse.AddTopicPartition("my_topic", 0, 2, nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataResponse)
|
|
|
|
|
|
prodSuccess := new(ProduceResponse)
|
|
|
- prodSuccess.AddTopicPartition("my_topic", 0, NoError)
|
|
|
+ prodSuccess.AddTopicPartition("my_topic", 0, ErrNoError)
|
|
|
leader.Returns(prodSuccess)
|
|
|
|
|
|
client, err := NewClient("client_id", []string{seedBroker.Addr()}, nil)
|
|
|
@@ -124,11 +124,11 @@ func TestProducer(t *testing.T) {
|
|
|
|
|
|
metadataResponse := new(MetadataResponse)
|
|
|
metadataResponse.AddBroker(leader.Addr(), leader.BrokerID())
|
|
|
- metadataResponse.AddTopicPartition("my_topic", 0, leader.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataResponse.AddTopicPartition("my_topic", 0, leader.BrokerID(), nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataResponse)
|
|
|
|
|
|
prodSuccess := new(ProduceResponse)
|
|
|
- prodSuccess.AddTopicPartition("my_topic", 0, NoError)
|
|
|
+ prodSuccess.AddTopicPartition("my_topic", 0, ErrNoError)
|
|
|
leader.Returns(prodSuccess)
|
|
|
|
|
|
client, err := NewClient("client_id", []string{seedBroker.Addr()}, nil)
|
|
|
@@ -176,11 +176,11 @@ func TestProducerMultipleFlushes(t *testing.T) {
|
|
|
|
|
|
metadataResponse := new(MetadataResponse)
|
|
|
metadataResponse.AddBroker(leader.Addr(), leader.BrokerID())
|
|
|
- metadataResponse.AddTopicPartition("my_topic", 0, leader.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataResponse.AddTopicPartition("my_topic", 0, leader.BrokerID(), nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataResponse)
|
|
|
|
|
|
prodSuccess := new(ProduceResponse)
|
|
|
- prodSuccess.AddTopicPartition("my_topic", 0, NoError)
|
|
|
+ prodSuccess.AddTopicPartition("my_topic", 0, ErrNoError)
|
|
|
leader.Returns(prodSuccess)
|
|
|
leader.Returns(prodSuccess)
|
|
|
leader.Returns(prodSuccess)
|
|
|
@@ -231,16 +231,16 @@ func TestProducerMultipleBrokers(t *testing.T) {
|
|
|
metadataResponse := new(MetadataResponse)
|
|
|
metadataResponse.AddBroker(leader0.Addr(), leader0.BrokerID())
|
|
|
metadataResponse.AddBroker(leader1.Addr(), leader1.BrokerID())
|
|
|
- metadataResponse.AddTopicPartition("my_topic", 0, leader0.BrokerID(), nil, nil, NoError)
|
|
|
- metadataResponse.AddTopicPartition("my_topic", 1, leader1.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataResponse.AddTopicPartition("my_topic", 0, leader0.BrokerID(), nil, nil, ErrNoError)
|
|
|
+ metadataResponse.AddTopicPartition("my_topic", 1, leader1.BrokerID(), nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataResponse)
|
|
|
|
|
|
prodResponse0 := new(ProduceResponse)
|
|
|
- prodResponse0.AddTopicPartition("my_topic", 0, NoError)
|
|
|
+ prodResponse0.AddTopicPartition("my_topic", 0, ErrNoError)
|
|
|
leader0.Returns(prodResponse0)
|
|
|
|
|
|
prodResponse1 := new(ProduceResponse)
|
|
|
- prodResponse1.AddTopicPartition("my_topic", 1, NoError)
|
|
|
+ prodResponse1.AddTopicPartition("my_topic", 1, ErrNoError)
|
|
|
leader1.Returns(prodResponse1)
|
|
|
|
|
|
client, err := NewClient("client_id", []string{seedBroker.Addr()}, nil)
|
|
|
@@ -288,7 +288,7 @@ func TestProducerFailureRetry(t *testing.T) {
|
|
|
|
|
|
metadataLeader1 := new(MetadataResponse)
|
|
|
metadataLeader1.AddBroker(leader1.Addr(), leader1.BrokerID())
|
|
|
- metadataLeader1.AddTopicPartition("my_topic", 0, leader1.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataLeader1.AddTopicPartition("my_topic", 0, leader1.BrokerID(), nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataLeader1)
|
|
|
|
|
|
client, err := NewClient("client_id", []string{seedBroker.Addr()}, nil)
|
|
|
@@ -310,16 +310,16 @@ func TestProducerFailureRetry(t *testing.T) {
|
|
|
producer.Input() <- &MessageToSend{Topic: "my_topic", Key: nil, Value: StringEncoder(TestMessage)}
|
|
|
}
|
|
|
prodNotLeader := new(ProduceResponse)
|
|
|
- prodNotLeader.AddTopicPartition("my_topic", 0, NotLeaderForPartition)
|
|
|
+ prodNotLeader.AddTopicPartition("my_topic", 0, ErrNotLeaderForPartition)
|
|
|
leader1.Returns(prodNotLeader)
|
|
|
|
|
|
metadataLeader2 := new(MetadataResponse)
|
|
|
metadataLeader2.AddBroker(leader2.Addr(), leader2.BrokerID())
|
|
|
- metadataLeader2.AddTopicPartition("my_topic", 0, leader2.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataLeader2.AddTopicPartition("my_topic", 0, leader2.BrokerID(), nil, nil, ErrNoError)
|
|
|
leader1.Returns(metadataLeader2)
|
|
|
|
|
|
prodSuccess := new(ProduceResponse)
|
|
|
- prodSuccess.AddTopicPartition("my_topic", 0, NoError)
|
|
|
+ prodSuccess.AddTopicPartition("my_topic", 0, ErrNoError)
|
|
|
leader2.Returns(prodSuccess)
|
|
|
for i := 0; i < 10; i++ {
|
|
|
select {
|
|
|
@@ -366,7 +366,7 @@ func TestProducerBrokerBounce(t *testing.T) {
|
|
|
|
|
|
metadataResponse := new(MetadataResponse)
|
|
|
metadataResponse.AddBroker(leaderAddr, leader.BrokerID())
|
|
|
- metadataResponse.AddTopicPartition("my_topic", 0, leader.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataResponse.AddTopicPartition("my_topic", 0, leader.BrokerID(), nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataResponse)
|
|
|
|
|
|
client, err := NewClient("client_id", []string{seedBroker.Addr()}, nil)
|
|
|
@@ -391,7 +391,7 @@ func TestProducerBrokerBounce(t *testing.T) {
|
|
|
seedBroker.Returns(metadataResponse) // tell it to go to broker 2 again
|
|
|
|
|
|
prodSuccess := new(ProduceResponse)
|
|
|
- prodSuccess.AddTopicPartition("my_topic", 0, NoError)
|
|
|
+ prodSuccess.AddTopicPartition("my_topic", 0, ErrNoError)
|
|
|
leader.Returns(prodSuccess)
|
|
|
for i := 0; i < 10; i++ {
|
|
|
select {
|
|
|
@@ -420,7 +420,7 @@ func TestProducerBrokerBounceWithStaleMetadata(t *testing.T) {
|
|
|
|
|
|
metadataLeader1 := new(MetadataResponse)
|
|
|
metadataLeader1.AddBroker(leader1.Addr(), leader1.BrokerID())
|
|
|
- metadataLeader1.AddTopicPartition("my_topic", 0, leader1.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataLeader1.AddTopicPartition("my_topic", 0, leader1.BrokerID(), nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataLeader1)
|
|
|
|
|
|
client, err := NewClient("client_id", []string{seedBroker.Addr()}, nil)
|
|
|
@@ -448,11 +448,11 @@ func TestProducerBrokerBounceWithStaleMetadata(t *testing.T) {
|
|
|
// ok fine, tell it to go to leader2 finally
|
|
|
metadataLeader2 := new(MetadataResponse)
|
|
|
metadataLeader2.AddBroker(leader2.Addr(), leader2.BrokerID())
|
|
|
- metadataLeader2.AddTopicPartition("my_topic", 0, leader2.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataLeader2.AddTopicPartition("my_topic", 0, leader2.BrokerID(), nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataLeader2)
|
|
|
|
|
|
prodSuccess := new(ProduceResponse)
|
|
|
- prodSuccess.AddTopicPartition("my_topic", 0, NoError)
|
|
|
+ prodSuccess.AddTopicPartition("my_topic", 0, ErrNoError)
|
|
|
leader2.Returns(prodSuccess)
|
|
|
for i := 0; i < 10; i++ {
|
|
|
select {
|
|
|
@@ -481,7 +481,7 @@ func TestProducerMultipleRetries(t *testing.T) {
|
|
|
|
|
|
metadataLeader1 := new(MetadataResponse)
|
|
|
metadataLeader1.AddBroker(leader1.Addr(), leader1.BrokerID())
|
|
|
- metadataLeader1.AddTopicPartition("my_topic", 0, leader1.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataLeader1.AddTopicPartition("my_topic", 0, leader1.BrokerID(), nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataLeader1)
|
|
|
|
|
|
client, err := NewClient("client_id", []string{seedBroker.Addr()}, nil)
|
|
|
@@ -503,12 +503,12 @@ func TestProducerMultipleRetries(t *testing.T) {
|
|
|
producer.Input() <- &MessageToSend{Topic: "my_topic", Key: nil, Value: StringEncoder(TestMessage)}
|
|
|
}
|
|
|
prodNotLeader := new(ProduceResponse)
|
|
|
- prodNotLeader.AddTopicPartition("my_topic", 0, NotLeaderForPartition)
|
|
|
+ prodNotLeader.AddTopicPartition("my_topic", 0, ErrNotLeaderForPartition)
|
|
|
leader1.Returns(prodNotLeader)
|
|
|
|
|
|
metadataLeader2 := new(MetadataResponse)
|
|
|
metadataLeader2.AddBroker(leader2.Addr(), leader2.BrokerID())
|
|
|
- metadataLeader2.AddTopicPartition("my_topic", 0, leader2.BrokerID(), nil, nil, NoError)
|
|
|
+ metadataLeader2.AddTopicPartition("my_topic", 0, leader2.BrokerID(), nil, nil, ErrNoError)
|
|
|
seedBroker.Returns(metadataLeader2)
|
|
|
leader2.Returns(prodNotLeader)
|
|
|
seedBroker.Returns(metadataLeader1)
|
|
|
@@ -518,7 +518,7 @@ func TestProducerMultipleRetries(t *testing.T) {
|
|
|
seedBroker.Returns(metadataLeader2)
|
|
|
|
|
|
prodSuccess := new(ProduceResponse)
|
|
|
- prodSuccess.AddTopicPartition("my_topic", 0, NoError)
|
|
|
+ prodSuccess.AddTopicPartition("my_topic", 0, ErrNoError)
|
|
|
leader2.Returns(prodSuccess)
|
|
|
for i := 0; i < 10; i++ {
|
|
|
select {
|