add protocol/apiversions unit test (segmentio#689)

duringnone · Jul 2, 2021 · 45672af · 45672af
1 parent 7113876
commit 45672af
Showing 1 changed file with 59 additions and 0 deletions.
diff --git a/protocol/apiversions/apiversions_test.go b/protocol/apiversions/apiversions_test.go
@@ -0,0 +1,59 @@
+package apiversions_test
+
+import (
+	"testing"
+
+	"github.com/segmentio/kafka-go/protocol/apiversions"
+	"github.com/segmentio/kafka-go/protocol/prototest"
+)
+
+const (
+	v0 = 0
+	v1 = 1
+	v2 = 2
+)
+
+func TestApiversionsRequest(t *testing.T) {
+	prototest.TestRequest(t, v0, &apiversions.Request{})
+
+	prototest.TestRequest(t, v1, &apiversions.Request{})
+
+	prototest.TestRequest(t, v2, &apiversions.Request{})
+}
+
+func TestApiversionsResponse(t *testing.T) {
+	prototest.TestResponse(t, v0, &apiversions.Response{
+		ErrorCode: 0,
+		ApiKeys: []apiversions.ApiKeyResponse{
+			{
+				ApiKey:     0,
+				MinVersion: 0,
+				MaxVersion: 2,
+			},
+		},
+	})
+
+	prototest.TestResponse(t, v1, &apiversions.Response{
+		ErrorCode: 0,
+		ApiKeys: []apiversions.ApiKeyResponse{
+			{
+				ApiKey:     0,
+				MinVersion: 0,
+				MaxVersion: 2,
+			},
+		},
+		ThrottleTimeMs: 10,
+	})
+
+	prototest.TestResponse(t, v2, &apiversions.Response{
+		ErrorCode: 0,
+		ApiKeys: []apiversions.ApiKeyResponse{
+			{
+				ApiKey:     0,
+				MinVersion: 0,
+				MaxVersion: 2,
+			},
+		},
+		ThrottleTimeMs: 50,
+	})
+}