Test case updates
[nonrtric.git] / test / mrstub / app / main.py
index fb6d674..0995428 100644 (file)
@@ -58,8 +58,8 @@ cntr_msg_responses_fetched=0
 # Request and response constants
 ORU_WRITE_URL="/events/unauthenticated.SEC_FAULT_OUTPUT"
 ORU_READ_URL="/events/unauthenticated.SEC_FAULT_OUTPUT/users/test/"
-AGENT_WRITE_URL="/events/A1-POLICY-AGENT-WRITE"
-AGENT_READ_URL="/events/A1-POLICY-AGENT-READ/users/policy-agent"
+A1PMS_WRITE_URL="/events/A1-POLICY-AGENT-WRITE"
+A1PMS_READ_URL="/events/A1-POLICY-AGENT-READ/users/policy-agent"
 APP_WRITE_URL="/send-request"
 APP_READ_URL="/receive-response"
 MIME_TEXT="text/plain"
@@ -69,11 +69,13 @@ SERVER_ERROR="Server error :"
 
 topic_write=""
 topic_read=""
+generic_topics_upload_baseurl=""
 
 uploader_thread=None
 downloader_thread=None
+generic_uploader_thread=None
 
-# Function to download messages from dmaap
+# Function to upload A1PMS messages to dmaap
 def dmaap_uploader():
     global msg_requests
     global cntr_msg_requests_fetched
@@ -107,7 +109,7 @@ def dmaap_uploader():
         sleep(0.01)
 
 
-# Function to upload messages to dmaap
+# Function to download A1PMS messages from dmaap
 def dmaap_downloader():
     global msg_responses
     global cntr_msg_responses_submitted
@@ -150,6 +152,49 @@ def dmaap_downloader():
         except Exception as e:
             sleep(1)
 
+# Function to upload generic messages to dmaap
+def dmaap_generic_uploader():
+    global msg_requests
+    global cntr_msg_requests_fetched
+
+    print("Starting generic uploader")
+
+    headers_json = {'Content-type': 'application/json', 'Accept': '*/*'}
+    headers_text = {'Content-type': 'text/plain', 'Accept': '*/*'}
+
+    while True:
+        if (len(generic_messages)):
+            keys_copy = list(generic_messages.keys())
+            for topicname in keys_copy:    #topicname contains the path of the topics, eg. "/event/<topic>"
+                topic_queue=generic_messages[topicname]
+                if (len(topic_queue)>0):
+                    if (topicname.endswith(".text")):
+                        msg=topic_queue[0]
+                        headers=headers_text
+                    else:
+                        msg=topic_queue[0]
+                        msg=json.dumps(msg)
+                        headers=headers_json
+                    url=generic_topics_upload_baseurl+topicname
+                    print("Sending to dmaap : "+ url)
+                    print("Sending to dmaap : "+ msg)
+                    print("Sending to dmaap : "+ str(headers))
+                    try:
+                        resp=requests.post(url, data=msg, headers=headers, timeout=10)
+                        if (resp.status_code<199 & resp.status_code > 299):
+                            print("Failed, response code: " + str(resp.status_code))
+                            sleep(1)
+                        else:
+                            print("Dmaap response code: " + str(resp.status_code))
+                            print("Dmaap response text: " + str(resp.text))
+                            with lock:
+                                topic_queue.pop(0)
+                                cntr_msg_requests_fetched += 1
+                    except Exception as e:
+                        print("Failed, exception: "+ str(e))
+                        sleep(1)
+        sleep(0.01)
+
 #I'm alive function
 @app.route('/',
     methods=['GET'])
@@ -157,7 +202,7 @@ def index():
     return 'OK', 200
 
 
-# Helper function to create a Dmaap request message
+# Helper function to create a Dmaap A1PMS request message
 # args : <GET|PUT|DELETE> <correlation-id> <json-string-payload - may be None> <url>
 # response: json formatted string of a complete Dmaap message
 def create_message(operation, correlation_id, payload, url):
@@ -171,7 +216,7 @@ def create_message(operation, correlation_id, payload, url):
 
 ### MR-stub interface, for MR control
 
-# Send a message to MR
+# Send a A1PMS message to MR
 # URI and parameters (PUT or POST): /send-request?operation=<GET|PUT|POST|DELETE>&url=<url>
 # response: <correlation-id> (http 200) o4 400 for parameter error or 500 for other errors
 @app.route(APP_WRITE_URL,
@@ -199,7 +244,7 @@ def sendrequest():
             print(APP_WRITE_URL+" operation="+oper+" url="+url)
             correlation_id=str(time.time_ns())
             payload=None
-            if (oper == "PUT") and (request.json is not None):
+            if (oper == "PUT") and len(request.data) > 0:
                 payload=json.dumps(request.json)
 
             msg=create_message(oper, correlation_id, payload, url)
@@ -212,7 +257,7 @@ def sendrequest():
             print(APP_WRITE_URL+"-"+CAUGHT_EXCEPTION+" "+str(e) + " "+traceback.format_exc())
             return Response(SERVER_ERROR+" "+str(e), status=500, mimetype=MIME_TEXT)
 
-# Receive a message response for MR for the included correlation id
+# Receive a A1PMS message response for MR for the included correlation id
 # URI and parameter, (GET): /receive-response?correlationid=<correlation-id>
 # response: <json-array of 1 response> 200 or empty 204 or other errors 500
 @app.route(APP_READ_URL,
@@ -243,10 +288,10 @@ def receiveresponse():
 
 ### Dmaap interface ###
 
-# Read messages stream. URI according to agent configuration.
+# Read A1PMS messages stream. URI according to agent configuration.
 # URI, (GET): /events/A1-POLICY-AGENT-READ/users/policy-agent
 # response: 200 <json array of request messages>, or 500 for other errors
-@app.route(AGENT_READ_URL,
+@app.route(A1PMS_READ_URL,
     methods=['GET'])
 def events_read():
     global msg_requests
@@ -288,10 +333,10 @@ def events_read():
                         cntr_msg_requests_fetched += 1
                         cntr=cntr+1
                     msgs='['+msgs+']'
-                    print(AGENT_READ_URL+" MSGs: "+json.dumps(json.loads(msgs), indent=2))
+                    print(A1PMS_READ_URL+" MSGs: "+json.dumps(json.loads(msgs), indent=2))
                     return Response(msgs, status=200, mimetype=MIME_JSON)
                 except Exception as e:
-                    print(AGENT_READ_URL+"-"+CAUGHT_EXCEPTION+" "+str(e) + " "+traceback.format_exc())
+                    print(A1PMS_READ_URL+"-"+CAUGHT_EXCEPTION+" "+str(e) + " "+traceback.format_exc())
                     return Response(SERVER_ERROR+" "+str(e), status=500, mimetype=MIME_TEXT)
         sleep(0.025) # sleep 25 milliseconds
         current_time=int(round(time.time() * 1000))
@@ -299,10 +344,10 @@ def events_read():
     print("timeout: "+str(timeout)+", start_time: "+str(start_time)+", current_time: "+str(current_time))
     return Response("[]", status=200, mimetype=MIME_JSON)
 
-# Write messages stream. URI according to agent configuration.
+# Write A1PMS messages stream. URI according to agent configuration.
 # URI and payload, (PUT or POST): /events/A1-POLICY-AGENT-WRITE <json array of response messages>
 # response: OK 200 or 400 for missing json parameters, 500 for other errors
-@app.route(AGENT_WRITE_URL,
+@app.route(A1PMS_WRITE_URL,
     methods=['PUT','POST'])
 def events_write():
     global msg_responses
@@ -312,10 +357,10 @@ def events_write():
         return Response('Url not available when running as mrstub frontend', status=404, mimetype=MIME_TEXT)
 
     with lock:
-        print("AGENT_WRITE_URL lock")
+        print("A1PMS_WRITE_URL lock")
         try:
             answer=request.json
-            print(AGENT_WRITE_URL+ " json=" + json.dumps(answer, indent=2))
+            print(A1PMS_WRITE_URL+ " json=" + json.dumps(answer, indent=2))
             if isinstance(answer, dict):
                 #Create a an array if the answer is a dict (single message)
                 answer_list=[]
@@ -325,15 +370,15 @@ def events_write():
             for item in answer:
                 cid=item['correlationId']
                 if (cid is None):
-                    print(AGENT_WRITE_URL+" parameter 'correlatonid' missing")
+                    print(A1PMS_WRITE_URL+" parameter 'correlatonid' missing")
                     return Response('Parameter <correlationid> missing in json', status=400, mimetype=MIME_TEXT)
                 msg=item['message']
                 if (msg is None):
-                    print(AGENT_WRITE_URL+" parameter 'msgs' missing")
+                    print(A1PMS_WRITE_URL+" parameter 'msgs' missing")
                     return Response('Parameter >message> missing in json', status=400, mimetype=MIME_TEXT)
                 status=item['status']
                 if (status is None):
-                    print(AGENT_WRITE_URL+" parameter 'status' missing")
+                    print(A1PMS_WRITE_URL+" parameter 'status' missing")
                     return Response('Parameter <status> missing in json', status=400, mimetype=MIME_TEXT)
                 if isinstance(msg, list) or isinstance(msg, dict):
                     msg_str=json.dumps(msg)+status[0:3]
@@ -341,9 +386,9 @@ def events_write():
                     msg_str=msg+status[0:3]
                 msg_responses[cid]=msg_str
                 cntr_msg_responses_submitted += 1
-                print(AGENT_WRITE_URL+ " msg+status (correlationid="+cid+") :" + str(msg_str))
+                print(A1PMS_WRITE_URL+ " msg+status (correlationid="+cid+") :" + str(msg_str))
         except Exception as e:
-            print(AGENT_WRITE_URL+"-"+CAUGHT_EXCEPTION+" "+str(e) + " "+traceback.format_exc())
+            print(A1PMS_WRITE_URL+"-"+CAUGHT_EXCEPTION+" "+str(e) + " "+traceback.format_exc())
             return Response('{"message": "' + SERVER_ERROR + ' ' + str(e) + '","status":"500"}', status=200, mimetype=MIME_JSON)
 
         return Response('{}', status=200, mimetype=MIME_JSON)
@@ -367,10 +412,10 @@ def oru_read():
         return Response(json.dumps(res), status=200, mimetype=MIME_JSON)
     return Response("[]", status=200, mimetype=MIME_JSON)
 
-# Generic POST/PUT catching all urls starting with /events/<topic>.
+# Generic POST catching all urls starting with /events/<topic>.
 # Writes the message in a que for that topic
 @app.route("/events/<path>",
-    methods=['PUT','POST'])
+    methods=['POST'])
 def generic_write(path):
     global generic_messages
     global cntr_msg_responses_submitted
@@ -378,8 +423,12 @@ def generic_write(path):
     write_method=str(request.method)
     with lock:
         try:
-            payload=request.json
-            print(write_method+" on "+urlkey+" json=" + json.dumps(payload))
+            if (urlkey.endswith(".text")):
+                payload=str(request.data.decode('UTF-8'))
+                print(write_method+" on "+urlkey+" text=" + payload)
+            else:
+                payload=request.json
+                print(write_method+" on "+urlkey+" json=" + json.dumps(payload))
             topicmsgs=[]
             if (urlkey in generic_messages.keys()):
                 topicmsgs=generic_messages[urlkey]
@@ -407,6 +456,9 @@ def generic_read(path):
     global generic_messages
     global cntr_msg_requests_fetched
 
+    if generic_topics_upload_baseurl:
+        return Response('Url not available when running as mrstub frontend', status=404, mimetype=MIME_TEXT)
+
     urlpath="/events/"+str(path)
     urlkey="/events/"+str(path).split("/")[0] #Extract topic
     print("GET on topic"+urlkey)
@@ -530,7 +582,14 @@ if os.getenv("TOPIC_READ") is not None:
         uploader_thread=Thread(target=dmaap_uploader)
         uploader_thread.start()
 
-else:
+if 'GENERIC_TOPICS_UPLOAD_BASEURL' in os.environ:
+    print("GENERIC_TOPICS_UPLOAD_BASEURL:"+os.environ['GENERIC_TOPICS_UPLOAD_BASEURL'])
+    generic_topics_upload_baseurl=os.environ['GENERIC_TOPICS_UPLOAD_BASEURL']
+    if generic_topics_upload_baseurl and generic_uploader_thread is None:
+        generic_uploader_thread=Thread(target=dmaap_generic_uploader)
+        generic_uploader_thread.start()
+
+if os.getenv("TOPIC_READ") is None or os.environ['GENERIC_TOPICS_UPLOAD_BASEURL'] is None:
     print("No env variables - OK")
 
 if __name__ == "__main__":