From c585e24f4f1bd290186025492fa346f61a0dfa2b Mon Sep 17 00:00:00 2001
From: "Pascal S. de Kloe" <pascal@quies.net>
Date: Thu, 18 Feb 2021 22:51:44 +0100
Subject: [PATCH] Add ReadSlices mock.

---
 mqtttest/mqtttest.go | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/mqtttest/mqtttest.go b/mqtttest/mqtttest.go
index e1a47cc..2df7765 100644
--- a/mqtttest/mqtttest.go
+++ b/mqtttest/mqtttest.go
@@ -26,6 +26,29 @@ type Transfer struct {
 	Err     error  // result
 }
 
+// NewReadSlicesMock returns a new mock for mqtt.Client ReadSlices, which
+// returns the Transfers in order of appearance.
+func NewReadSlicesMock(t testing.TB, want ...Transfer) func() (message, topic []byte, err error) {
+	t.Helper()
+
+	var wantIndex uint64
+
+	return func() (message, topic []byte, err error) {
+		i := atomic.AddUint64(&wantIndex, 1) - 1
+		if i >= uint64(len(want)) {
+			err = errors.New("unwanted MQTT ReadSlices")
+			t.Error(err)
+			return
+		}
+
+		// use copies to prevent some hard to trace issues
+		message = make([]byte, len(want[i].Message))
+		copy(message, want[i].Message)
+		topic = []byte(want[i].Topic)
+		return message, topic, want[i].Err
+	}
+}
+
 // NewPublishMock returns a new mock for mqtt.Client Publish, which compares the
 // invocation with want in order of appearance.
 func NewPublishMock(t testing.TB, want ...Transfer) func(message []byte, topic string) error {