goreplay/input_kafka_test.go

package main

import (
	"github.com/Shopify/sarama"
	"github.com/Shopify/sarama/mocks"
	"testing"
)

func TestInputKafkaRAW(t *testing.T) {
	consumer := mocks.NewConsumer(t, nil)
	defer consumer.Close()

	consumer.ExpectConsumePartition("test", 0, mocks.AnyOffset).YieldMessage(&sarama.ConsumerMessage{Value: []byte("1 2 3\nGET / HTTP1.1\r\nHeader: 1\r\n\r\n")})
	consumer.SetTopicMetadata(
		map[string][]int32{"test": []int32{0}},
	)

	input := NewKafkaInput("", &KafkaConfig{
		consumer: consumer,
		topic:    "test",
		useJSON:  false,
	})

	buf := make([]byte, 1024)
	n, err := input.Read(buf)

	if err != nil {
		t.Fatal(err)
	}

	if string(buf[:n]) != "1 2 3\nGET / HTTP1.1\r\nHeader: 1\r\n\r\n" {
		t.Error("Message not properly decoded: ", string(buf[:n]), n)
	}
}

func TestInputKafkaJSON(t *testing.T) {
	consumer := mocks.NewConsumer(t, nil)
	defer consumer.Close()

	consumer.ExpectConsumePartition("test", 0, mocks.AnyOffset).YieldMessage(&sarama.ConsumerMessage{Value: []byte(`{"Req_URL":"/","Req_Type":"1","Req_ID":"2","Req_Ts":"3","Req_Method":"GET","Req_Headers":{"Header":"1"}}`)})
	consumer.SetTopicMetadata(
		map[string][]int32{"test": []int32{0}},
	)

	input := NewKafkaInput("", &KafkaConfig{
		consumer: consumer,
		topic:    "test",
		useJSON:  true,
	})

	buf := make([]byte, 1024)
	n, err := input.Read(buf)

	if err != nil {
		t.Fatal(err)
	}

	if string(buf[:n]) != "1 2 3\nGET / HTTP/1.1\r\nHeader: 1\r\n\r\n" {
		t.Error("Message not properly decoded: ", string(buf[:n]), n)
	}
}