Add test for DescribeLogDirsResponse

describe_log_dirs_response_test.go

@@ -0,0 +1,79 @@
+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{
+		DescribeLogDirsResponseDirMetadata{
+			ErrorCode: 0,
+			Path:      "/kafka",
+			Topics: []DescribeLogDirsResponseTopic{
+				DescribeLogDirsResponseTopic{
+					Topic: "random",
+					Partitions: []DescribeLogDirsResponsePartition{
+						DescribeLogDirsResponsePartition{
+							PartitionID: 25,
+							Size:        125,
+							OffsetLag:   0,
+							IsTemporary: false,
+						},
+						DescribeLogDirsResponsePartition{
+							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")
+	}
+}