change return type of MockJoinGroupResponse.For & MockLeaveGroupRespo…

…nse.For to encoderWithHeader
IBM · Jul 12, 2020 · ebbd858 · ebbd858
1 parent ac5ca79
commit ebbd858
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mockresponses.go b/mockresponses.go
@@ -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,
@@ -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,
 	}