test/common/create_policies_process.py

   1 #  ============LICENSE_START===============================================
   2 #  Copyright (C) 2020 Nordix Foundation. All rights reserved.
   3 #  ========================================================================
   4 #  Licensed under the Apache License, Version 2.0 (the "License");
   5 #  you may not use this file except in compliance with the License.
   6 #  You may obtain a copy of the License at
   7 #
   8 #       http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 #  Unless required by applicable law or agreed to in writing, software
  11 #  distributed under the License is distributed on an "AS IS" BASIS,
  12 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 #  See the License for the specific language governing permissions and
  14 #  limitations under the License.
  15 #  ============LICENSE_END=================================================
  16 #
  17
  18 # This script create/update policies spread over a number rics
  19 # Intended for parallel processing
  20 # Returns a string with result, either "0" for ok, or "1<fault description>"
  21
  22 import os
  23 import json
  24 import sys
  25 import requests
  26 import traceback
  27 from time import sleep
  28
  29 # disable warning about unverified https requests
  30 from requests.packages import urllib3
  31
  32 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
  33
  34 #arg responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id proxy
  35 data_out=""
  36 url_out=""
  37 try:
  38
  39     if len(sys.argv) < 12:
  40         print("1Expected 12/15 args, got "+str(len(sys.argv)-1))
  41         print (sys.argv[1:])
  42         sys.exit()
  43     responsecode=int(sys.argv[1])
  44     baseurl=str(sys.argv[2])
  45     ric_base=str(sys.argv[3])
  46     num_rics=int(sys.argv[4])
  47     uuid=str(sys.argv[5])
  48     start=int(sys.argv[6])
  49     httpproxy="NOPROXY"
  50     if ("/v2/" in baseurl) or ("a1policymanagement/v1/" in baseurl):
  51         if len(sys.argv) != 16:
  52             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")
  53             print (sys.argv[1:])
  54             sys.exit()
  55
  56         serv=str(sys.argv[7])
  57         pt=str(sys.argv[8])
  58         trans=str(sys.argv[9])
  59         noti=str(sys.argv[10])
  60         templatepath=str(sys.argv[11])
  61         count=int(sys.argv[12])
  62         pids=int(sys.argv[13])
  63         pid_id=int(sys.argv[14])
  64         httpproxy=str(sys.argv[15])
  65     else:
  66         if len(sys.argv) != 12:
  67             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")
  68             print (sys.argv[1:])
  69             sys.exit()
  70
  71         templatepath=str(sys.argv[7])
  72         count=int(sys.argv[8])
  73         pids=int(sys.argv[9])
  74         pid_id=int(sys.argv[10])
  75         httpproxy=str(sys.argv[11])
  76
  77     proxydict=None
  78     if httpproxy != "NOPROXY":
  79         proxydict = {
  80             "http" : httpproxy,
  81             "https" : httpproxy
  82         }
  83     if uuid == "NOUUID":
  84         uuid=""
  85
  86     with open(templatepath, 'r') as file:
  87         template = file.read()
  88
  89         start=start
  90         stop=count*num_rics+start
  91
  92         http_retry_count=0
  93         connect_retry_count=0
  94
  95         for i in range(start,stop):
  96             if (i%pids == (pid_id-1)):
  97                 payload=template.replace("XXX",str(i))
  98                 ric_id=(i%num_rics)+1
  99                 ric=ric_base+str(ric_id)
 100
 101                 connect_ok=False
 102                 retry_cnt=5
 103                 while(retry_cnt>0):
 104                     try:
 105                         headers = {'Content-type': 'application/json'}
 106                         if ("/v2/" in baseurl):
 107                             url=baseurl
 108
 109                             data={}
 110                             data["ric_id"]=ric
 111                             data["policy_id"]=uuid+str(i)
 112                             data["service_id"]=serv
 113                             if (trans != "NOTRANSIENT"):
 114                                 data["transient"]=trans
 115                             if (pt != "NOTYPE"):
 116                                 data["policytype_id"]=pt
 117                             else:
 118                                 data["policytype_id"]=""
 119                             data["policy_data"]=json.loads(payload)
 120
 121                             url_out=url
 122                             data_out=json.dumps(data)
 123                             if proxydict is None:
 124                                 resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90)
 125                             else:
 126                                 resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90, proxies=proxydict)
 127
 128                         elif ("a1policymanagement/v1/" in baseurl):
 129                             url=baseurl
 130
 131                             data={}
 132                             data["nearRtRicId"]=ric
 133                             data["serviceId"]=serv
 134                             if (trans != "NOTRANSIENT"):
 135                                 data["transient"]=trans
 136                             if (pt != "NOTYPE"):
 137                                 data["policyTypeId"]=pt
 138                             else:
 139                                 data["policyTypeId"]=""
 140                             if (noti != "NOURL"):
 141                                 data["statusNotificationUri"]=noti
 142                             data["policyObject"]=json.loads(payload)
 143
 144                             url_out=url
 145                             data_out=json.dumps(data)
 146                             if proxydict is None:
 147                                 resp=requests.post(url, data_out, headers=headers, verify=False, timeout=90)
 148                             else:
 149                                 resp=requests.post(url, data_out, headers=headers, verify=False, timeout=90, proxies=proxydict)
 150                         else:
 151                             url=baseurl+"&id="+uuid+str(i)+"&ric="+str(ric)
 152                             url_out=url
 153                             data_out=json.dumps(json.loads(payload))
 154                             if proxydict is None:
 155                                 resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90)
 156                             else:
 157                                 resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90, proxies=proxydict)
 158                         connect_ok=True
 159                     except Exception as e1:
 160                         if (retry_cnt > 1):
 161                             sleep(0.1)
 162                             retry_cnt -= 1
 163                             connect_retry_count += 1
 164                         else:
 165                             print("1Put failed for id:"+uuid+str(i)+ ", "+str(e1) + " "+traceback.format_exc())
 166                             sys.exit()
 167
 168                     if (connect_ok == True):
 169                         if (resp.status_code == None):
 170                             print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: None")
 171                             sys.exit()
 172
 173                         if (resp.status_code != responsecode):
 174                             if (resp.status_code >= 500) and (http_retry_count < 600 ) and (retry_cnt > 1):
 175                                 sleep(0.1)
 176                                 retry_cnt -= 1
 177                                 http_retry_count += 1
 178                             else:
 179                                 print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: "+str(resp.status_code))
 180                                 print(url_out)
 181                                 print(str(data_out))
 182                                 sys.exit()
 183                         else:
 184                             retry_cnt=-1
 185
 186     print("0 http retries:"+str(http_retry_count) + ", connect retries: "+str(connect_retry_count))
 187     sys.exit()
 188
 189 except Exception as e:
 190     print("1"+str(e))
 191     traceback.print_exc()
 192     print(str(data_out))
 193 sys.exit()