From 120655a1bbf78778635008214dc6fdd1c5c88f7b Mon Sep 17 00:00:00 2001
From: DasSkelett <dasskelett@gmail.com>
Date: Sun, 28 Aug 2022 15:24:10 +0000
Subject: [PATCH] Add more tests for worker/mqtt.py

---
 wgkex/worker/mqtt_test.py | 42 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/wgkex/worker/mqtt_test.py b/wgkex/worker/mqtt_test.py
index c70151c..c317cf4 100644
--- a/wgkex/worker/mqtt_test.py
+++ b/wgkex/worker/mqtt_test.py
@@ -1,8 +1,13 @@
 """Unit tests for mqtt.py"""
+import socket
 import threading
 import unittest
+from time import sleep
+
 import mock
+import paho.mqtt.client
 
+from wgkex.common.mqtt import TOPIC_CONNECTED_PEERS
 from wgkex.worker import mqtt
 
 
@@ -90,6 +95,43 @@ def test_on_message_wireguard_fails_no_domain(self, config_mock, link_mock):
         with self.assertRaises(ValueError):
             mqtt.on_message_wireguard(None, None, mqtt_msg)
 
+    @mock.patch.object(mqtt, "get_config")
+    @mock.patch.object(mqtt, "get_connected_peers_count")
+    def test_publish_metrics_loop_success(self, conn_peers_mock, config_mock):
+        config_mock.return_value = _get_config_mock()
+        conn_peers_mock.return_value = 20
+        mqtt_client = mock.MagicMock(spec=paho.mqtt.client.Client)
+
+        ee = threading.Event()
+        thread = threading.Thread(
+            target=mqtt.publish_metrics_loop,
+            args=(ee, mqtt_client, "_ffmuc_domain.one"),
+        )
+        thread.start()
+
+        i = 0
+        while i < 20 and not mqtt_client.publish.called:
+            i += 1
+            sleep(0.1)
+
+        conn_peers_mock.assert_called_with("wg-domain.one")
+        mqtt_client.publish.assert_called_with(
+            TOPIC_CONNECTED_PEERS.format(
+                domain="_ffmuc_domain.one", worker=socket.gethostname()
+            ),
+            20,
+            retain=True,
+        )
+
+        ee.set()
+
+        i = 0
+        while i < 20 and thread.is_alive():
+            i += 1
+            sleep(0.1)
+
+        self.assertFalse(thread.is_alive())
+
 
 if __name__ == "__main__":
     unittest.main()