New tests added

tb_device_mqtt.py

@@ -1,11 +1,11 @@
 # Copyright 2024. ThingsBoard
-# 
+#
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
-# 
+#
 #  http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -125,20 +125,21 @@ class TBPublishInfo:
     def __init__(self, message_info):
         self.message_info = message_info
 
+
     # pylint: disable=invalid-name
     def rc(self):
         if isinstance(self.message_info, list):
             for info in self.message_info:
-                if isinstance(info.rc, ReasonCodes):
+                if isinstance(info.rc, ReasonCodes) or hasattr(info.rc, 'value'):
                     if info.rc.value == 0:
                         continue
-                    return info.rc
+                    return info.rc.value
                 else:
                     if info.rc != 0:
                         return info.rc
             return self.TB_ERR_SUCCESS
         else:
-            if isinstance(self.message_info.rc, ReasonCodes):
+            if isinstance(self.message_info.rc, ReasonCodes) or hasattr(self.message_info.rc, 'value'):
                 return self.message_info.rc.value
             return self.message_info.rc
 
@@ -239,6 +240,11 @@ def set_limit(self, rate_limit, percentage=80):
             old_rate_limit_dict = deepcopy(self._rate_limit_dict)
             self._rate_limit_dict = {}
             self.percentage = percentage if percentage != 0 else self.percentage
+            if rate_limit.strip() == "0:0,":
+                self._rate_limit_dict.clear()
+                self._no_limit = True
+                log.debug("Rate limit set to NO_LIMIT from '0:0,' directive.")
+                return
             rate_configs = rate_limit.split(";")
             if "," in rate_limit:
                 rate_configs = rate_limit.split(",")
@@ -673,6 +679,7 @@ def send_rpc_reply(self, req_id, resp, quality_of_service=None, wait_for_publish
         info = self._publish_data(resp, RPC_RESPONSE_TOPIC + req_id, quality_of_service)
         if wait_for_publish:
             info.get()
+        return info
 
     def send_rpc_call(self, method, params, callback):
         """Send RPC call to ThingsBoard. The callback will be called when the response is received."""

tests/count_data_points_message_tests.py

@@ -0,0 +1,78 @@
+# Copyright 2025. ThingsBoard
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+from tb_device_mqtt import TBDeviceMqttClient
+
+
+class TestCountDataPointsInMessage(unittest.TestCase):
+    def test_simple_dict_no_device(self):
+        data = {
+            "ts": 123456789,
+            "values": {
+                "temp": 22.5,
+                "humidity": 55
+            }
+        }
+        result = TBDeviceMqttClient._count_datapoints_in_message(data)
+        self.assertEqual(result, 2)
+
+    def test_list_of_dict_no_device(self):
+        data = [
+            {"ts": 123456789, "values": {"temp": 22.5, "humidity": 55}},
+            {"ts": 123456799, "values": {"light": 100, "pressure": 760}}
+        ]
+        result = TBDeviceMqttClient._count_datapoints_in_message(data)
+        self.assertEqual(result, 4)
+
+    def test_with_device_dict_inside(self):
+        data = {
+            "MyDevice": {
+                "ts": 123456789,
+                "values": {"temp": 22.5, "humidity": 55}
+            },
+            "OtherKey": "some_value"
+        }
+        result = TBDeviceMqttClient._count_datapoints_in_message(data, device="MyDevice")
+        self.assertEqual(result, 2)
+
+    def test_with_device_list_inside(self):
+        data = {
+            "Sensor": [
+                {"ts": 1, "values": {"v1": 10}},
+                {"ts": 2, "values": {"v2": 20, "v3": 30}}
+            ]
+        }
+        result = TBDeviceMqttClient._count_datapoints_in_message(data, device="Sensor")
+        self.assertEqual(result, 3)
+
+    def test_empty_dict_no_device(self):
+        data = {}
+        result = TBDeviceMqttClient._count_datapoints_in_message(data)
+        self.assertEqual(result, 0)
+
+    def test_missing_device_key(self):
+
+        data = {"some_unrelated_key": 42}
+        result = TBDeviceMqttClient._count_datapoints_in_message(data, device="NotExistingDeviceKey")
+        self.assertEqual(result, 1)
+
+    def test_data_is_string_no_device(self):
+        data = "just a string"
+        result = TBDeviceMqttClient._count_datapoints_in_message(data)
+        self.assertEqual(result, 1)
+
+
+if __name__ == '__main__':
+    unittest.main()