Skip to content

Commit

Permalink
change return type of MockJoinGroupResponse.For, MockHeartbeatRespons…
Browse files Browse the repository at this point in the history
…e.For & MockLeaveGroupResponse.For to encoderWithHeader
  • Loading branch information
Kranti Deep committed Jul 12, 2020
1 parent ac5ca79 commit 94dce3e
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions mockresponses.go
Expand Up @@ -1098,7 +1098,7 @@ func NewMockJoinGroupResponse(t TestReporter) *MockJoinGroupResponse {
}
}

func (m *MockJoinGroupResponse) For(reqBody versionedDecoder) encoder {
func (m *MockJoinGroupResponse) For(reqBody versionedDecoder) encoderWithHeader {
req := reqBody.(*JoinGroupRequest)
resp := &JoinGroupResponse{
Version: req.Version,
Expand Down Expand Up @@ -1162,7 +1162,7 @@ func NewMockLeaveGroupResponse(t TestReporter) *MockLeaveGroupResponse {
return &MockLeaveGroupResponse{t: t}
}

func (m *MockLeaveGroupResponse) For(reqBody versionedDecoder) encoder {
func (m *MockLeaveGroupResponse) For(reqBody versionedDecoder) encoderWithHeader {
resp := &LeaveGroupResponse{
Err: m.Err,
}
Expand All @@ -1185,7 +1185,7 @@ func NewMockSyncGroupResponse(t TestReporter) *MockSyncGroupResponse {
return &MockSyncGroupResponse{t: t}
}

func (m *MockSyncGroupResponse) For(reqBody versionedDecoder) encoder {
func (m *MockSyncGroupResponse) For(reqBody versionedDecoder) encoderWithHeader {
resp := &SyncGroupResponse{
Err: m.Err,
MemberAssignment: m.MemberAssignment,
Expand Down Expand Up @@ -1217,10 +1217,8 @@ func NewMockHeartbeatResponse(t TestReporter) *MockHeartbeatResponse {
return &MockHeartbeatResponse{t: t}
}

func (m *MockHeartbeatResponse) For(reqBody versionedDecoder) encoder {
resp := &HeartbeatResponse{
Err: m.Err,
}
func (m *MockHeartbeatResponse) For(reqBody versionedDecoder) encoderWithHeader {
resp := &HeartbeatResponse{}
return resp
}

Expand Down

0 comments on commit 94dce3e

Please sign in to comment.