32 lines
703 B
Go
32 lines
703 B
Go
package kafka
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/influxdata/telegraf/plugins/serializers"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestConnectAndWrite(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("Skipping integration test in short mode")
|
|
}
|
|
|
|
brokers := []string{testutil.GetLocalHost() + ":9092"}
|
|
s, _ := serializers.NewInfluxSerializer()
|
|
k := &Kafka{
|
|
Brokers: brokers,
|
|
Topic: "Test",
|
|
serializer: s,
|
|
}
|
|
|
|
// Verify that we can connect to the Kafka broker
|
|
err := k.Connect()
|
|
require.NoError(t, err)
|
|
|
|
// Verify that we can successfully write data to the kafka broker
|
|
err = k.Write(testutil.MockMetrics())
|
|
require.NoError(t, err)
|
|
}
|