X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=test%2Fusecases%2Flinkfailure%2Fsimulators%2Fmessage_generator.py;h=1b0676caac30b020f1c94c2d0015351b80eeb1d7;hb=d2c7d37bcf7e9193e2fafdff06802d233294a711;hp=f6b4e25bc005a095ec40224e6ebb9d0713194d1c;hpb=2772b3e52b693924df131a195859ff8e44458bb3;p=nonrtric.git

diff --git a/test/usecases/linkfailure/simulators/message_generator.py b/test/usecases/linkfailure/simulators/message_generator.py
index f6b4e25b..1b0676ca 100644
--- a/test/usecases/linkfailure/simulators/message_generator.py
+++ b/test/usecases/linkfailure/simulators/message_generator.py
@@ -16,16 +16,19 @@
 #  ============LICENSE_END=================================================
 #
 
+import json
+import os
+import random
 import requests
 import time
-import random
-import json
 
 # Randomly, between 0 and 10 seconds sends a "CUS Link Failure" alarm event to the Message Router. The ID of the O-RU is also
 # randomly generated between 0 and 9.
 # When the modulo of the ID is 1, a "heartbeat" message will also be sent to MR.
 
-MR_PATH = "http://localhost:3904/events/unauthenticated.SEC_FAULT_OUTPUT"
+mr_host = "http://localhost"
+mr_port = "3904"
+MR_PATH = "/events/unauthenticated.SEC_FAULT_OUTPUT"
 
 linkFailureMessage = {
     "event": {
@@ -87,17 +90,27 @@ heartBeatMessage = {
    }
  }
 
-while True:
-    random_time = int(10 * random.random())
-    if (random_time % 3 == 1):
-        print("Sent heart beat")
-        requests.post(MR_PATH, json=heartBeatMessage);
+if __name__ == "__main__":
+    if os.getenv("MR-HOST") is not None:
+        mr_host = os.getenv("MR-HOST")
+        print("Using MR Host from os: " + mr_host)
+    if os.getenv("MR-PORT") is not None:
+        mr_port = os.getenv("MR-PORT")
+        print("Using MR Port from os: " + mr_port)
+
+    mr_url = mr_host + ":" + mr_port + MR_PATH
+    print(mr_url)
+    while True:
+        random_time = int(10 * random.random())
+        if (random_time % 3 == 1):
+            print("Sent heart beat")
+            requests.post(mr_url, json=heartBeatMessage);
 
-    o_ru_id = "O-RAN-RU-0" + str(random_time)
-    print("Sent link failure for O-RAN-RU: " + o_ru_id)
-    msg_as_json = json.loads(json.dumps(linkFailureMessage))
-    msg_as_json["event"]["commonEventHeader"]["sourceName"] = o_ru_id
-    requests.post(MR_PATH, json=msg_as_json);
+        o_ru_id = "ERICSSON-O-RU-1122" + str(random_time)
+        print("Sent link failure for O-RAN-RU: " + o_ru_id)
+        msg_as_json = json.loads(json.dumps(linkFailureMessage))
+        msg_as_json["event"]["commonEventHeader"]["sourceName"] = o_ru_id
+        requests.post(mr_url, json=msg_as_json);
 
-    time.sleep(random_time)
+        time.sleep(random_time)