| 403 | } |
| 404 | |
| 405 | func (m *mockLeaveGroupCapture) For(reqBody versionedDecoder) encoderWithHeader { |
| 406 | req := reqBody.(*LeaveGroupRequest) |
| 407 | m.mu.Lock() |
| 408 | for _, member := range req.Members { |
| 409 | m.reasons = append(m.reasons, member.Reason) |
| 410 | } |
| 411 | select { |
| 412 | case <-m.captured: |
| 413 | default: |
| 414 | close(m.captured) |
| 415 | } |
| 416 | m.mu.Unlock() |
| 417 | return m.inner.For(reqBody) |
| 418 | } |
| 419 | |
| 420 | // mockHeartbeatRebalanceResponse returns ErrNoError for the first N heartbeats, |
| 421 | // then ErrRebalanceInProgress for all subsequent heartbeats. |