Function TestMurmur2Partitioner

partitioner_test.go:346–412

(t *testing.T)

}

func TestMurmur2Partitioner(t *testing.T) {
	partitioner := NewMurmur2Partitioner("mytopic")

	t.Run("single partition always returns 0", func(t *testing.T) {
		choice, err := partitioner.Partition(&ProducerMessage{Key: StringEncoder("foo")}, 1)
		if err != nil {
			t.Fatal(err)
		}
		if choice != 0 {
			t.Errorf("expected partition 0, got %d", choice)
		}
	})

	t.Run("nil key returns random partition in range", func(t *testing.T) {
		for range 50 {
			choice, err := partitioner.Partition(&ProducerMessage{}, 50)
			if err != nil {
				t.Fatal(err)
			}
			if choice < 0 || choice >= 50 {
				t.Errorf("partition %d out of range [0, 50)", choice)
			}
		}
	})

	t.Run("consistent for same key", func(t *testing.T) {
		assertPartitioningConsistent(t, partitioner, &ProducerMessage{Key: StringEncoder("foo")}, 100)
		assertPartitioningConsistent(t, partitioner, &ProducerMessage{Key: StringEncoder("bar")}, 100)
	})

	t.Run("partition in range for various keys", func(t *testing.T) {
		keys := []string{"foo", "bar", "baz", "kafka", "sarama", "1468509572224"}
		for _, key := range keys {
			choice, err := partitioner.Partition(&ProducerMessage{Key: StringEncoder(key)}, 100)
			if err != nil {
				t.Fatalf("key %q: %v", key, err)
			}
			if choice < 0 || choice >= 100 {
				t.Errorf("key %q: partition %d out of range [0, 100)", key, choice)
			}
		}
	})

	t.Run("known Java-compatible partition assignments", func(t *testing.T) {
		// these expected partitions were verified against the Apache Kafka Java
		// client's DefaultPartitioner with 100 partitions
		testCases := []struct {
			key       string
			partition int32
		}{
			{"foo", 16},
			{"bar", 45},
			{"baz", 80},
			{"kafka", 80},
			{"sarama", 11},
		}
		for _, tc := range testCases {
			choice, err := partitioner.Partition(&ProducerMessage{Key: StringEncoder(tc.key)}, 100)