Adding script for Performance tests of HV-VES on cloud

Change-Id: If71b0e8b04ce23a5daafef632788667ccd50c1a1
Issue-ID: DCAEGEN2-1664
Signed-off-by: kjaniak <kornel.janiak@nokia.com>

1 files changed, 148 insertions, 0 deletions

diff --git a/tools/performance/cloud/cloud-based-performance-test.sh b/tools/performance/cloud/cloud-based-performance-test.sh
new file mode 100755
index 00000000..95bd10a3
--- /dev/null
+++ b/tools/performance/cloud/cloud-based-performance-test.sh
@@ -0,0 +1,148 @@
+#!/usr/bin/env bash
+
+SCRIPT_DIRECTORY="$(pwd "$0")"
+CONTAINERS_COUNT=1
+PROPERTIES_FILE=${SCRIPT_DIRECTORY}/test.properties
+CONFIG_MAP_NAME=performance-test-config
+PRODUCER_APPS_LABEL=hv-collector-producer
+CONSUMER_APPS_LABEL=hv-collector-kafka-consumer
+ONAP_NAMESPACE=onap
+MAXIMUM_BACK_OFF_CHECK_ITERATIONS=30
+CHECK_NUMBER=0
+NAME_REASON_PATTERN="custom-columns=NAME:.metadata.name,REASON:.status.containerStatuses[].state.waiting.reason"
+
+function clean() {
+    echo "Cleaning up environment"
+
+    echo "Attempting to delete ConfigMap"
+    kubectl delete configmap ${CONFIG_MAP_NAME} -n ${ONAP_NAMESPACE}
+
+    echo "Attempting to delete consumer deployments"
+    kubectl delete deployments -l app=${CONSUMER_APPS_LABEL} -n ${ONAP_NAMESPACE}
+
+    echo "Attempting to delete producer pods"
+    kubectl delete pods -l app=${PRODUCER_APPS_LABEL} -n ${ONAP_NAMESPACE}
+
+    echo "Environment clean up finished!"
+}
+
+function create_producers() {
+    set -e
+    for i in $(seq 1 ${CONTAINERS_COUNT});
+    do
+        echo "Creating ${i}/${CONTAINERS_COUNT} producer"
+        kubectl create -f producer-pod.yaml -n ${ONAP_NAMESPACE}
+    done
+    echo "Producers created"
+    set +e
+}
+
+function usage() {
+    echo ""
+    echo "Run cloud based HV-VES performance test"
+    echo "Usage $0 setup|start|clean|help"
+    echo "  setup    : set up ConfigMap and consumers"
+    echo "  start    : create producers - start the performance test"
+    echo "    Optional parameters:"
+    echo "      --containers      : number of producer containers to create (1)"
+    echo "      --properties-file : path to file with benchmark properties (./test.properties)"
+    echo "  clean    : remove ConfigMap, HV-VES consumers and producers"
+    echo "  help     : print usage"
+    echo "Example invocations:"
+    echo "./cloud-based-performance-test.sh setup"
+    echo "./cloud-based-performance-test.sh start"
+    echo "./cloud-based-performance-test.sh start --containers 10"
+    echo "./cloud-based-performance-test.sh start --containers 10"
+    echo "./cloud-based-performance-test.sh clean"
+    exit 1
+}
+
+function setup_environment() {
+    echo "Setting up environment"
+    echo "Creating ConfigMap from: $PROPERTIES_FILE"
+    kubectl create configmap ${CONFIG_MAP_NAME} --from-env-file=${PROPERTIES_FILE} -n ${ONAP_NAMESPACE}
+
+    echo "Creating consumer deployment"
+    kubectl apply -f consumer-deployment.yaml -n ${ONAP_NAMESPACE}
+
+    echo "Waiting for consumers to be running."
+    while [[ $(kubectl get pods -l app=${CONSUMER_APPS_LABEL} -n ${ONAP_NAMESPACE} | grep -c "unhealthy\|starting") -ne 0 ]] ; do
+        sleep 1
+    done
+    echo "Setting up environment finished!"
+}
+
+function start_performance_test() {
+    echo "Starting cloud based performance tests"
+    echo "________________________________________"
+    echo "Test configuration:"
+    echo "Producer containers count: ${CONTAINERS_COUNT}"
+    echo "Properties file path: ${PROPERTIES_FILE}"
+    echo "________________________________________"
+
+    create_producers
+
+    echo "Waiting for producers completion"
+    while :; do
+        COMPLETED_PRODUCERS=$(kubectl get pods -l app=${PRODUCER_APPS_LABEL} -n ${ONAP_NAMESPACE} | grep -c "Completed")
+        IMAGE_PULL_BACK_OFFS=$(kubectl get pods -l app=${PRODUCER_APPS_LABEL} -n ${ONAP_NAMESPACE} -o ${NAME_REASON_PATTERN} | grep -c "ImagePullBackOff \| ErrImagePull")
+
+        if [[ ${IMAGE_PULL_BACK_OFFS} -gt 0 ]]; then
+            CHECK_NUMBER=$((CHECK_NUMBER + 1))
+            if [[ ${CHECK_NUMBER} -gt ${MAXIMUM_BACK_OFF_CHECK_ITERATIONS} ]]; then
+                echo "Error: Image pull problem"
+                exit 1
+            fi
+        fi
+        
+        [[ ${COMPLETED_PRODUCERS} -eq ${CONTAINERS_COUNT} || ${CHECK_NUMBER} -gt ${MAXIMUM_BACK_OFF_CHECK_ITERATIONS} ]] && break
+        sleep 1
+    done
+
+    echo "Performance test finished"
+    exit 0
+}
+
+cd ${SCRIPT_DIRECTORY}
+
+if [[ $# -eq 0 ]]; then
+    usage
+else
+    for arg in ${@}
+    do
+        case ${arg} in
+            setup)
+            setup_environment
+            ;;
+            start)
+            shift 1
+            while [[ $(($#)) -gt 0 ]]; do
+                case "${1}" in
+                    --containers)
+                        CONTAINERS_COUNT=${2}
+                        ;;
+                    --properties-file)
+                        PROPERTIES_FILE=${2}
+                        ;;
+                    *)
+                        echo "Unknown option: ${1}"
+                        usage
+                        ;;
+                esac
+                shift 2
+            done
+            start_performance_test
+            ;;
+            clean)
+            clean
+            ;;
+            help)
+            usage
+            ;;
+            *)
+            echo "Unknown action: ${arg}" >&2
+            usage
+            ;;
+        esac
+    done
+fi
\ No newline at end of file