From aa71d6f9311d3b815b281094ce2e7cfc9b9e3f2f Mon Sep 17 00:00:00 2001
From: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
Date: Wed, 2 Aug 2023 16:46:48 +0000
Subject: [PATCH] metrics: Add network latency test

This PR adds network latency test for kata metrics.

Fixes #7526

Signed-off-by: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
(cherry picked from commit 64fdb98704e68ec91f4683624872f06037c388ed)
---
 latency_kubernetes/latency-network.sh | 85 +++++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100755 latency_kubernetes/latency-network.sh

diff --git a/latency_kubernetes/latency-network.sh b/latency_kubernetes/latency-network.sh
new file mode 100755
index 0000000000..19c84ef14e
--- /dev/null
+++ b/latency_kubernetes/latency-network.sh
@@ -0,0 +1,85 @@
+#!/bin/bash
+#
+# Copyright (c) 2022 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+set -o pipefail
+
+SCRIPT_PATH=$(dirname "$(readlink -f "$0")")
+
+source "${SCRIPT_PATH}/../../lib/common.bash"
+latency_file=$(mktemp latencyresults.XXXXXXXXXX)
+TEST_NAME="${TEST_NAME:-latency}"
+
+function remove_tmp_file() {
+	rm -rf "${latency_file}"
+}
+
+trap remove_tmp_file EXIT
+
+function main() {
+	init_env
+	cmds=("bc" "jq")
+	check_cmds "${cmds[@]}"
+
+	# Check no processes are left behind
+	check_processes
+
+	wait_time=20
+	sleep_time=2
+
+	# Create server
+	kubectl create -f "${SCRIPT_PATH}/runtimeclass_workloads/latency-server.yaml"
+
+	# Get the names of the server pod
+	export server_pod_name="latency-server"
+
+	# Verify the server pod is working
+	local cmd="kubectl get pod $server_pod_name -o yaml | grep 'phase: Running'"
+	waitForProcess "$wait_time" "$sleep_time" "$cmd"
+
+	# Create client
+	kubectl create -f "${SCRIPT_PATH}/runtimeclass_workloads/latency-client.yaml"
+
+	# Get the names of the client pod
+	export client_pod_name="latency-client"
+
+	# Verify the client pod is working
+	local cmd="kubectl get pod $client_pod_name -o yaml | grep 'phase: Running'"
+	waitForProcess "$wait_time" "$sleep_time" "$cmd"
+
+	# Get the ip address of the server pod
+	export server_ip_add=$(kubectl get pod "$server_pod_name" -o jsonpath='{.status.podIP}')
+
+	# Number of packets (sent)
+	local number="${number:-30}"
+
+	local client_command="ping -c ${number} ${server_ip_add}"
+
+	kubectl exec "$client_pod_name" -- sh -c "$client_command" > "$latency_file"
+
+	metrics_json_init
+
+	local latency=$(cat $latency_file | grep avg | cut -f2 -d'=' | sed 's/[[:blank:]]//g' | cut -f2 -d'/')
+
+	metrics_json_start_array
+
+	local json="$(cat << EOF
+	{
+		"latency": {
+			"Result" : $latency,
+			"Units" : "ms"
+		}
+	}
+EOF
+)"
+
+	metrics_json_add_array_element "$json"
+	metrics_json_end_array "Results"
+	metrics_json_save
+
+	kubectl delete pod "$client_pod_name" "$server_pod_name"
+	check_processes
+}
+main "$@"