Skip to content

Commit

Permalink
adding base/kafka_test.go for code coverage (#2990)
Browse files Browse the repository at this point in the history
Signed-off-by: zhaque44 <haque.zubair@gmail.com>
  • Loading branch information
zhaque44 committed Jan 27, 2024
1 parent 03ba32d commit 16c4dc1
Showing 1 changed file with 105 additions and 0 deletions.
105 changes: 105 additions & 0 deletions eventbus/kafka/base/kafka_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,105 @@
package base

import (
"testing"

"github.com/IBM/sarama"
eventbusv1alpha1 "github.com/argoproj/argo-events/pkg/apis/eventbus/v1alpha1"
"github.com/stretchr/testify/assert"
"go.uber.org/zap"
)

func TestBrokers(t *testing.T) {
config := &eventbusv1alpha1.KafkaBus{
URL: "broker1:9092,broker2:9092",
}

logger := zap.NewNop().Sugar()
kafka := NewKafka(config, logger)

expectedBrokers := []string{"broker1:9092", "broker2:9092"}
actualBrokers := kafka.Brokers()

assert.Equal(t, expectedBrokers, actualBrokers)
}

func TestConfig(t *testing.T) {
config := &eventbusv1alpha1.KafkaBus{
URL: "localhost:9092",
}

logger := zap.NewNop().Sugar()

kafka := NewKafka(config, logger)

saramaConfig, err := kafka.Config()

assert.NoError(t, err)
assert.NotNil(t, saramaConfig)
assert.Equal(t, sarama.OffsetNewest, saramaConfig.Consumer.Offsets.Initial)
assert.Equal(t, sarama.WaitForAll, saramaConfig.Producer.RequiredAcks)
}

func TestConfig_StartOldest(t *testing.T) {
config := &eventbusv1alpha1.KafkaBus{
URL: "localhost:9092",
ConsumerGroup: &eventbusv1alpha1.KafkaConsumerGroup{
StartOldest: true,
},
}

logger := zap.NewNop().Sugar()

kafka := NewKafka(config, logger)

saramaConfig, err := kafka.Config()

assert.NoError(t, err)
assert.NotNil(t, saramaConfig)
assert.Equal(t, sarama.OffsetOldest, saramaConfig.Consumer.Offsets.Initial)
}

func TestConfig_NoSASL(t *testing.T) {
config := &eventbusv1alpha1.KafkaBus{
URL: "localhost:9092",
SASL: nil,
}

logger := zap.NewNop().Sugar()

kafka := NewKafka(config, logger)

saramaConfig, err := kafka.Config()

assert.NoError(t, err)
assert.NotNil(t, saramaConfig)
assert.False(t, saramaConfig.Net.SASL.Enable)
}

func TestNewKafka(t *testing.T) {
config := &eventbusv1alpha1.KafkaBus{
URL: "localhost:9092",
}

logger := zap.NewNop().Sugar()

kafka := NewKafka(config, logger)

assert.NotNil(t, kafka)
assert.NotNil(t, kafka.Logger)
assert.NotNil(t, kafka.config)
}

func TestNewKafka_EmptyURL(t *testing.T) {
config := &eventbusv1alpha1.KafkaBus{
URL: "",
}

logger := zap.NewNop().Sugar()

kafka := NewKafka(config, logger)

assert.NotNil(t, kafka)
assert.NotNil(t, kafka.Logger)
assert.NotNil(t, kafka.config)
}

0 comments on commit 16c4dc1

Please sign in to comment.