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
  35 data_out=""
  36 url_out=""
  37 try:
  38
  39     if len(sys.argv) < 11:
  40         print("1Expected 11/14 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     if ("/v2/" in baseurl):
  50         if len(sys.argv) != 15:
  51             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")
  52             print (sys.argv[1:])
  53             sys.exit()
  54
  55         serv=str(sys.argv[7])
  56         pt=str(sys.argv[8])
  57         trans=str(sys.argv[9])
  58         noti=str(sys.argv[10])
  59         templatepath=str(sys.argv[11])
  60         count=int(sys.argv[12])
  61         pids=int(sys.argv[13])
  62         pid_id=int(sys.argv[14])
  63     else:
  64         if len(sys.argv) != 11:
  65             print("1Expected 10 args, got "+str(len(sys.argv)-1)+ ". Args: responsecode baseurl ric_base num_rics uuid startid templatepath count pids pid_id")
  66             print (sys.argv[1:])
  67             sys.exit()
  68
  69         templatepath=str(sys.argv[7])
  70         count=int(sys.argv[8])
  71         pids=int(sys.argv[9])
  72         pid_id=int(sys.argv[10])
  73
  74     if uuid == "NOUUID":
  75         uuid=""
  76
  77     with open(templatepath, 'r') as file:
  78         template = file.read()
  79
  80         start=start
  81         stop=count*num_rics+start
  82
  83         total_retry_count=0
  84
  85         for i in range(start,stop):
  86             if (i%pids == (pid_id-1)):
  87                 payload=template.replace("XXX",str(i))
  88                 ric_id=(i%num_rics)+1
  89                 ric=ric_base+str(ric_id)
  90
  91                 retry_cnt=5
  92                 while(retry_cnt>0):
  93                     try:
  94                         headers = {'Content-type': 'application/json'}
  95                         if ("/v2/" in baseurl):
  96                             url=baseurl
  97
  98                             data={}
  99                             data["ric_id"]=ric
 100                             data["policy_id"]=uuid+str(i)
 101                             data["service_id"]=serv
 102                             if (trans != "NOTRANSIENT"):
 103                                 data["transient"]=trans
 104                             if (pt != "NOTYPE"):
 105                                 data["policytype_id"]=pt
 106                             else:
 107                                 data["policytype_id"]=""
 108                             if (noti != "NOURL"):
 109                                 data["status_notification_uri"]=noti
 110                             data["policy_data"]=json.loads(payload)
 111
 112                             url_out=url
 113                             data_out=json.dumps(data)
 114                         else:
 115                             url=baseurl+"&id="+uuid+str(i)+"&ric="+str(ric)
 116                             url_out=url
 117                             data_out=json.dumps(json.loads(payload))
 118
 119                         resp=requests.put(url, data_out, headers=headers, verify=False, timeout=90)
 120                     except Exception as e1:
 121                         print("1Put failed for id:"+uuid+str(i)+ ", "+str(e1) + " "+traceback.format_exc())
 122                         sys.exit()
 123
 124                     if (resp.status_code == None):
 125                         print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: None")
 126                         sys.exit()
 127
 128                     if (resp.status_code != responsecode):
 129                         if (resp.status_code == 503 ) and (retry_cnt > 1):
 130                             sleep(0.1)
 131                             retry_cnt -= 1
 132                             total_retry_count += 1
 133                         else:
 134                             print("1Put failed for id:"+uuid+str(i)+ ", expected response code: "+str(responsecode)+", got: "+str(resp.status_code))
 135                             print(url_out)
 136                             print(str(data_out))
 137                             sys.exit()
 138                     else:
 139                         retry_cnt=-1
 140
 141     if (total_retry_count > 0):
 142         print("0 retries:"+str(total_retry_count))
 143     else:
 144         print("0")
 145     sys.exit()
 146
 147 except Exception as e:
 148     print("1"+str(e))
 149     traceback.print_exc()
 150     print(str(data_out))
 151 sys.exit()