X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=src%2Fbindings%2Frmr-python%2Frmr%2Frmr_mocks%2Frmr_mocks.py;h=315536f363ac5a705598a7522ecfa93f9e96b358;hb=refs%2Fchanges%2F92%2F792%2F3;hp=395bfb39cd6658b5a3e9c8f9b0d122c62864a6d2;hpb=46ae0f44cd0ea713601f16adc5ea709e607eb2e8;p=ric-plt%2Flib%2Frmr.git

diff --git a/src/bindings/rmr-python/rmr/rmr_mocks/rmr_mocks.py b/src/bindings/rmr-python/rmr/rmr_mocks/rmr_mocks.py
index 395bfb3..315536f 100644
--- a/src/bindings/rmr-python/rmr/rmr_mocks/rmr_mocks.py
+++ b/src/bindings/rmr-python/rmr/rmr_mocks/rmr_mocks.py
@@ -34,6 +34,10 @@ def rcv_mock_generator(msg_payload, msg_type, msg_state, jsonb, timeout=0):
         sbuf.contents.payload = payload
         sbuf.contents.len = len(payload)
         sbuf.contents.state = msg_state
+        if msg_state != 0:  # set something in transport state if 'error'
+            sbuf.contents.tp_state = 99
+        else:
+            sbuf.contents.tp_state = 0
         return sbuf
 
     return f
@@ -48,6 +52,10 @@ def send_mock_generator(msg_state):
 
     def f(_unused, sbuf):
         sbuf.contents.state = msg_state
+        if msg_state != 0:  # set something in transport state if 'error'
+            sbuf.contents.tp_state = 99
+        else:
+            sbuf.contents.tp_state = 0
         return sbuf
 
     return f
@@ -63,6 +71,7 @@ class _Sbuf_Contents:
         self.payload = ""
         self.xaction = uuid.uuid1().hex.encode("utf-8")
         self.sub_id = 0
+        self.tp_state = 0
 
     def __str__(self):
         return str(
@@ -73,6 +82,7 @@ class _Sbuf_Contents:
                 "payload": self.payload,
                 "xaction": self.xaction,
                 "sub_id": self.sub_id,
+                "tp_state": self.tp_state,
             }
         )
 
@@ -116,5 +126,5 @@ def patch_rmr(monkeypatch):
     monkeypatch.setattr("rmr.rmr.generate_and_set_transaction_id", fake_generate_and_set_transaction_id)
     monkeypatch.setattr("rmr.rmr.get_payload", fake_get_payload)
     monkeypatch.setattr("rmr.rmr.get_src", fake_get_src)
-    monkeypatch.setattr("rmr.rmr.get_meid", fake_get_meid)
+    monkeypatch.setattr("rmr.rmr.rmr_get_meid", fake_get_meid)
     monkeypatch.setattr("rmr.rmr.rmr_payload_size", fake_rmr_payload_size)