Mock Adapter: Switch back to handlers once the queue is empty.

Change-Id: Ifc85a80498521ef218a34072cce470bf32cead9c
Signed-off-by: Milan Lenco <milan.lenco@pantheon.tech>

1 files changed, 5 insertions, 0 deletions

diff --git a/adapter/mock/mock_adapter.go b/adapter/mock/mock_adapter.go
index 8c88030..e6b4bc4 100644
--- a/adapter/mock/mock_adapter.go
+++ b/adapter/mock/mock_adapter.go
@@ -274,6 +274,11 @@ func (a *VppAdapter) SendMsg(clientID uint32, data []byte) error {
 		}
 		if len(replies) > 0 {
 			replies = []api.Message{}
+			if len(replyHandlers) > 0 {
+				// Switch back to handlers once the queue is empty to revert back
+				// the fallthrough effect.
+				mode = useReplyHandlers
+			}
 			return nil
 		}