Forráskód Böngészése

Enable errcheck on subpackages, and address error handling that was missed before.

Willem van Bergen 10 éve
szülő
commit
25161444de
3 módosított fájl, 33 hozzáadás és 11 törlés
  1. 1 1
      .travis.yml
  2. 9 3
      mocks/producer_test.go
  3. 23 7
      mocks/sync_producer_test.go

+ 1 - 1
.travis.yml

@@ -22,7 +22,7 @@ before_install:
 script:
 - go test -v -race ./...
 - go vet ./...
-- errcheck github.com/Shopify/sarama/
+- errcheck github.com/Shopify/sarama/...
 - if [[ -n $(go fmt ./...) ]]; then echo "Please run go fmt on your code." && exit 1; fi
 
 notifications:

+ 9 - 3
mocks/producer_test.go

@@ -55,7 +55,9 @@ func TestProducerReturnsExpectationsToChannels(t *testing.T) {
 		t.Error("Expected message 3 to be returned as error")
 	}
 
-	mp.Close()
+	if err := mp.Close(); err != nil {
+		t.Error(err)
+	}
 }
 
 func TestProducerWithTooFewExpectations(t *testing.T) {
@@ -66,7 +68,9 @@ func TestProducerWithTooFewExpectations(t *testing.T) {
 	mp.Input() <- &sarama.ProducerMessage{Topic: "test"}
 	mp.Input() <- &sarama.ProducerMessage{Topic: "test"}
 
-	mp.Close()
+	if err := mp.Close(); err != nil {
+		t.Error(err)
+	}
 
 	if len(trm.errors) != 1 {
 		t.Error("Expected to report an error")
@@ -80,7 +84,9 @@ func TestProducerWithTooManyExpectations(t *testing.T) {
 	mp.ExpectInputAndFail(sarama.ErrOutOfBrokers)
 
 	mp.Input() <- &sarama.ProducerMessage{Topic: "test"}
-	mp.Close()
+	if err := mp.Close(); err != nil {
+		t.Error(err)
+	}
 
 	if len(trm.errors) != 1 {
 		t.Error("Expected to report an error")

+ 23 - 7
mocks/sync_producer_test.go

@@ -15,7 +15,11 @@ func TestMockSyncProducerImplementsSyncProducerInterface(t *testing.T) {
 
 func TestSyncProducerReturnsExpectationsToSendMessage(t *testing.T) {
 	sp := NewSyncProducer(t, nil)
-	defer sp.Close()
+	defer func() {
+		if err := sp.Close(); err != nil {
+			t.Error(err)
+		}
+	}()
 
 	sp.ExpectSendMessageAndSucceed()
 	sp.ExpectSendMessageAndSucceed()
@@ -47,7 +51,9 @@ func TestSyncProducerReturnsExpectationsToSendMessage(t *testing.T) {
 		t.Errorf("The third message should not have been produced successfully")
 	}
 
-	sp.Close()
+	if err := sp.Close(); err != nil {
+		t.Error(err)
+	}
 }
 
 func TestSyncProducerWithTooManyExpectations(t *testing.T) {
@@ -57,9 +63,13 @@ func TestSyncProducerWithTooManyExpectations(t *testing.T) {
 	sp.ExpectSendMessageAndSucceed()
 	sp.ExpectSendMessageAndFail(sarama.ErrOutOfBrokers)
 
-	sp.SendMessage("test", nil, sarama.StringEncoder("test"))
+	if _, _, err := sp.SendMessage("test", nil, sarama.StringEncoder("test")); err != nil {
+		t.Error("No error expected on first SendMessage call", err)
+	}
 
-	sp.Close()
+	if err := sp.Close(); err != nil {
+		t.Error(err)
+	}
 
 	if len(trm.errors) != 1 {
 		t.Error("Expected to report an error")
@@ -72,10 +82,16 @@ func TestSyncProducerWithTooFewExpectations(t *testing.T) {
 	sp := NewSyncProducer(trm, nil)
 	sp.ExpectSendMessageAndSucceed()
 
-	sp.SendMessage("test", nil, sarama.StringEncoder("test"))
-	sp.SendMessage("test", nil, sarama.StringEncoder("test"))
+	if _, _, err := sp.SendMessage("test", nil, sarama.StringEncoder("test")); err != nil {
+		t.Error("No error expected on first SendMessage call", err)
+	}
+	if _, _, err := sp.SendMessage("test", nil, sarama.StringEncoder("test")); err != errOutOfExpectations {
+		t.Error("errOutOfExpectations expected on second SendMessage call, found:", err)
+	}
 
-	sp.Close()
+	if err := sp.Close(); err != nil {
+		t.Error(err)
+	}
 
 	if len(trm.errors) != 1 {
 		t.Error("Expected to report an error")