File: heartbeat_test.go

package info (click to toggle)
golang-github-segmentio-kafka-go 0.4.49%2Bds1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,292 kB
  • sloc: sh: 17; makefile: 10
file content (44 lines) | stat: -rw-r--r-- 1,019 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package heartbeat_test

import (
	"testing"

	"github.com/segmentio/kafka-go/protocol/heartbeat"
	"github.com/segmentio/kafka-go/protocol/prototest"
)

func TestHeartbeatRequest(t *testing.T) {
	// Versions 0-3 have all the same fields.
	for _, version := range []int16{0, 1, 2, 3} {
		prototest.TestRequest(t, version, &heartbeat.Request{
			GroupID:      "group-1",
			GenerationID: 1,
			MemberID:     "member-1",
		})
	}

	for _, version := range []int16{4} {
		prototest.TestRequest(t, version, &heartbeat.Request{
			GroupID:         "group-2",
			GenerationID:    10,
			MemberID:        "member-2",
			GroupInstanceID: "instace-1",
		})
	}
}

func TestHeartbeatResponse(t *testing.T) {
	for _, version := range []int16{0} {
		prototest.TestResponse(t, version, &heartbeat.Response{
			ErrorCode: 4,
		})
	}

	// Versions 1-4 have all the same fields.
	for _, version := range []int16{1, 2, 3, 4} {
		prototest.TestResponse(t, version, &heartbeat.Response{
			ErrorCode:      4,
			ThrottleTimeMs: 10,
		})
	}
}