X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=test%2Fcommon%2Fcreate_policies_process.py;h=480d5cbdf2f80196a2a1b01ece280ae975c8f10a;hb=1b732d17463fad74721391b3a87a2a12172da63c;hp=8d37678f0d8347c8530dce31e08d4d8d83327c75;hpb=8d10bf1fee8093ec4c2a3d3530fded002daf83b6;p=nonrtric.git diff --git a/test/common/create_policies_process.py b/test/common/create_policies_process.py index 8d37678f..480d5cbd 100644 --- a/test/common/create_policies_process.py +++ b/test/common/create_policies_process.py @@ -24,29 +24,62 @@ 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 10 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id") - sys.exit() + 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]) - baseurl=sys.argv[2] - ric_base=sys.argv[3] + baseurl=str(sys.argv[2]) + ric_base=str(sys.argv[3]) num_rics=int(sys.argv[4]) - uuid=sys.argv[5] + uuid=str(sys.argv[5]) start=int(sys.argv[6]) - templatepath=sys.argv[7] - count=int(sys.argv[8]) - pids=int(sys.argv[9]) - pid_id=int(sys.argv[10]) + httpproxy="NOPROXY" + if ("/v2/" in baseurl): + 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() + + serv=str(sys.argv[7]) + pt=str(sys.argv[8]) + trans=str(sys.argv[9]) + noti=str(sys.argv[10]) + templatepath=str(sys.argv[11]) + 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) != 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() + templatepath=str(sys.argv[7]) + 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="" @@ -56,28 +89,80 @@ 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) - url=baseurl+"&id="+uuid+str(i)+"&ric="+str(ric) - try: - headers = {'Content-type': 'application/json'} - resp=requests.put(url, json.dumps(json.loads(payload)), 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: "+responsecode+", got: None") - sys.exit() - if (resp.status_code != responsecode): - print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+responsecode+", got: "+str(resp.status_code)) - sys.exit() - - print("0") + + 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: + 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: print("1"+str(e)) + traceback.print_exc() + print(str(data_out)) sys.exit() \ No newline at end of file