Migrate Get Requests invocations(libraries)

[integration/test.git] / csit / libraries / ClusterStateLibrary.py

diff --git a/csit/libraries/ClusterStateLibrary.py b/csit/libraries/ClusterStateLibrary.py
index ab5a643bb29d97a7ffb88f1ca612e5ef93cecd7d..54fc0839925358916df265bbdfd4f418b117ae99 100644 (file)
--- a/csit/libraries/ClusterStateLibrary.py
+++ b/csit/libraries/ClusterStateLibrary.py
@@ -11,7 +11,9 @@ __license__ = "New-style BSD"
 __email__ = "syedbahm@cisco.com"
 
 
-def getClusterRoles(shardName, numOfShards=3, numOfTries=3, sleepBetweenRetriesInSecs=3, port=8181, *ips):
+def getClusterRoles(
+    shardName, numOfShards=3, numOfTries=3, sleepBetweenRetriesInSecs=3, port=8181, *ips
+):
     """Given a shardname (e.g. shard-inventory-config), number of shards and bunch of ips
 
     determines what role each ip has in an Akka (Raft based) cluster
@@ -22,18 +24,22 @@ def getClusterRoles(shardName, numOfShards=3, numOfTries=3, sleepBetweenRetriesI
     for ip in ips:
         i = 1
         dict[ip] = None
-        print "numOfShards => " + str(numOfShards)
+        print("numOfShards => ", str(numOfShards))
         while i <= numOfShards:
             shardMemberName = "member-" + str(i) + "-" + shardName
             j = 1
-            print 'j => ' + str(j)
-            print 'numOfTries => ' + str(numOfTries)
+            print("j => ", str(j))
+            print("numOfTries => ", str(numOfTries))
             while int(j) <= int(numOfTries):
                 print("Try number " + str(j))
                 try:
-                    print("getting role of " + ip + "  for shardName = " + shardMemberName)
-                    url = SettingsLibrary.getJolokiaURL(ip, str(port), str(i), shardName)
-                    print url
+                    print(
+                        "getting role of " + ip + "  for shardName = " + shardMemberName
+                    )
+                    url = SettingsLibrary.getJolokiaURL(
+                        ip, str(port), str(i), shardName
+                    )
+                    print(url)
                     resp = UtilLibrary.get(url)
                     print(resp)
                     if resp.status_code != 200:
@@ -41,14 +47,20 @@ def getClusterRoles(shardName, numOfShards=3, numOfTries=3, sleepBetweenRetriesI
                         continue
                     print(resp.text)
                     data = json.loads(resp.text)
-                    if 'value' in data:
-                        dataValue = data['value']
-                        print("datavalue RaftState is", dataValue['RaftState'])
-                        dict[ip] = dataValue['RaftState']
-                except:
+                    if "value" in data:
+                        dataValue = data["value"]
+                        print("datavalue RaftState is", dataValue["RaftState"])
+                        dict[ip] = dataValue["RaftState"]
+                except Exception:
                     e = sys.exc_info()[0]
-                    print("Try" + str(j) + ":An error occurred when finding leader on" + ip +
-                          " for shardName:" + shardMemberName)
+                    print(
+                        "Try"
+                        + str(j)
+                        + ":An error occurred when finding leader on"
+                        + ip
+                        + " for shardName:"
+                        + shardMemberName
+                    )
                     print(e)
                     sleep(sleepBetweenRetriesInSecs)
                     continue
@@ -60,48 +72,68 @@ def getClusterRoles(shardName, numOfShards=3, numOfTries=3, sleepBetweenRetriesI
     return dict
 
 
-def isRole(role, shardName, ipAddress, numOfShards=3, numOfRetries=1, sleepFor=3, port=8181):
+def isRole(
+    role, shardName, ipAddress, numOfShards=3, numOfRetries=1, sleepFor=3, port=8181
+):
     """Given a role (Leader, Follower, Candidate, or IsolatedLeader),
     shardname (e.g. shard-inventory-config), controller IP address,
     and number of shards on the controller,this function determines if the controller,
     has that role for the specified shard.
     """
-    ip = getClusterRoles(shardName, numOfShards, numOfRetries, sleepFor, port, ipAddress)
+    ip = getClusterRoles(
+        shardName, numOfShards, numOfRetries, sleepFor, port, ipAddress
+    )
     print(ip)
     if ip[ipAddress] == role:
         return True
     return False
 
 
-def getLeader(shardName, numOfShards=3, numOfTries=3, sleepBetweenRetriesInSecs=1, port=8181, *ips):
+def getLeader(
+    shardName, numOfShards=3, numOfTries=3, sleepBetweenRetriesInSecs=1, port=8181, *ips
+):
     """Returns the leader of the shard given a set of IPs Or None"""
     for i in range(3):  # Try 3 times to find a leader
-        dict = getClusterRoles(shardName, numOfShards, numOfTries, sleepBetweenRetriesInSecs, port, *ips)
+        dict = getClusterRoles(
+            shardName, numOfShards, numOfTries, sleepBetweenRetriesInSecs, port, *ips
+        )
         for ip in dict.keys():
-            if dict[ip] == 'Leader':
+            if dict[ip] == "Leader":
                 return ip
     return None
 
 
-def getFollowers(shardName, numOfShards=3, numOfTries=3, sleepBetweenRetriesInSecs=1, port=8181, *ips):
+def getFollowers(
+    shardName, numOfShards=3, numOfTries=3, sleepBetweenRetriesInSecs=1, port=8181, *ips
+):
     """Returns the follower list of a shard given a set of IPs Or []"""
     for i in range(6):  # Try 6 times to find all followers
-        dict = getClusterRoles(shardName, numOfShards, numOfTries, sleepBetweenRetriesInSecs, port, *ips)
+        dict = getClusterRoles(
+            shardName, numOfShards, numOfTries, sleepBetweenRetriesInSecs, port, *ips
+        )
         result = []
 
         for ip in dict.keys():
-            if dict[ip] == 'Follower':
+            if dict[ip] == "Follower":
                 result.append(ip)
-        print "i=", i, "result=", result
-        if (len(result) == (len(ips) - 1)):
+        print("i=%s result=%s" % (i, result))
+        if len(result) == (len(ips) - 1):
             break
         sleep(1)
     return result
 
 
 def testGetClusterRoles():
-    dict = getClusterRoles("shard-inventory-config", 3, 1, 1, 8181,
-                           "10.194.126.116", "10.194.126.117", "10.194.126.118")
+    dict = getClusterRoles(
+        "shard-inventory-config",
+        3,
+        1,
+        1,
+        8181,
+        "10.194.126.116",
+        "10.194.126.117",
+        "10.194.126.118",
+    )
     print(dict)
 
     for ip in dict.keys():
@@ -114,18 +146,35 @@ def testGetClusterRoles():
 
 
 def testGetLeader():
-    leader = getLeader("shard-inventory-config", 3, 1, 1, 8181,
-                       "10.194.126.116", "10.194.126.117", "10.194.126.118")
-    print leader
+    leader = getLeader(
+        "shard-inventory-config",
+        3,
+        1,
+        1,
+        8181,
+        "10.194.126.116",
+        "10.194.126.117",
+        "10.194.126.118",
+    )
+    print(leader)
     return leader
 
 
 def testGetFollowers():
-    followers = getFollowers("shard-inventory-config", 3, 1, 1, 8181,
-                             "10.194.126.116", "10.194.126.117", "10.194.126.118")
+    followers = getFollowers(
+        "shard-inventory-config",
+        3,
+        1,
+        1,
+        8181,
+        "10.194.126.116",
+        "10.194.126.117",
+        "10.194.126.118",
+    )
     print(followers)
     return followers
 
+
 # testGetClusterRoles()
 # testGetLeader()
 # testGetFollowers()