aboutsummaryrefslogtreecommitdiffstats
path: root/resources/tools/ab
diff options
context:
space:
mode:
authorpmikus <peter.mikus@protonmail.ch>2023-01-20 07:46:58 +0000
committerpmikus <peter.mikus@protonmail.ch>2023-01-20 07:46:58 +0000
commit6e2a3e51f363fe8ff478bc46cd97394dab98dc9a (patch)
tree4f6e56e4c61edeb34f0df1291114df052f90d3d1 /resources/tools/ab
parent464eb6e9d244575b585e5b557044e152f6ef694f (diff)
fix(core): AB library
Signed-off-by: pmikus <peter.mikus@protonmail.ch> Change-Id: Ib79a3f010abd28a57248c03ca62e8070df000c77
Diffstat (limited to 'resources/tools/ab')
-rw-r--r--resources/tools/ab/ABTools.py58
1 files changed, 29 insertions, 29 deletions
diff --git a/resources/tools/ab/ABTools.py b/resources/tools/ab/ABTools.py
index f3e97a720a..bb21ea351f 100644
--- a/resources/tools/ab/ABTools.py
+++ b/resources/tools/ab/ABTools.py
@@ -38,29 +38,29 @@ class ABTools:
:rtype: OptionString
"""
cmd = OptionString()
- cmd.add(u"python3")
+ cmd.add("python3")
dirname = f"{Constants.REMOTE_FW_DIR}/resources/tools/ab"
cmd.add(f"{dirname}/ABFork.py")
- cmd_options = OptionString(prefix=u"-")
+ cmd_options = OptionString(prefix="-")
# Number of requests to perform.
- cmd_options.add_with_value_from_dict(u"r", u"requests", kwargs)
+ cmd_options.add_with_value_from_dict("r", "requests", kwargs)
# Server port number to use.
- cmd_options.add_with_value_from_dict(u"p", u"port", kwargs)
+ cmd_options.add_with_value_from_dict("p", "port", kwargs)
# Number of clients being processed at the same time.
- cmd_options.add_with_value_from_dict(u"c", u"clients", kwargs)
+ cmd_options.add_with_value_from_dict("c", "clients", kwargs)
# Filename to be requested from the servers.
- cmd_options.add_with_value_from_dict(u"f", u"files", kwargs)
+ cmd_options.add_with_value_from_dict("f", "files", kwargs)
# Server ip address.
- cmd_options.add_with_value_from_dict(u"i", u"ip", kwargs)
+ cmd_options.add_with_value_from_dict("i", "ip", kwargs)
# tg ip address.
- cmd_options.add_with_value_from_dict(u"g", u"tip", kwargs)
+ cmd_options.add_with_value_from_dict("g", "tip", kwargs)
# Specify SSL/TLS cipher suite.
- cmd_options.add_with_value_from_dict(u"z", u"cipher", kwargs, default=0)
+ cmd_options.add_with_value_from_dict("z", "cipher", kwargs, default=0)
# Specify SSL/TLS protocol.
- cmd_options.add_with_value_from_dict(u"t", u"protocol", kwargs,
+ cmd_options.add_with_value_from_dict("t", "protocol", kwargs,
default=0)
# Mode: RPS or CPS.
- cmd_options.add_with_value_from_dict(u"m", u"mode", kwargs)
+ cmd_options.add_with_value_from_dict("m", "mode", kwargs)
return cmd.extend(cmd_options)
@staticmethod
@@ -73,11 +73,11 @@ class ABTools:
command is not available.
"""
- if tg_node[u"type"] != NodeType.TG:
- raise RuntimeError(u"Node type is not a TG!")
+ if tg_node["type"] != NodeType.TG:
+ raise RuntimeError("Node type is not a TG!")
- cmd = u"command -v ab"
- message = u"ab not installed on TG node!"
+ cmd = "command -v ab"
+ message = "ab not installed on TG node!"
exec_cmd_no_error(tg_node, cmd, message=message)
@staticmethod
@@ -101,13 +101,13 @@ class ABTools:
:rtype: str
"""
command = f"ab -V | head -1 | cut -d',' -f2"
- message = u"Get AB version failed!"
+ message = "Get AB version failed!"
stdout, _ = exec_cmd_no_error(node, command, message=message)
return stdout.strip()
@staticmethod
def run_ab(tg_node, ip_addr, tg_addr, tls_tcp, cipher, files_num, rps_cps,
- r_total, c_total, port, protocol=u"TLS1.3"):
+ r_total, c_total, port, protocol="TLS1.3"):
""" Run ab test.
:param tg_node: Topology node.
@@ -138,7 +138,7 @@ class ABTools:
:raises: RuntimeError if node type is not a TG.
"""
if files_num == 0:
- files = u"return"
+ files = "return"
elif files_num >= 1024:
files = f"{int(files_num / 1024)}KB.json"
else:
@@ -156,7 +156,7 @@ class ABTools:
mode=rps_cps,
)
stdout, _ = exec_cmd_no_error(
- tg_node, cmd, timeout=180, sudo=True, message=u"ab runtime error!"
+ tg_node, cmd, timeout=180, sudo=True, message="ab runtime error!"
)
rate_unit = rps_cps
@@ -167,18 +167,18 @@ class ABTools:
failed_requests = None
for line in stdout.splitlines():
if f"Connection {rps_cps} rate:" in line:
- rate = float(line.split(u" ")[3])
- elif u"Transfer Rate:" in line:
- bandwidth = float(line.split(u" ")[2]) * 8000
- elif u"Latency:" in line:
- latency = float(line.split(u" ")[1])
- elif u"Completed requests:" in line:
- completed_requests = int(line.split(u" ")[2])
- elif u"Failed requests" in line:
- failed_requests = int(line.split(u" ")[2])
+ rate = float(line.split(" ")[3])
+ elif "Transfer Rate:" in line:
+ bandwidth = float(line.split(" ")[2]) * 8000
+ elif "Latency:" in line:
+ latency = float(line.split(" ")[1])
+ elif "Completed requests:" in line:
+ completed_requests = int(line.split(" ")[2])
+ elif "Failed requests" in line:
+ failed_requests = int(line.split(" ")[2])
export_hoststack_results(
- rate, rate_unit, bandwidth, latency, failed_requests,
+ bandwidth, rate, rate_unit, latency, failed_requests,
completed_requests
)