Fix tests

reportportal · Jan 8, 2025 · f920127 · f920127
1 parent 1dc9349
commit f920127
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/robotframework_reportportal/listener.py b/robotframework_reportportal/listener.py
@@ -150,7 +150,9 @@ def _build_msg_struct(self, message: Dict[str, Any]) -> LogMessage:
         else:
             msg = LogMessage(message["message"])
             msg.level = message["level"]
-        msg.item_id = self.current_item.rp_item_id
+        current_item = self.current_item
+        if current_item:
+            msg.item_id = current_item.rp_item_id
 
         message_str = msg.message
         if is_binary(message_str):

diff --git a/tests/unit/test_model.py b/tests/unit/test_model.py
@@ -12,6 +12,8 @@
 limitations under the License.
 """
 
+from unittest import mock
+
 import pytest
 
 from robotframework_reportportal.model import Keyword
@@ -28,6 +30,8 @@
 )
 def test_keyword_get_type(kwd_attributes, self_type, parent_type, expected):
     """Test for the get_type() method of the Keyword model."""
-    kwd = Keyword(name="Test keyword", robot_attributes=kwd_attributes, parent_type=parent_type)
+    parent = mock.Mock()
+    parent.type = parent_type
+    kwd = Keyword(name="Test keyword", robot_attributes=kwd_attributes, parent=parent)
     kwd.keyword_type = self_type
     assert kwd.get_type() == expected