Function TestReaderAssignTopicPartitions

consumergroup_test.go:120–242

(t *testing.T)

}

func TestReaderAssignTopicPartitions(t *testing.T) {
	conn := &mockCoordinator{
		readPartitionsFunc: func(...string) ([]Partition, error) {
			return []Partition{
				{
					Topic: "topic-1",
					ID:    0,
				},
				{
					Topic: "topic-1",
					ID:    1,
				},
				{
					Topic: "topic-1",
					ID:    2,
				},
				{
					Topic: "topic-2",
					ID:    0,
				},
			}, nil
		},
	}

	newJoinGroupResponse := func(topicsByMemberID map[string][]string) func(v apiVersion) joinGroupResponse {
		return func(v apiVersion) joinGroupResponse {
			resp := joinGroupResponse{
				v:             v,
				GroupProtocol: RoundRobinGroupBalancer{}.ProtocolName(),
			}

			for memberID, topics := range topicsByMemberID {
				resp.Members = append(resp.Members, joinGroupResponseMember{
					MemberID: memberID,
					MemberMetadata: groupMetadata{
						Topics: topics,
					}.bytes(),
				})
			}

			return resp
		}
	}

	testCases := map[string]struct {
		MembersFunc func(v apiVersion) joinGroupResponse
		Assignments GroupMemberAssignments
	}{
		"nil": {
			MembersFunc: newJoinGroupResponse(nil),
			Assignments: GroupMemberAssignments{},
		},
		"one member, one topic": {
			MembersFunc: newJoinGroupResponse(map[string][]string{
				"member-1": {"topic-1"},
			}),
			Assignments: GroupMemberAssignments{
				"member-1": map[string][]int{