Auto-generated patch by python-black
[integration/test.git] / csit / libraries / CrudLibrary.py
index baaf5f4547f30e12bdc45e503e3993a226607636..0f80253a3a5c81ac80c56b9cd082a521e4c8376a 100644 (file)
@@ -15,12 +15,17 @@ def initCar(hostname, port):
     x = 0
     strId = str(x)
     payload = SettingsLibrary.add_car_init_payload_template.substitute(
-        id=strId, category="category" + strId, model="model" + strId,
+        id=strId,
+        category="category" + strId,
+        model="model" + strId,
         manufacturer="manufacturer" + strId,
-        year=(2000 + x % 100))
+        year=(2000 + x % 100),
+    )
     print("Initialization payload=")
     print(payload)
-    resp = UtilLibrary.post(SettingsLibrary.getAddCarInitUrl(hostname, port), "admin", "admin", payload)
+    resp = UtilLibrary.post(
+        SettingsLibrary.getAddCarInitUrl(hostname, port), "admin", "admin", payload
+    )
     print("the response of the POST to add car=")
     print(resp)
     return resp
@@ -31,19 +36,27 @@ def addCar(hostname, port, numberOfCars, *expected):
     for x in range(1, numberOfCars + 1):
         strId = str(x)
         payload = SettingsLibrary.add_car_payload_template.substitute(
-            id=strId, category="category" + strId, model="model" + strId,
+            id=strId,
+            category="category" + strId,
+            model="model" + strId,
             manufacturer="manufacturer" + strId,
-            year=(2000 + x % 100))
+            year=(2000 + x % 100),
+        )
         print("payload formed after template substitution=")
         print(payload)
         # Send the POST request
-        resp = UtilLibrary.post(SettingsLibrary.getAddCarUrl(hostname, port), "admin", "admin", payload)
+        resp = UtilLibrary.post(
+            SettingsLibrary.getAddCarUrl(hostname, port), "admin", "admin", payload
+        )
 
         print("the response of the POST to add car=")
         print(resp)
         if expected and str(resp.status_code) not in expected:
-            raise RuntimeError('Add car failed for {}:{} with status {}'.
-                               format(hostname, port, resp.status_code))
+            raise RuntimeError(
+                "Add car failed for {}:{} with status {}".format(
+                    hostname, port, resp.status_code
+                )
+            )
 
     return resp
 
@@ -57,19 +70,24 @@ def addPerson(hostname, port, numberOfPersons, *expected):
     </note>
     """
     # FOR RPC TO WORK PROPERLY THE FIRST ENTRY SHOULD BE VIA RESTCONF
-    if (numberOfPersons == 0):
+    if numberOfPersons == 0:
         strId = str(numberOfPersons)
         payload = SettingsLibrary.add_person_payload_template.substitute(
-            personId="user" + strId, gender="unknown", age=0,
+            personId="user" + strId,
+            gender="unknown",
+            age=0,
             address=strId + "Way, Some Country, Some Zip  " + strId,
-            contactNo="some number" + strId)
+            contactNo="some number" + strId,
+        )
         # Send the POST request using RESTCONF
-        resp = UtilLibrary.nonprintpost(SettingsLibrary.getAddPersonUrl(hostname, port), "admin", "admin", payload)
+        resp = UtilLibrary.nonprintpost(
+            SettingsLibrary.getAddPersonUrl(hostname, port), "admin", "admin", payload
+        )
         return resp
 
     genderToggle = "Male"
     for x in range(1, numberOfPersons + 1):
-        if(genderToggle == "Male"):
+        if genderToggle == "Male":
             genderToggle = "Female"
         else:
             genderToggle = "Male"
@@ -77,19 +95,30 @@ def addPerson(hostname, port, numberOfPersons, *expected):
         strId = str(x)
 
         payload = SettingsLibrary.add_person_rpc_payload_template.substitute(
-            personId="user" + strId, gender=genderToggle, age=(20 + x % 100),
+            personId="user" + strId,
+            gender=genderToggle,
+            age=(20 + x % 100),
             address=strId + "Way, Some Country, Some Zip  " + str(x % 1000),
-            contactNo="some number" + strId)
+            contactNo="some number" + strId,
+        )
         # Send the POST request using RPC
-        resp = UtilLibrary.post(SettingsLibrary.getAddPersonRpcUrl(hostname, port), "admin", "admin", payload)
+        resp = UtilLibrary.post(
+            SettingsLibrary.getAddPersonRpcUrl(hostname, port),
+            "admin",
+            "admin",
+            payload,
+        )
 
         print("payload formed after template substitution=")
         print(payload)
         print("the response of the POST to add person=")
         print(resp)
         if expected and str(resp.status_code) not in expected:
-            raise RuntimeError('Add person failed for {}:{} with status {}'.
-                               format(hostname, port, resp.status_code))
+            raise RuntimeError(
+                "Add person failed for {}:{} with status {}".format(
+                    hostname, port, resp.status_code
+                )
+            )
 
     return resp
 
@@ -105,21 +134,34 @@ def addCarPerson(hostname, port, numberOfCarPersons):
     </note>
     """
     # FOR RPC TO WORK PROPERLY THE FIRST ENTRY SHOULD BE VIA RESTCONF
-    if (numberOfCarPersons == 0):
+    if numberOfCarPersons == 0:
         payload = SettingsLibrary.add_car_person_template.substitute(
-            Id=str(numberOfCarPersons), personId="user" + str(numberOfCarPersons))
+            Id=str(numberOfCarPersons), personId="user" + str(numberOfCarPersons)
+        )
         # Send the POST request REST CONF
-        resp = UtilLibrary.nonprintpost(SettingsLibrary.getAddCarPersonUrl(hostname, port), "admin", "admin", payload)
+        resp = UtilLibrary.nonprintpost(
+            SettingsLibrary.getAddCarPersonUrl(hostname, port),
+            "admin",
+            "admin",
+            payload,
+        )
 
         return resp
 
     for x in range(1, numberOfCarPersons + 1):
         strId = str(x)
 
-        payload = SettingsLibrary.add_car_person_template.substitute(Id=strId, personId="user" + strId)
+        payload = SettingsLibrary.add_car_person_template.substitute(
+            Id=strId, personId="user" + strId
+        )
 
         # Send the POST request REST CONF
-        resp = UtilLibrary.post(SettingsLibrary.getAddCarPersonUrl(hostname, port), "admin", "admin", payload)
+        resp = UtilLibrary.post(
+            SettingsLibrary.getAddCarPersonUrl(hostname, port),
+            "admin",
+            "admin",
+            payload,
+        )
 
         print("payload formed after template substitution=")
         print(payload)
@@ -146,23 +188,30 @@ def buyCar(hostname, port, numberOfCarBuyers, start=0):
     for x in range(start, start + numberOfCarBuyers):
         strId = str(x + 1)
 
-        payload = SettingsLibrary.buy_car_rpc_template.substitute(personId="user" + strId, carId=strId)
+        payload = SettingsLibrary.buy_car_rpc_template.substitute(
+            personId="user" + strId, carId=strId
+        )
 
         # Send the POST request using RPC
-        resp = UtilLibrary.post(SettingsLibrary.getBuyCarRpcUrl(hostname, port), "admin", "admin", payload)
+        resp = UtilLibrary.post(
+            SettingsLibrary.getBuyCarRpcUrl(hostname, port), "admin", "admin", payload
+        )
 
         print(resp)
         print(resp.text)
 
-        if (resp.status_code != 200):
-            raise RuntimeError('Buy car failed for {}:{} with status {}'.
-                               format(hostname, port, resp.status_code))
+        if resp.status_code != 200:
+            raise RuntimeError(
+                "Buy car failed for {}:{} with status {}".format(
+                    hostname, port, resp.status_code
+                )
+            )
 
 
 def getCars(hostname, port, ignore):
     """Uses the GET on car:cars resource to get all cars in the store using RESTCONF"""
     resp = UtilLibrary.get(SettingsLibrary.getCarsUrl(hostname, port), "admin", "admin")
-    resp.encoding = 'utf-8'
+    resp.encoding = "utf-8"
     print(resp.text)
     return resp
 
@@ -175,8 +224,10 @@ def getPersons(hostname, port, ignore):
         with personId being user0
     </note>
     """
-    resp = UtilLibrary.get(SettingsLibrary.getPersonsUrl(hostname, port), "admin", "admin")
-    resp.encoding = 'utf-8'
+    resp = UtilLibrary.get(
+        SettingsLibrary.getPersonsUrl(hostname, port), "admin", "admin"
+    )
+    resp.encoding = "utf-8"
     print(resp.text)
     return resp
 
@@ -190,8 +241,10 @@ def getCarPersonMappings(hostname, port, ignore):
         with personId being user0
     </note>
     """
-    resp = UtilLibrary.get(SettingsLibrary.getCarPersonUrl(hostname, port), "admin", "admin")
-    resp.encoding = 'utf-8'
+    resp = UtilLibrary.get(
+        SettingsLibrary.getCarPersonUrl(hostname, port), "admin", "admin"
+    )
+    resp.encoding = "utf-8"
     print(resp)
 
     return resp
@@ -213,7 +266,9 @@ def deleteAllPersons(hostname, port, ignore):
 
 def deleteAllCarsPersons(hostname, port, ignore):
     """delete all car -poeple s in the store using RESTCONF"""
-    UtilLibrary.delete(SettingsLibrary.getCarPersonUrl(hostname, port), "admin", "admin")
+    UtilLibrary.delete(
+        SettingsLibrary.getCarPersonUrl(hostname, port), "admin", "admin"
+    )
     resp = getPersons(hostname, port, ignore)
     print("Persons in store after deletion:" + str(resp))
 
@@ -273,27 +328,33 @@ def testlongevity(inputtime, port, *ips):
                     print("Pass: car person data matches")
             else:
                 print("Fail: car person addition failed")
-            time.sleep(60)    # sleep before next host starts working
+            time.sleep(60)  # sleep before next host starts working
 
 
 #
 # Usage message shown to user
 #
 
-def options():
 
-    command = 'ac=Add Car\n\t\tap=Add Person \n\t\tbc=Buy Car\n\t\tgc=Get Cars\n\t\tgp=Get Persons\n\t\t' \
-              'gcp=Get Car-Person Mappings\n\t\tdc=Delete All Cars\n\t\tdp=Delete All Persons)'
+def options():
 
-    param = '\n\t<param> is\n\t\t' \
-            'number of cars to be added if <command>=ac\n\t\t' \
-            'number of persons to be added if <command>=ap\n\t\t' \
-            'number of car buyers if <command>=bc\n\t\t'\
-            'pass 0 if <command>=gc or gp or gcp or dc or dp'\
+    command = (
+        "ac=Add Car\n\t\tap=Add Person \n\t\tbc=Buy Car\n\t\tgc=Get Cars\n\t\tgp=Get Persons\n\t\t"
+        "gcp=Get Car-Person Mappings\n\t\tdc=Delete All Cars\n\t\tdp=Delete All Persons)"
+    )
 
+    param = (
+        "\n\t<param> is\n\t\t"
+        "number of cars to be added if <command>=ac\n\t\t"
+        "number of persons to be added if <command>=ap\n\t\t"
+        "number of car buyers if <command>=bc\n\t\t"
+        "pass 0 if <command>=gc or gp or gcp or dc or dp"
+    )
 
-    usageString = 'usage: python crud <ipaddress> <command> <param>\nwhere\n\t<ipaddress> = ODL server ip address' \
-                  '\n\t<command> = any of the following commands \n\t\t'
+    usageString = (
+        "usage: python crud <ipaddress> <command> <param>\nwhere\n\t<ipaddress> = ODL server ip address"
+        "\n\t<command> = any of the following commands \n\t\t"
+    )
 
     usageString = usageString + command + param
 
@@ -304,14 +365,23 @@ def options():
 # entry point for command executions
 #
 
+
 def main():
     if len(sys.argv) < 4:
         options()
         quit(0)
     SettingsLibrary.hostname = sys.argv[1]
-    SettingsLibrary.port = '8181'
-    call = dict(ac=addCar, ap=addPerson, bc=buyCar,
-                gc=getCars, gp=getPersons, gcp=getCarPersonMappings, dc=deleteAllCars, dp=deleteAllPersons)
+    SettingsLibrary.port = "8181"
+    call = dict(
+        ac=addCar,
+        ap=addPerson,
+        bc=buyCar,
+        gc=getCars,
+        gp=getPersons,
+        gcp=getCarPersonMappings,
+        dc=deleteAllCars,
+        dp=deleteAllPersons,
+    )
 
     # FOR RPC TO WORK PROPERLY THE FIRST PERSON SHOULD BE ADDED VIA RESTCONF
     addPerson(SettingsLibrary.hostname, SettingsLibrary.port, 0)