chore: error handling in consumer

kafka_test.go

@@ -274,6 +274,29 @@ func TestDefaultClient_Send(t *testing.T) {
 	}
 }
 
+func TestKafka_ConsumerErrorReturns(t *testing.T) {
+	k := &defaultClient{
+		consumer: &MockConsumer{
+			read: func(timeout time.Duration) (*kafka.Message, error) {
+				return &kafka.Message{}, kafka.NewError(kafka.ErrUnknownTopicOrPart, "test", false)
+			},
+			subscribe: func(topics []string, rebalanceCb kafka.RebalanceCb) (err error) {
+				return nil
+			},
+		},
+	}
+	handlers := map[string]func(msg []byte) error{
+		"topic": func(msg []byte) error {
+			return nil
+		},
+	}
+
+	err := k.Consume(handlers)
+	if err == nil {
+		t.Error("error expected")
+	}
+}
+
 func TestDefaultClient_Close(t *testing.T) {
 	consumerClosed := false
 	producerClosed := false