Add MQTTClient.PublishSynced (#21)

wirenboard · Dec 28, 2024 · 54c8b28 · 54c8b28
1 parent 6bd9cd9
commit 54c8b28
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/mqtt.go b/mqtt.go
@@ -23,6 +23,7 @@ type MQTTClient interface {
 	Start()
 	Stop()
 	Publish(message MQTTMessage)
+	PublishSynced(message MQTTMessage)
 	Subscribe(callback MQTTMessageHandler, topics ...string)
 	Unsubscribe(topics ...string)
 }

diff --git a/testutils/fake_mqtt.go b/testutils/fake_mqtt.go
@@ -296,6 +296,11 @@ func (client *FakeMQTTClient) Publish(message wbgong.MQTTMessage) {
 	client.broker.Publish(client.id, message)
 }
 
+func (client *FakeMQTTClient) PublishSynced(message wbgong.MQTTMessage) {
+	client.ensureStarted()
+	client.broker.Publish(client.id, message)
+}
+
 func (client *FakeMQTTClient) Subscribe(callback wbgong.MQTTMessageHandler, topics ...string) {
 	client.Lock()
 	defer client.Unlock()