chore: add balancer tests

balancer.go

@@ -24,6 +24,6 @@ func GetBalancerReferenceHash() Balancer {
 	return &kafka.ReferenceHash{}
 }
 
-func GetBalancerRoundRobinh() Balancer {
+func GetBalancerRoundRobin() Balancer {
 	return &kafka.RoundRobin{}
 }

balancer_test.go

@@ -0,0 +1,66 @@
+package kafka
+
+import (
+	"reflect"
+	"testing"
+)
+
+func TestGetBalancerCRC32(t *testing.T) {
+	balancer := GetBalancerCRC32()
+	if balancer == nil {
+		t.Error("Expected non-nil balancer, got nil")
+	}
+	if reflect.TypeOf(balancer).String() != "*kafka.CRC32Balancer" {
+		t.Errorf("Expected *kafka.CRC32Balancer, got %s", reflect.TypeOf(balancer).String())
+	}
+}
+
+func TestGetBalancerHash(t *testing.T) {
+	balancer := GetBalancerHash()
+	if balancer == nil {
+		t.Error("Expected non-nil balancer, got nil")
+	}
+	if reflect.TypeOf(balancer).String() != "*kafka.Hash" {
+		t.Errorf("Expected *kafka.Hash, got %s", reflect.TypeOf(balancer).String())
+	}
+}
+
+func TestGetBalancerLeastBytes(t *testing.T) {
+	balancer := GetBalancerLeastBytes()
+	if balancer == nil {
+		t.Error("Expected non-nil balancer, got nil")
+	}
+	if reflect.TypeOf(balancer).String() != "*kafka.LeastBytes" {
+		t.Errorf("Expected *kafka.LeastBytes, got %s", reflect.TypeOf(balancer).String())
+	}
+}
+
+func TestGetBalancerMurmur2Balancer(t *testing.T) {
+	balancer := GetBalancerMurmur2Balancer()
+	if balancer == nil {
+		t.Error("Expected non-nil balancer, got nil")
+	}
+	if reflect.TypeOf(balancer).String() != "*kafka.Murmur2Balancer" {
+		t.Errorf("Expected *kafka.Murmur2Balancer, got %s", reflect.TypeOf(balancer).String())
+	}
+}
+
+func TestGetBalancerReferenceHash(t *testing.T) {
+	balancer := GetBalancerReferenceHash()
+	if balancer == nil {
+		t.Error("Expected non-nil balancer, got nil")
+	}
+	if reflect.TypeOf(balancer).String() != "*kafka.ReferenceHash" {
+		t.Errorf("Expected *kafka.ReferenceHash, got %s", reflect.TypeOf(balancer).String())
+	}
+}
+
+func TestGetBalancerRoundRobinh(t *testing.T) {
+	balancer := GetBalancerRoundRobin()
+	if balancer == nil {
+		t.Error("Expected non-nil balancer, got nil")
+	}
+	if reflect.TypeOf(balancer).String() != "*kafka.RoundRobin" {
+		t.Errorf("Expected *kafka.RoundRobin, got %s", reflect.TypeOf(balancer).String())
+	}
+}