diff options
| -rw-r--r-- | adapter/mock/mock_vpp_adapter.go | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/adapter/mock/mock_vpp_adapter.go b/adapter/mock/mock_vpp_adapter.go index 9dca2ba..c05148d 100644 --- a/adapter/mock/mock_vpp_adapter.go +++ b/adapter/mock/mock_vpp_adapter.go @@ -372,6 +372,16 @@ func (a *VppAdapter) MockReplyHandler(replyHandler ReplyHandler) { a.mode = useReplyHandlers } +// MockClearReplyHanders clears all reply handlers that were registered +// Will also set the mode to useReplyHandlers +func (a *VppAdapter) MockClearReplyHandlers () { + a.repliesLock.Lock() + defer a.repliesLock.Unlock() + + a.replyHandlers = a.replyHandlers[:0] + a.mode = useReplyHandlers +} + func setSeqNum(context uint32, seqNum uint16) (newContext uint32) { context &= 0xffff0000 context |= uint32(seqNum) |