Add more tests

Author: nanomad

tests/messages/xml/test_map.py

@@ -2,9 +2,9 @@
 
 import pytest
 
-from deebot_client.events import MinorMapEvent
+from deebot_client.events import MapTraceEvent, MinorMapEvent
 from deebot_client.message import HandlingState
-from deebot_client.messages.xml import MapP
+from deebot_client.messages.xml import MapP, Trace
 from tests.messages import assert_message, assert_message_failure
 
 
@@ -29,3 +29,28 @@ def test_MapP(pid: int, data: str) -> None:
 )
 def test_MapP_error(xml_message: str) -> None:
     assert_message_failure(MapP, xml_message, HandlingState.ANALYSE_LOGGED)
+
+
+@pytest.mark.parametrize(("tf", "tt", "tr"), [(13, 42, "base64data")])
+def test_Trace(tf: int, tt: int, tr: str) -> None:
+    xml_message = f"<ctl td='trace' trid='631369' tf='{tf}' tt='{tt}' tr='{tr}'/>"
+    assert_message(
+        Trace,
+        xml_message,
+        MapTraceEvent(start=tf, total=tt, data=tr),
+    )
+
+
+@pytest.mark.parametrize(
+    "xml_message",
+    {
+        "<ctl td='trace' trid='631369' tt='17' tr='XQAABAAKAAAAAG0/wEAAA2cAS5AAAA=='/>",
+        "<ctl td='trace' trid='631369' tf='XXX' tt='17' tr='XQAABAAKAAAAAG0/wEAAA2cAS5AAAA=='/>",
+        "<ctl td='trace' trid='631369' tf='16' tr='XQAABAAKAAAAAG0/wEAAA2cAS5AAAA=='/>",
+        "<ctl td='trace' trid='631369' tf='16' tt='XXX' tr='XQAABAAKAAAAAG0/wEAAA2cAS5AAAA=='/>",
+        "<ctl td='trace' trid='631369' tf='16' tt='16' />",
+        "<ctl td='trace' trid='631369' />",
+    },
+)
+def test_Trace_error(xml_message: str) -> None:
+    assert_message_failure(MapP, xml_message, HandlingState.ANALYSE_LOGGED)