| import random |
| import string |
| import time |
| import datetime |
| import sys |
| import collections |
| import json |
| import tzlocal |
| import os |
| import fcntl |
| import logging |
| from locust import HttpLocust, TaskSet, task |
| from decimal import Decimal |
| |
| |
| class UserBehavior(TaskSet): |
| base = "/ecomp/mso/infra/e2eServiceInstances/v3" |
| headers = {"Accept":"application/json","Content-Type":"application/json","Authorization":"Basic SW5mcmFQb3J0YWxDbGllbnQ6cGFzc3dvcmQxJA=="} |
| service_creation_body = "{\"service\": {\"name\": \"E2E_volte_%s\", \"description\": \"E2E_volte_ONAP_deploy\", \"serviceDefId\": \"a16eb184-4a81-4c8c-89df-c287d390315a\", \"templateId\": \"012c3446-51db-4a2a-9e64-a936f10a5e3c\", \"parameters\": { \"globalSubscriberId\": \"Demonstration\", \"subscriberName\": \"Demonstration\", \"serviceType\": \"vIMS\", \"templateName\": \"VoLTE e2e Service:null\", \"resources\": [ { \"resourceName\": \"VL OVERLAYTUNNEL\", \"resourceDefId\": \"671d4757-b018-47ab-9df3-351c3bda0a98\", \"resourceId\": \"e859b0fd-d928-4cc8-969e-0fee7795d623\", \"nsParameters\": { \"locationConstraints\": [], \"additionalParamForNs\": { \"site2_vni\": \"5010\", \"site1_localNetworkAll\": \"false\", \"site1_vni\": \"5010\", \"site1_exportRT1\": \"11:1\", \"description\": \"overlay\", \"site2_localNetworkAll\": \"false\", \"site1_routerId\": \"9.9.9.9\", \"site1_fireWallEnable\": \"false\", \"site1_networkName\": \"network1\", \"site2_description\": \"overlay\", \"site1_importRT1\": \"11:1\", \"site1_description\": \"overlay\", \"site2_networkName\": \"network3\", \"name\": \"overlay\", \"site2_fireWallEnable\": \"false\", \"site2_id\": \"ZTE-DCI-Controller\", \"site2_routerId\": \"9.9.9.9\", \"site2_importRT1\": \"11:1\", \"site2_exportRT1\": \"11:1\", \"site2_fireWallId\": \"false\", \"site1_id\": \"DCI-Controller-1\", \"tunnelType\": \"L3-DCI\" } } },{\"resourceName\": \"VL UNDERLAYVPN\", \"resourceDefId\": \"4f5d692b-4022-43ab-b878-a93deb5b2061\", \"resourceId\": \"b977ec47-45b2-41f6-aa03-bf6554dc9620\", \"nsParameters\": { \"locationConstraints\": [], \"additionalParamForNs\": { \"topology\": \"full-mesh\", \"site2_name\": \"site2\", \"sna2_name\": \"site2_sna\", \"description\": \"underlay\", \"sna1_name\": \"site1_sna\", \"ac1_route\": \"3.3.3.12/30:dc84ce88-99f7\", \"ac2_peer_ip\": \"3.3.3.20/30\", \"technology\": \"mpls\", \"ac2_route\": \"3.3.3.20/30:98928302-3287\", \"ac2_id\": \"84d937a4-b227-375f-a744-2b778f36e04e\", \"ac1_protocol\": \"STATIC\", \"ac2_svlan\": \"4004\", \"serviceType\": \"l3vpn-ipwan\", \"ac2_ip\": \"3.3.3.21/30\", \"pe2_id\": \"4412d3f0-c296-314d-9284-b72fc5d485e8\", \"ac1_id\": \"b4f01ac0-c1e1-3e58-a8be-325e4372c960\", \"af_type\": \"ipv4\", \"ac1_svlan\": \"4002\", \"ac1_peer_ip\": \"3.3.3.12/30\", \"ac1_ip\": \"3.3.3.13/30\", \"version\": \"1.0\", \"name\": \"testunderlay\", \"id\": \"123124141\", \"pe1_id\": \"2ef788f0-407c-3070-b756-3a5cd71fde18\", \"ac2_protocol\": \"STATIC\", \"site1_name\": \"stie1\" } } } ] } } }" |
| # following class variables to make them unique across all users |
| transaction_file= open("transaction.log", "w+") |
| operation_file = open("operation.log", "w+") |
| |
| def on_start(self): |
| """ on_start is called when a Locust start before any task is scheduled """ |
| self.init() |
| |
| def init(self): |
| pass |
| |
| def myconverter(self, o): |
| if isinstance(o, datetime.datetime): |
| return o.__str__() |
| |
| @task(1) |
| def create_service(self): |
| # Post a E2E service instantiation request to SO |
| method = "POST" |
| url = self.base |
| service_instance_name = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(10)) |
| data = self.service_creation_body % service_instance_name |
| |
| t1 = datetime.datetime.now(tzlocal.get_localzone()) |
| response = self.client.request(method, url, headers=self.headers, data=data) |
| t2 = datetime.datetime.now(tzlocal.get_localzone()) |
| delta = t2 - t1 |
| data = collections.OrderedDict() |
| data['datetime'] = datetime.datetime.now(tzlocal.get_localzone()).strftime("%Y-%m-%dT%H:%M:%S%Z") |
| data['method'] = method |
| data['url'] = url |
| data['status_code'] = response.status_code |
| data['transaction_time'] = (delta.seconds*10^6 + delta.microseconds)/1000 |
| fcntl.flock(self.transaction_file, fcntl.LOCK_EX) |
| self.transaction_file.write(json.dumps(data, default = self.myconverter) + "\n") |
| self.transaction_file.flush() |
| os.fsync(self.transaction_file) |
| fcntl.flock(self.transaction_file, fcntl.LOCK_UN) |
| serviceId = response.json()['service']['serviceId'] |
| operationId = response.json()['service']['operationId'] |
| |
| # Get the request status |
| method = "GET" |
| url = self.base + "/" + serviceId + "/operations/" + operationId |
| url1 = "/ecomp/mso/infra/e2eServiceInstances/v3/{serviceId}/operations/{operationId}" |
| count = 1 |
| while count < 50: |
| tt1 = datetime.datetime.now() |
| response = self.client.request(method, url, name=url1, headers=self.headers) |
| tt2 = datetime.datetime.now() |
| delta = tt2 - tt1 |
| result = response.json()['operationStatus']['result'] |
| progress = response.json()['operationStatus']['progress'] |
| data = collections.OrderedDict() |
| data['datetime'] = datetime.datetime.now(tzlocal.get_localzone()).strftime("%Y-%m-%dT%H:%M:%S%Z") |
| data['method'] = method |
| data['url'] = url1 |
| data['status_code'] = response.status_code |
| data['count'] = count |
| data['result'] = result |
| data['progress'] = progress |
| data['transaction_time'] = (delta.seconds*10^6 + delta.microseconds)/1000 |
| fcntl.flock(self.transaction_file, fcntl.LOCK_EX) |
| self.transaction_file.write(json.dumps(data, default = self.myconverter) + "\n") |
| self.transaction_file.flush() |
| os.fsync(self.transaction_file) |
| fcntl.flock(self.transaction_file, fcntl.LOCK_UN) |
| if result == "finished" or result == "error": |
| break |
| else: |
| time.sleep(1) |
| count = count + 1 |
| |
| if result == "finished": |
| result = "success" |
| else: |
| result = "failure" |
| t3 = datetime.datetime.now(tzlocal.get_localzone()) |
| delta = t3 - t1 |
| data = collections.OrderedDict() |
| data['datetime'] = t1.strftime("%Y-%m-%dT%H:%M:%S%Z") |
| data['operation'] = "volte_create" |
| data['result'] = result |
| data['duration'] = round(delta.seconds + Decimal(delta.microseconds/1000000.0), 3) |
| fcntl.flock(self.operation_file, fcntl.LOCK_EX) |
| self.operation_file.write(json.dumps(data, default = self.myconverter) + "\n") |
| self.operation_file.flush() |
| os.fsync(self.operation_file) |
| fcntl.flock(self.operation_file, fcntl.LOCK_UN) |
| |
| self.delete_service(serviceId) |
| |
| def delete_service(self, serviceId): |
| method = "DELETE" |
| url = self.base + "/" + serviceId |
| data = "{\"globalSubscriberId\":\"Demonstration\", \"serviceType\":\"vIMS\"}" |
| t1 = datetime.datetime.now(tzlocal.get_localzone()) |
| response = self.client.request(method, url, name=self.base, headers=self.headers, data=data) |
| t2 = datetime.datetime.now(tzlocal.get_localzone()) |
| delta = t2 - t1 |
| data = collections.OrderedDict() |
| data['datetime'] = datetime.datetime.now(tzlocal.get_localzone()).strftime("%Y-%m-%dT%H:%M:%S%Z") |
| data['method'] = method |
| data['url'] = self.base |
| data['status_code'] = response.status_code |
| data['transaction_time'] = (delta.seconds*10^6 + delta.microseconds)/1000 |
| fcntl.flock(self.transaction_file, fcntl.LOCK_EX) |
| self.transaction_file.write(json.dumps(data, default = self.myconverter) + "\n") |
| self.transaction_file.flush() |
| os.fsync(self.transaction_file) |
| fcntl.flock(self.transaction_file, fcntl.LOCK_UN) |
| operationId = response.json()['operationId'] |
| |
| # Get the request status |
| method = "GET" |
| url = self.base + "/" + serviceId + "/operations/" + operationId |
| url1 = "/ecomp/mso/infra/e2eServiceInstances/v3/{serviceId}/operations/{operationId}" |
| count = 1 |
| while count < 50: |
| tt1 = datetime.datetime.now(tzlocal.get_localzone()) |
| response = self.client.request(method, url, name=url1, headers=self.headers) |
| tt2 = datetime.datetime.now(tzlocal.get_localzone()) |
| delta = tt2 - tt1 |
| result = response.json()['operationStatus']['result'] |
| progress = response.json()['operationStatus']['progress'] |
| data = collections.OrderedDict() |
| data['datetime'] = datetime.datetime.now(tzlocal.get_localzone()).strftime("%Y-%m-%dT%H:%M:%S%Z") |
| data['method'] = method |
| data['url'] = url1 |
| data['status_code'] = response.status_code |
| data['count'] = count |
| data['result'] = result |
| data['progress'] = progress |
| data['transaction_time'] = (delta.seconds*10^6 + delta.microseconds)/1000 |
| fcntl.flock(self.transaction_file, fcntl.LOCK_EX) |
| self.transaction_file.write(json.dumps(data, default = self.myconverter) + "\n") |
| self.transaction_file.flush() |
| os.fsync(self.transaction_file) |
| fcntl.flock(self.transaction_file, fcntl.LOCK_UN) |
| if result == "finished" or result == "error": |
| break |
| else: |
| time.sleep(1) |
| count = count + 1 |
| |
| if result == "finished": |
| result = "success" |
| else: |
| result = "failure" |
| t3 = datetime.datetime.now(tzlocal.get_localzone()) |
| delta = t3 - t1 |
| data = collections.OrderedDict() |
| data['datetime'] = t1.strftime("%Y-%m-%dT%H:%M:%S%Z") |
| data['operation'] = "volte_delete" |
| data['result'] = result |
| data['duration'] = round(delta.seconds + Decimal(delta.microseconds/1000000.0), 3) |
| fcntl.flock(self.operation_file, fcntl.LOCK_EX) |
| self.operation_file.write(json.dumps(data, default = self.myconverter) + "\n") |
| self.operation_file.flush() |
| os.fsync(self.operation_file) |
| fcntl.flock(self.operation_file, fcntl.LOCK_UN) |
| |
| |
| class WebsiteUser(HttpLocust): |
| task_set = UserBehavior |
| min_wait = 1000 |
| max_wait = 3000 |