X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=test%2Fcommon%2Fcreate_policies_process.py;h=480d5cbdf2f80196a2a1b01ece280ae975c8f10a;hb=4dd321bda70fab4987c1ea7aa0dffc5e1eb74252;hp=d1bfd6f44d0061bf490b2814b2ff17f5726fc389;hpb=8489de0f43f28c5cd62206b4e242c0308f9864b9;p=nonrtric.git diff --git a/test/common/create_policies_process.py b/test/common/create_policies_process.py index d1bfd6f4..480d5cbd 100644 --- a/test/common/create_policies_process.py +++ b/test/common/create_policies_process.py @@ -24,19 +24,20 @@ import json import sys import requests import traceback +from time import sleep # disable warning about unverified https requests from requests.packages import urllib3 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) -#arg responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id +#arg responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id proxy data_out="" url_out="" try: - if len(sys.argv) < 11: - print("1Expected 11/14 args, got "+str(len(sys.argv)-1)) + if len(sys.argv) < 12: + print("1Expected 12/15 args, got "+str(len(sys.argv)-1)) print (sys.argv[1:]) sys.exit() responsecode=int(sys.argv[1]) @@ -45,9 +46,10 @@ try: num_rics=int(sys.argv[4]) uuid=str(sys.argv[5]) start=int(sys.argv[6]) + httpproxy="NOPROXY" if ("/v2/" in baseurl): - if len(sys.argv) != 15: - print("1Expected 14 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid service type transient notification-url templatepath count pids pid_id") + if len(sys.argv) != 16: + print("1Expected 15 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid service type transient notification-url templatepath count pids pid_id proxy") print (sys.argv[1:]) sys.exit() @@ -59,9 +61,10 @@ try: count=int(sys.argv[12]) pids=int(sys.argv[13]) pid_id=int(sys.argv[14]) + httpproxy=str(sys.argv[15]) else: - if len(sys.argv) != 11: - print("1Expected 10 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id") + if len(sys.argv) != 12: + print("1Expected 11 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id proxy") print (sys.argv[1:]) sys.exit() @@ -69,7 +72,14 @@ try: count=int(sys.argv[8]) pids=int(sys.argv[9]) pid_id=int(sys.argv[10]) - + httpproxy=str(sys.argv[11]) + + proxydict=None + if httpproxy != "NOPROXY": + proxydict = { + "http" : httpproxy, + "https" : httpproxy + } if uuid == "NOUUID": uuid="" @@ -79,52 +89,76 @@ try: start=start stop=count*num_rics+start + http_retry_count=0 + connect_retry_count=0 + for i in range(start,stop): if (i%pids == (pid_id-1)): payload=template.replace("XXX",str(i)) ric_id=(i%num_rics)+1 ric=ric_base+str(ric_id) - try: - headers = {'Content-type': 'application/json'} - if ("/v2/" in baseurl): - url=baseurl - - data={} - data["ric_id"]=ric - data["policy_id"]=uuid+str(i) - data["service_id"]=serv - if (trans != "NOTRANSIENT"): - data["transient"]=trans - if (pt != "NOTYPE"): - data["policy_type_id"]=pt + connect_ok=False + retry_cnt=5 + while(retry_cnt>0): + try: + headers = {'Content-type': 'application/json'} + if ("/v2/" in baseurl): + url=baseurl + + data={} + data["ric_id"]=ric + data["policy_id"]=uuid+str(i) + data["service_id"]=serv + if (trans != "NOTRANSIENT"): + data["transient"]=trans + if (pt != "NOTYPE"): + data["policytype_id"]=pt + else: + data["policytype_id"]="" + if (noti != "NOURL"): + data["status_notification_uri"]=noti + data["policy_data"]=json.loads(payload) + + url_out=url + data_out=json.dumps(data) + else: + url=baseurl+"&id="+uuid+str(i)+"&ric="+str(ric) + url_out=url + data_out=json.dumps(json.loads(payload)) + if proxydict is None: + resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90) else: - data["policy_type_id"]="" - if (noti != "NOURL"): - data["status_notification_uri"]=noti - data["policy_data"]=json.loads(payload) - - url_out=url - data_out=json.dumps(data) - resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90) - else: - url=baseurl+"&id="+uuid+str(i)+"&ric="+str(ric) - url_out=url - data_out=json.dumps(json.loads(payload)) - resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90) - except Exception as e1: - print("1Put failed for id:"+uuid+str(i)+ ", "+str(e1) + " "+traceback.format_exc()) - sys.exit() - if (resp.status_code == None): - print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: None") - sys.exit() - if (resp.status_code != responsecode): - print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: "+str(resp.status_code)) - print(url_out) - print(str(data_out)) - sys.exit() - - print("0") + resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90, proxies=proxydict) + connect_ok=True + except Exception as e1: + if (retry_cnt > 1): + sleep(0.1) + retry_cnt -= 1 + connect_retry_count += 1 + else: + print("1Put failed for id:"+uuid+str(i)+ ", "+str(e1) + " "+traceback.format_exc()) + sys.exit() + + if (connect_ok == True): + if (resp.status_code == None): + print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: None") + sys.exit() + + if (resp.status_code != responsecode): + if (resp.status_code >= 500) and (http_retry_count < 600 ) and (retry_cnt > 1): + sleep(0.1) + retry_cnt -= 1 + http_retry_count += 1 + else: + print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: "+str(resp.status_code)) + print(url_out) + print(str(data_out)) + sys.exit() + else: + retry_cnt=-1 + + print("0 http retries:"+str(http_retry_count) + ", connect retries: "+str(connect_retry_count)) sys.exit() except Exception as e: