sarama/describe_log_dirs_response_test.go

package sarama

import (
	"testing"
)

var (
	describeLogDirsResponseEmpty = []byte{
		0, 0, 0, 0, // no throttle time
		0, 0, 0, 0, // no log dirs
	}

	describeLogDirsResponseTwoPartitions = []byte{
		0, 0, 0, 0, // no throttle time
		0, 0, 0, 1, // One describe log dir (array length)
		0, 0, // No error code
		0, 6, // Character length of path (6 chars)
		'/', 'k', 'a', 'f', 'k', 'a',
		0, 0, 0, 1, // One DescribeLogDirsResponseTopic (array length)
		0, 6, // Character length of "random" topic (6 chars)
		'r', 'a', 'n', 'd', 'o', 'm', // Topic name
		0, 0, 0, 2, // Two DescribeLogDirsResponsePartition (array length)
		0, 0, 0, 25, // PartitionID 25
		0, 0, 0, 0, 0, 0, 0, 125, // Log Size
		0, 0, 0, 0, 0, 0, 0, 0, // OffsetLag
		0,           // IsTemporary = false
		0, 0, 0, 26, // PartitionID 25
		0, 0, 0, 0, 0, 0, 0, 100, // Log Size
		0, 0, 0, 0, 0, 0, 0, 0, // OffsetLag
		0, // IsTemporary = false
	}
)

func TestDescribeLogDirsResponse(t *testing.T) {
	// Test empty response
	response := &DescribeLogDirsResponse{
		LogDirs: []DescribeLogDirsResponseDirMetadata{},
	}
	testVersionDecodable(t, "empty", response, describeLogDirsResponseEmpty, 0)
	if len(response.LogDirs) != 0 {
		t.Error("Expected no log dirs")
	}

	response.LogDirs = []DescribeLogDirsResponseDirMetadata{
		{
			ErrorCode: 0,
			Path:      "/kafka",
			Topics: []DescribeLogDirsResponseTopic{
				{
					Topic: "random",
					Partitions: []DescribeLogDirsResponsePartition{
						{
							PartitionID: 25,
							Size:        125,
							OffsetLag:   0,
							IsTemporary: false,
						},
						{
							PartitionID: 26,
							Size:        100,
							OffsetLag:   0,
							IsTemporary: false,
						},
					},
				},
			},
		},
	}
	testVersionDecodable(t, "two partitions", response, describeLogDirsResponseTwoPartitions, 0)
	if len(response.LogDirs) != 1 {
		t.Error("Expected one log dirs")
	}
	if len(response.LogDirs[0].Topics) != 1 {
		t.Error("Expected one topic in log dirs")
	}
	if len(response.LogDirs[0].Topics[0].Partitions) != 2 {
		t.Error("Expected two partitions")
	}
}