test/common/testcase_common.sh

   1 #!/bin/bash
   2
   3 #  ============LICENSE_START===============================================
   4 #  Copyright (C) 2020 Nordix Foundation. All rights reserved.
   5 #  ========================================================================
   6 #  Licensed under the Apache License, Version 2.0 (the "License");
   7 #  you may not use this file except in compliance with the License.
   8 #  You may obtain a copy of the License at
   9 #
  10 #       http://www.apache.org/licenses/LICENSE-2.0
  11 #
  12 #  Unless required by applicable law or agreed to in writing, software
  13 #  distributed under the License is distributed on an "AS IS" BASIS,
  14 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15 #  See the License for the specific language governing permissions and
  16 #  limitations under the License.
  17 #  ============LICENSE_END=================================================
  18 #
  19
  20 # This is a script that contains all the common functions needed for auto test.
  21 # Specific test function are defined in scripts  XXXX_functions.sh
  22
  23 . ../common/api_curl.sh
  24 . ../common/testengine_config.sh
  25
  26 __print_args() {
  27         echo "Args: remote|remote-remove docker|kube --env-file <environment-filename> [release] [auto-clean] [--stop-at-error] "
  28         echo "      [--ricsim-prefix <prefix> ] [--use-local-image <app-nam>+]  [--use-snapshot-image <app-nam>+]"
  29         echo "      [--use-staging-image <app-nam>+] [--use-release-image <app-nam>+] [--image-repo <repo-address]"
  30         echo "      [--repo-policy local|remote] [--cluster-timeout <timeout-in seconds>] [--print-stats]"
  31 }
  32
  33 if [ $# -eq 1 ] && [ "$1" == "help" ]; then
  34
  35         if [ ! -z "$TC_ONELINE_DESCR" ]; then
  36                 echo "Test script description:"
  37                 echo $TC_ONELINE_DESCR
  38                 echo ""
  39         fi
  40         __print_args
  41         echo ""
  42         echo "remote                -  Use images from remote repositories. Can be overridden for individual images using the '--use_xxx' flags"
  43         echo "remote-remove         -  Same as 'remote' but will also try to pull fresh images from remote repositories"
  44         echo "docker                -  Test executed in docker environment"
  45         echo "kube                  -  Test executed in kubernetes environment - requires an already started kubernetes environment"
  46         echo "--env-file            -  The script will use the supplied file to read environment variables from"
  47         echo "release               -  If this flag is given the script will use release version of the images"
  48         echo "auto-clean            -  If the function 'auto_clean_containers' is present in the end of the test script then all containers will be stopped and removed. If 'auto-clean' is not given then the function has no effect."
  49     echo "--stop-at-error       -  The script will stop when the first failed test or configuration"
  50         echo "--ricsim-prefix       -  The a1 simulator will use the supplied string as container prefix instead of 'ricsim'"
  51         echo "--use-local-image     -  The script will use local images for the supplied apps, space separated list of app short names"
  52         echo "--use-snapshot-image  -  The script will use images from the nexus snapshot repo for the supplied apps, space separated list of app short names"
  53         echo "--use-staging-image   -  The script will use images from the nexus staging repo for the supplied apps, space separated list of app short names"
  54         echo "--use-release-image   -  The script will use images from the nexus release repo for the supplied apps, space separated list of app short names"
  55         echo "--image-repo          -  Url to optional image repo. Only locally built images will be re-tagged and pushed to this repo"
  56         echo "--repo-policy         -  Policy controlling which images to re-tag and push if param --image-repo is set. Default is 'local'"
  57         echo "--cluster-timeout     -  Optional timeout for cluster where it takes time to obtain external ip/host-name. Timeout in seconds. "
  58         echo "--print-stats         -  Print current test stats after each test."
  59
  60         echo ""
  61         echo "List of app short names supported: "$APP_SHORT_NAMES
  62         exit 0
  63 fi
  64
  65 AUTOTEST_HOME=$PWD
  66 # Create a test case id, ATC (Auto Test Case), from the name of the test case script.
  67 # FTC1.sh -> ATC == FTC1
  68 ATC=$(basename "${BASH_SOURCE[$i+1]}" .sh)
  69
  70 #Create result file (containing '1' for error) for this test case
  71 #Will be replaced with a file containing '0' if all test cases pass
  72 echo "1" > "$PWD/.result$ATC.txt"
  73
  74 #Formatting for 'echo' cmd
  75 BOLD="\033[1m"
  76 EBOLD="\033[0m"
  77 RED="\033[31m\033[1m"
  78 ERED="\033[0m"
  79 GREEN="\033[32m\033[1m"
  80 EGREEN="\033[0m"
  81 YELLOW="\033[33m\033[1m"
  82 EYELLOW="\033[0m"
  83 SAMELINE="\033[0K\r"
  84
  85 # Just resetting any previous echo formatting...
  86 echo -ne $EBOLD
  87
  88 # default test environment variables
  89 TEST_ENV_VAR_FILE=""
  90
  91 echo "Test case started as: ${BASH_SOURCE[$i+1]} "$@
  92
  93 #Localhost constants
  94 LOCALHOST_NAME="localhost"
  95 LOCALHOST_HTTP="http://localhost"
  96 LOCALHOST_HTTPS="https://localhost"
  97
  98 # Var to hold 'auto' in case containers shall be stopped when test case ends
  99 AUTO_CLEAN=""
 100
 101 # Var to hold the app names to use local images for
 102 USE_LOCAL_IMAGES=""
 103
 104 # Var to hold the app names to use remote snapshot images for
 105 USE_SNAPSHOT_IMAGES=""
 106
 107 # Var to hold the app names to use remote staging images for
 108 USE_STAGING_IMAGES=""
 109
 110 # Var to hold the app names to use remote release images for
 111 USE_RELEASE_IMAGES=""
 112
 113
 114 # Use this var (STOP_AT_ERROR=1 in the test script) for debugging/trouble shooting to take all logs and exit at first FAIL test case
 115 STOP_AT_ERROR=0
 116
 117 # The default value "DEV" indicate that development image tags (SNAPSHOT) and nexus repos (nexus port 10002) are used.
 118 # The value "RELEASE" indicate that relase image tag and nexus repos (nexus port) are used
 119 # Applies only to images defined in the test-env files with image names and tags defined as XXXX_RELEASE
 120 IMAGE_CATEGORY="DEV"
 121
 122 # Function to indent cmd output with one space
 123 indent1() { sed 's/^/ /'; }
 124
 125 # Function to indent cmd output with two spaces
 126 indent2() { sed 's/^/  /'; }
 127
 128 # Set a description string for the test case
 129 if [ -z "$TC_ONELINE_DESCR" ]; then
 130         TC_ONELINE_DESCR="<no-description>"
 131         echo "No test case description found, TC_ONELINE_DESCR should be set on in the test script , using "$TC_ONELINE_DESCR
 132 fi
 133
 134 # Counter for test suites
 135 if [ -f .tmp_tcsuite_ctr ]; then
 136         tmpval=$(< .tmp_tcsuite_ctr)
 137         ((tmpval++))
 138         echo $tmpval > .tmp_tcsuite_ctr
 139 fi
 140
 141 # Create the logs dir if not already created in the current dir
 142 if [ ! -d "logs" ]; then
 143     mkdir logs
 144 fi
 145 TESTLOGS=$PWD/logs
 146
 147 # Create the tmp dir for temporary files that is not needed after the test
 148 # hidden files for the test env is still stored in the current dir
 149 # files in the ./tmp is moved to ./tmp/prev when a new test is started
 150 if [ ! -d "tmp" ]; then
 151     mkdir tmp
 152 fi
 153 curdir=$PWD
 154 cd tmp
 155 if [ $? -ne 0 ]; then
 156         echo "Cannot cd to $PWD/tmp"
 157         echo "Dir cannot be created. Exiting...."
 158 fi
 159 if [ ! -d "prev" ]; then
 160     mkdir prev
 161 fi
 162 cd $curdir
 163 mv ./tmp/* ./tmp/prev 2> /dev/null
 164
 165 # Create a http message log for this testcase
 166 HTTPLOG=$PWD"/.httplog_"$ATC".txt"
 167 echo "" > $HTTPLOG
 168
 169 # Create a log dir for the test case
 170 mkdir -p $TESTLOGS/$ATC
 171
 172 # Save create for current logs
 173 mkdir -p $TESTLOGS/$ATC/previous
 174
 175 rm $TESTLOGS/$ATC/previous/*.log &> /dev/null
 176 rm $TESTLOGS/$ATC/previous/*.txt &> /dev/null
 177 rm $TESTLOGS/$ATC/previous/*.json &> /dev/null
 178
 179 mv  $TESTLOGS/$ATC/*.log $TESTLOGS/$ATC/previous &> /dev/null
 180 mv  $TESTLOGS/$ATC/*.txt $TESTLOGS/$ATC/previous &> /dev/null
 181 mv  $TESTLOGS/$ATC/*.txt $TESTLOGS/$ATC/previous &> /dev/null
 182
 183 # Clear the log dir for the test case
 184 rm $TESTLOGS/$ATC/*.log &> /dev/null
 185 rm $TESTLOGS/$ATC/*.txt &> /dev/null
 186 rm $TESTLOGS/$ATC/*.json &> /dev/null
 187
 188 # Log all output from the test case to a TC log
 189 TCLOG=$TESTLOGS/$ATC/TC.log
 190 exec &>  >(tee ${TCLOG})
 191
 192 #Variables for counting tests as well as passed and failed tests
 193 RES_TEST=0
 194 RES_PASS=0
 195 RES_FAIL=0
 196 RES_CONF_FAIL=0
 197 RES_DEVIATION=0
 198
 199 #Var to control if current stats shall be printed
 200 PRINT_CURRENT_STATS=0
 201
 202 #File to keep deviation messages
 203 DEVIATION_FILE=".tmp_deviations"
 204 rm $DEVIATION_FILE &> /dev/null
 205
 206 # Trap "command not found" and make the script fail
 207 trap_fnc() {
 208
 209         if [ $? -eq 127 ]; then
 210                 echo -e $RED"Function not found, setting script to FAIL"$ERED
 211                 ((RES_CONF_FAIL++))
 212                 __print_current_stats
 213         fi
 214 }
 215 trap trap_fnc ERR
 216
 217 # Counter for tests
 218 TEST_SEQUENCE_NR=1
 219
 220 # Function to log the start of a test case
 221 __log_test_start() {
 222         TIMESTAMP=$(date "+%Y-%m-%d %H:%M:%S")
 223         echo -e $BOLD"TEST $TEST_SEQUENCE_NR (${BASH_LINENO[1]}): ${FUNCNAME[1]}" $@ $EBOLD
 224     echo "TEST $TEST_SEQUENCE_NR - ${TIMESTAMP}: (${BASH_LINENO[1]}): ${FUNCNAME[1]}" $@ >> $HTTPLOG
 225         ((RES_TEST++))
 226         ((TEST_SEQUENCE_NR++))
 227 }
 228
 229 # Function to print current statistics
 230 __print_current_stats() {
 231         if [ $PRINT_CURRENT_STATS -ne 0 ]; then
 232                 echo " Currrent stats - tests, passes, fails, conf fails, deviations: $RES_TEST, $RES_PASS, $RES_FAIL, $RES_CONF_FAIL, $RES_DEVIATION"
 233         fi
 234 }
 235
 236 # General function to log a failed test case
 237 __log_test_fail_general() {
 238         echo -e $RED" FAIL."$1 $ERED
 239         ((RES_FAIL++))
 240         __print_current_stats
 241         __check_stop_at_error
 242 }
 243
 244 # Function to log a test case failed due to incorrect response code
 245 __log_test_fail_status_code() {
 246         echo -e $RED" FAIL. Exepected status "$1", got "$2 $3 $ERED
 247         ((RES_FAIL++))
 248         __print_current_stats
 249         __check_stop_at_error
 250 }
 251
 252 # Function to log a test case failed due to incorrect response body
 253 __log_test_fail_body() {
 254         echo -e $RED" FAIL, returned body not correct"$ERED
 255         ((RES_FAIL++))
 256         __print_current_stats
 257         __check_stop_at_error
 258 }
 259
 260 # Function to log a test case that is not supported
 261 __log_test_fail_not_supported() {
 262         echo -e $RED" FAIL, function not supported"$ERED
 263         ((RES_FAIL++))
 264         __print_current_stats
 265         __check_stop_at_error
 266 }
 267
 268 # General function to log a passed test case
 269 __log_test_pass() {
 270         if [ $# -gt 0 ]; then
 271                 echo $@
 272         fi
 273         ((RES_PASS++))
 274         echo -e $GREEN" PASS"$EGREEN
 275         __print_current_stats
 276 }
 277
 278 #Counter for configurations
 279 CONF_SEQUENCE_NR=1
 280
 281 # Function to log the start of a configuration setup
 282 __log_conf_start() {
 283         TIMESTAMP=$(date "+%Y-%m-%d %H:%M:%S")
 284         echo -e $BOLD"CONF $CONF_SEQUENCE_NR (${BASH_LINENO[1]}): "${FUNCNAME[1]} $@ $EBOLD
 285         echo "CONF $CONF_SEQUENCE_NR - ${TIMESTAMP}: (${BASH_LINENO[1]}): "${FUNCNAME[1]} $@  >> $HTTPLOG
 286         ((CONF_SEQUENCE_NR++))
 287 }
 288
 289 # Function to log a failed configuration setup
 290 __log_conf_fail_general() {
 291         echo -e $RED" FAIL."$1 $ERED
 292         ((RES_CONF_FAIL++))
 293         __print_current_stats
 294         __check_stop_at_error
 295 }
 296
 297 # Function to log a failed configuration setup due to incorrect response code
 298 __log_conf_fail_status_code() {
 299         echo -e $RED" FAIL. Exepected status "$1", got "$2 $3 $ERED
 300         ((RES_CONF_FAIL++))
 301         __print_current_stats
 302         __check_stop_at_error
 303 }
 304
 305 # Function to log a failed configuration setup due to incorrect response body
 306 __log_conf_fail_body() {
 307         echo -e $RED" FAIL, returned body not correct"$ERED
 308         ((RES_CONF_FAIL++))
 309         __print_current_stats
 310         __check_stop_at_error
 311 }
 312
 313 # Function to log a passed configuration setup
 314 __log_conf_ok() {
 315         if [ $# -gt 0 ]; then
 316                 echo $@
 317         fi
 318         echo -e $GREEN" OK"$EGREEN
 319         __print_current_stats
 320 }
 321
 322 #Var for measuring execution time
 323 TCTEST_START=$SECONDS
 324
 325 #File to save timer measurement results
 326 TIMER_MEASUREMENTS=".timer_measurement.txt"
 327 echo -e "Activity \t Duration" > $TIMER_MEASUREMENTS
 328
 329 # If this is set, some images (control by the parameter repo-polcy) will be re-tagged and pushed to this repo before any
 330 IMAGE_REPO_ADR=""
 331 IMAGE_REPO_POLICY="local"
 332 CLUSTER_TIME_OUT=0
 333
 334 echo "-------------------------------------------------------------------------------------------------"
 335 echo "-----------------------------------      Test case: "$ATC
 336 echo "-----------------------------------      Started:   "$(date)
 337 echo "-------------------------------------------------------------------------------------------------"
 338 echo "-- Description: "$TC_ONELINE_DESCR
 339 echo "-------------------------------------------------------------------------------------------------"
 340 echo "-----------------------------------      Test case setup      -----------------------------------"
 341
 342 echo "Setting AUTOTEST_HOME="$AUTOTEST_HOME
 343 START_ARG=$1
 344 paramerror=0
 345 paramerror_str=""
 346 if [ $# -lt 1 ]; then
 347         paramerror=1
 348 fi
 349 if [ $paramerror -eq 0 ]; then
 350         if [ "$1" != "remote" ] && [ "$1" != "remote-remove" ]; then
 351                 paramerror=1
 352                 if [ -z "$paramerror_str" ]; then
 353                         paramerror_str="First arg shall be 'remote' or 'remote-remove'"
 354                 fi
 355         else
 356                 shift;
 357         fi
 358 fi
 359 if [ $paramerror -eq 0 ]; then
 360         if [ "$1" != "docker" ] && [ "$1" != "kube" ]; then
 361                 paramerror=1
 362                 if [ -z "$paramerror_str" ]; then
 363                         paramerror_str="Second arg shall be 'docker' or 'kube'"
 364                 fi
 365         else
 366                 if [ $1 == "docker" ]; then
 367                         RUNMODE="DOCKER"
 368                         echo "Setting RUNMODE=DOCKER"
 369                 fi
 370                 if [ $1 == "kube" ]; then
 371                         RUNMODE="KUBE"
 372                         echo "Setting RUNMODE=KUBE"
 373                 fi
 374                 shift;
 375         fi
 376 fi
 377 foundparm=0
 378 while [ $paramerror -eq 0 ] && [ $foundparm -eq 0 ]; do
 379         foundparm=1
 380         if [ $paramerror -eq 0 ]; then
 381                 if [ "$1" == "release" ]; then
 382                         IMAGE_CATEGORY="RELEASE"
 383                         echo "Option set - Release image tags used for applicable images "
 384                         shift;
 385                         foundparm=0
 386                 fi
 387         fi
 388         if [ $paramerror -eq 0 ]; then
 389                 if [ "$1" == "auto-clean" ]; then
 390                         AUTO_CLEAN="auto"
 391                         echo "Option set - Auto clean at end of test script"
 392                         shift;
 393                         foundparm=0
 394                 fi
 395         fi
 396         if [ $paramerror -eq 0 ]; then
 397                 if [ "$1" == "--stop-at-error" ]; then
 398                         STOP_AT_ERROR=1
 399                         echo "Option set - Stop at first error"
 400                         shift;
 401                         foundparm=0
 402                 fi
 403         fi
 404         if [ $paramerror -eq 0 ]; then
 405                 if [ "$1" == "--ricsim-prefix" ]; then
 406                         shift;
 407                         TMP_RIC_SIM_PREFIX=$1  #RIC_SIM_PREFIX need to be updated after sourcing of the env file
 408                         if [ -z "$1" ]; then
 409                                 paramerror=1
 410                                 if [ -z "$paramerror_str" ]; then
 411                                         paramerror_str="No prefix found for flag: '--ricsim-prefix'"
 412                                 fi
 413                         else
 414                                 echo "Option set - Overriding RIC_SIM_PREFIX with: "$1
 415                                 shift;
 416                                 foundparm=0
 417                         fi
 418                 fi
 419         fi
 420         if [ $paramerror -eq 0 ]; then
 421                 if [ "$1" == "--env-file" ]; then
 422                         shift;
 423                         TEST_ENV_VAR_FILE=$1
 424                         if [ -z "$1" ]; then
 425                                 paramerror=1
 426                                 if [ -z "$paramerror_str" ]; then
 427                                         paramerror_str="No env file found for flag: '--env-file'"
 428                                 fi
 429                         else
 430                                 echo "Option set - Reading test env from: "$1
 431                                 shift;
 432                                 foundparm=0
 433                         fi
 434                 fi
 435         fi
 436         if [ $paramerror -eq 0 ]; then
 437                 if [ "$1" == "--use-local-image" ]; then
 438                         USE_LOCAL_IMAGES=""
 439                         shift
 440                         while [ $# -gt 0 ] && [[ "$1" != "--"* ]]; do
 441                                 USE_LOCAL_IMAGES=$USE_LOCAL_IMAGES" "$1
 442                                 if [[ "$AVAILABLE_IMAGES_OVERRIDE" != *"$1"* ]]; then
 443                                         paramerror=1
 444                                         if [ -z "$paramerror_str" ]; then
 445                                                 paramerror_str="App name $1 is not available for local override for flag: '--use-local-image'"
 446                                         fi
 447                                 fi
 448                                 shift;
 449                         done
 450                         foundparm=0
 451                         if [ -z "$USE_LOCAL_IMAGES" ]; then
 452                                 paramerror=1
 453                                 if [ -z "$paramerror_str" ]; then
 454                                         paramerror_str="No app name found for flag: '--use-local-image'"
 455                                 fi
 456                         else
 457                                 echo "Option set - Overriding with local images for app(s):"$USE_LOCAL_IMAGES
 458                         fi
 459                 fi
 460         fi
 461         if [ $paramerror -eq 0 ]; then
 462                 if [ "$1" == "--use-snapshot-image" ]; then
 463                         USE_SNAPSHOT_IMAGES=""
 464                         shift
 465                         while [ $# -gt 0 ] && [[ "$1" != "--"* ]]; do
 466                                 USE_SNAPSHOT_IMAGES=$USE_SNAPSHOT_IMAGES" "$1
 467                                 if [[ "$AVAILABLE_IMAGES_OVERRIDE" != *"$1"* ]]; then
 468                                         paramerror=1
 469                                         if [ -z "$paramerror_str" ]; then
 470                                                 paramerror_str="App name $1 is not available for snapshot override for flag: '--use-snapshot-image'"
 471                                         fi
 472                                 fi
 473                                 shift;
 474                         done
 475                         foundparm=0
 476                         if [ -z "$USE_SNAPSHOT_IMAGES" ]; then
 477                                 paramerror=1
 478                                 if [ -z "$paramerror_str" ]; then
 479                                         paramerror_str="No app name found for flag: '--use-snapshot-image'"
 480                                 fi
 481                         else
 482                                 echo "Option set - Overriding with snapshot images for app(s):"$USE_SNAPSHOT_IMAGES
 483                         fi
 484                 fi
 485         fi
 486         if [ $paramerror -eq 0 ]; then
 487                 if [ "$1" == "--use-staging-image" ]; then
 488                         USE_STAGING_IMAGES=""
 489                         shift
 490                         while [ $# -gt 0 ] && [[ "$1" != "--"* ]]; do
 491                                 USE_STAGING_IMAGES=$USE_STAGING_IMAGES" "$1
 492                                 if [[ "$AVAILABLE_IMAGES_OVERRIDE" != *"$1"* ]]; then
 493                                         paramerror=1
 494                                         if [ -z "$paramerror_str" ]; then
 495                                                 paramerror_str="App name $1 is not available for staging override for flag: '--use-staging-image'"
 496                                         fi
 497                                 fi
 498                                 shift;
 499                         done
 500                         foundparm=0
 501                         if [ -z "$USE_STAGING_IMAGES" ]; then
 502                                 paramerror=1
 503                                 if [ -z "$paramerror_str" ]; then
 504                                         paramerror_str="No app name found for flag: '--use-staging-image'"
 505                                 fi
 506                         else
 507                                 echo "Option set - Overriding with staging images for app(s):"$USE_STAGING_IMAGES
 508                         fi
 509                 fi
 510         fi
 511         if [ $paramerror -eq 0 ]; then
 512                 if [ "$1" == "--use-release-image" ]; then
 513                         USE_RELEASE_IMAGES=""
 514                         shift
 515                         while [ $# -gt 0 ] && [[ "$1" != "--"* ]]; do
 516                                 USE_RELEASE_IMAGES=$USE_RELEASE_IMAGES" "$1
 517                                 if [[ "$AVAILABLE_IMAGES_OVERRIDE" != *"$1"* ]]; then
 518                                         paramerror=1
 519                                         if [ -z "$paramerror_str" ]; then
 520                                                 paramerror_str="App name $1 is not available for release override for flag: '--use-release-image'"
 521                                         fi
 522                                 fi
 523                                 shift;
 524                         done
 525                         foundparm=0
 526                         if [ -z "$USE_RELEASE_IMAGES" ]; then
 527                                 paramerror=1
 528                                 if [ -z "$paramerror_str" ]; then
 529                                         paramerror_str="No app name found for flag: '--use-release-image'"
 530                                 fi
 531                         else
 532                                 echo "Option set - Overriding with release images for app(s):"$USE_RELEASE_IMAGES
 533                         fi
 534                 fi
 535         fi
 536         if [ $paramerror -eq 0 ]; then
 537                 if [ "$1" == "--image-repo" ]; then
 538                         shift;
 539                         IMAGE_REPO_ADR=$1
 540                         if [ -z "$1" ]; then
 541                                 paramerror=1
 542                                 if [ -z "$paramerror_str" ]; then
 543                                         paramerror_str="No image repo url found for : '--image-repo'"
 544                                 fi
 545                         else
 546                                 echo "Option set - Image repo url: "$1
 547                                 shift;
 548                                 foundparm=0
 549                         fi
 550                 fi
 551         fi
 552         if [ $paramerror -eq 0 ]; then
 553                 if [ "$1" == "--repo-policy" ]; then
 554                         shift;
 555                         IMAGE_REPO_POLICY=$1
 556                         if [ -z "$1" ]; then
 557                                 paramerror=1
 558                                 if [ -z "$paramerror_str" ]; then
 559                                         paramerror_str="No policy found for : '--repo-policy'"
 560                                 fi
 561                         else
 562                             if [ "$1" == "local" ] || [ "$1" == "remote" ]; then
 563                                         echo "Option set - Image repo policy: "$1
 564                                         shift;
 565                                         foundparm=0
 566                                 else
 567                                         paramerror=1
 568                                         if [ -z "$paramerror_str" ]; then
 569                                                 paramerror_str="Repo policy shall be 'local' or 'remote'"
 570                                         fi
 571                                 fi
 572                         fi
 573                 fi
 574         fi
 575         if [ $paramerror -eq 0 ]; then
 576                 if [ "$1" == "--cluster-timeout" ]; then
 577                         shift;
 578                         CLUSTER_TIME_OUT=$1
 579                         if [ -z "$1" ]; then
 580                                 paramerror=1
 581                                 if [ -z "$paramerror_str" ]; then
 582                                         paramerror_str="No timeout value found for : '--cluster-timeout'"
 583                                 fi
 584                         else
 585                                 #Check if positive int
 586                                 case ${CLUSTER_TIME_OUT#[+]} in
 587                                         *[!0-9]* | '')
 588                                                 paramerror=1
 589                                                 if [ -z "$paramerror_str" ]; then
 590                                                         paramerror_str="Value for '--cluster-timeout' not an int : "$CLUSTER_TIME_OUT
 591                                                 fi
 592                                                 ;;
 593                                         * ) ;; # Ok
 594                                 esac
 595                                 echo "Option set - Cluster timeout: "$1
 596                                 shift;
 597                                 foundparm=0
 598                         fi
 599                 fi
 600         fi
 601         if [ $paramerror -eq 0 ]; then
 602                 if [ "$1" == "--print-stats" ]; then
 603                         PRINT_CURRENT_STATS=1
 604                         echo "Option set - Print stats"
 605                         shift;
 606                         foundparm=0
 607                 fi
 608         fi
 609 done
 610 echo ""
 611
 612 #Still params left?
 613 if [ $paramerror -eq 0 ] && [ $# -gt 0 ]; then
 614         paramerror=1
 615         if [ -z "$paramerror_str" ]; then
 616                 paramerror_str="Unknown parameter(s): "$@
 617         fi
 618 fi
 619
 620 if [ $paramerror -eq 1 ]; then
 621         echo -e $RED"Incorrect arg list: "$paramerror_str$ERED
 622         __print_args
 623         exit 1
 624 fi
 625
 626 # sourcing the selected env variables for the test case
 627 if [ -f "$TEST_ENV_VAR_FILE" ]; then
 628         echo -e $BOLD"Sourcing env vars from: "$TEST_ENV_VAR_FILE$EBOLD
 629         . $TEST_ENV_VAR_FILE
 630
 631         if [ -z "$TEST_ENV_PROFILE" ] || [ -z "$SUPPORTED_PROFILES" ]; then
 632                 echo -e $YELLOW"This test case may not work with selected test env file. TEST_ENV_PROFILE is missing in test_env file or SUPPORTED_PROFILES is missing in test case file"$EYELLOW
 633         else
 634                 found_profile=0
 635                 for prof in $SUPPORTED_PROFILES; do
 636                         if [ "$TEST_ENV_PROFILE" == "$prof" ]; then
 637                                 echo -e $GREEN"Test case supports the selected test env file"$EGREEN
 638                                 found_profile=1
 639                         fi
 640                 done
 641                 if [ $found_profile -ne 1 ]; then
 642                         echo -e $RED"Test case does not support the selected test env file"$ERED
 643                         echo "Profile: "$TEST_ENV_PROFILE"     Supported profiles: "$SUPPORTED_PROFILES
 644                         echo -e $RED"Exiting...."$ERED
 645                         exit 1
 646                 fi
 647         fi
 648 else
 649         echo -e $RED"Selected env var file does not exist: "$TEST_ENV_VAR_FILE$ERED
 650         echo " Select one of following env var file matching the intended target of the test"
 651         echo " Restart the test using the flag '--env-file <path-to-env-file>"
 652         ls $AUTOTEST_HOME/../common/test_env* | indent1
 653         exit 1
 654 fi
 655
 656 #This var need be preserved from the command line option, if set, when env var is sourced.
 657 if [ ! -z "$TMP_RIC_SIM_PREFIX" ]; then
 658         RIC_SIM_PREFIX=$TMP_RIC_SIM_PREFIX
 659 fi
 660
 661 if [ -z "$PROJECT_IMAGES_APP_NAMES" ]; then
 662         echo -e $RED"Var PROJECT_IMAGES_APP_NAMES must be defined in: "$TEST_ENV_VAR_FILE $ERED
 663         exit 1
 664 fi
 665
 666 if [[ $SUPPORTED_RUNMODES != *"$RUNMODE"* ]]; then
 667         echo -e $RED"This test script does not support RUNMODE $RUNMODE"$ERED
 668         echo "Supported RUNMODEs: "$SUPPORTED_RUNMODES
 669         exit 1
 670 fi
 671
 672 # Choose list of included apps depending on run-mode
 673 if [ $RUNMODE == "KUBE" ]; then
 674         INCLUDED_IMAGES=$KUBE_INCLUDED_IMAGES
 675 else
 676         INCLUDED_IMAGES=$DOCKER_INCLUDED_IMAGES
 677 fi
 678
 679 # Check needed installed sw
 680 tmp=$(which python3)
 681 if [ $? -ne 0 ] || [ -z tmp ]; then
 682         echo -e $RED"python3 is required to run the test environment, pls install"$ERED
 683         exit 1
 684 fi
 685 tmp=$(which docker)
 686 if [ $? -ne 0 ] || [ -z tmp ]; then
 687         echo -e $RED"docker is required to run the test environment, pls install"$ERED
 688         exit 1
 689 fi
 690
 691 tmp=$(which docker-compose)
 692 if [ $? -ne 0 ] || [ -z tmp ]; then
 693         if [ $RUNMODE == "DOCKER" ]; then
 694                 echo -e $RED"docker-compose is required to run the test environment, pls install"$ERED
 695                 exit 1
 696         fi
 697 fi
 698
 699 tmp=$(which kubectl)
 700 if [ $? -ne 0 ] || [ -z tmp ]; then
 701         if [ $RUNMODE == "KUBE" ]; then
 702                 echo -e $RED"kubectl is required to run the test environment in kubernetes mode, pls install"$ERED
 703                 exit 1
 704         fi
 705 else
 706         if [ $RUNMODE == "KUBE" ]; then
 707                 res=$(kubectl cluster-info 2>&1)
 708                 if [ $? -ne 0 ]; then
 709                         echo -e "$BOLD$RED############################################# $ERED$EBOLD"
 710                         echo -e  $BOLD$RED"Command 'kubectl cluster-info' returned error $ERED$EBOLD"
 711                         echo -e "$BOLD$RED############################################# $ERED$EBOLD"
 712                         echo " "
 713                         echo "kubectl response:"
 714                         echo $res
 715                         echo " "
 716                         echo "This script may have been started with user with no permission to run kubectl"
 717                         echo "Try running with 'sudo' or set 'KUBECONFIG'"
 718                         echo "Do either 1, 2 or 3 "
 719                         echo " "
 720                         echo "1"
 721                         echo "Run with sudo"
 722                         echo -e $BOLD"sudo <test-script-and-parameters>"$EBOLD
 723                         echo " "
 724                         echo "2"
 725                         echo "Export KUBECONFIG and pass env to sudo - (replace user)"
 726                         echo -e $BOLD"export KUBECONFIG='/home/<user>/.kube/config'"$EBOLD
 727                         echo -e $BOLD"sudo -E <test-script-and-parameters>"$EBOLD
 728                         echo " "
 729                         echo "3"
 730                         echo "Set KUBECONFIG inline (replace user)"
 731                         echo -e $BOLD"sudo  KUBECONFIG='/home/<user>/.kube/config' <test-script-and-parameters>"$EBOLD
 732
 733                         exit 1
 734                 fi
 735         fi
 736 fi
 737
 738 echo -e $BOLD"Checking configured image setting for this test case"$EBOLD
 739
 740 #Temp var to check for image variable name errors
 741 IMAGE_ERR=0
 742 #Create a file with image info for later printing as a table
 743 image_list_file="./tmp/.image-list"
 744 echo -e "Application\tApp short name\tImage\ttag\ttag-switch" > $image_list_file
 745
 746 # Check if image env var is set and if so export the env var with image to use (used by docker compose files)
 747 # arg: <app-short-name> <target-variable-name> <image-variable-name> <image-tag-variable-name> <tag-suffix> <image name>
 748 __check_and_create_image_var() {
 749
 750         if [ $# -ne 6 ]; then
 751                 echo "Expected arg: <app-short-name> <target-variable-name> <image-variable-name> <image-tag-variable-name> <tag-suffix> <image name>"
 752                 ((IMAGE_ERR++))
 753                 return
 754         fi
 755
 756         __check_included_image $1
 757         if [ $? -ne 0 ]; then
 758                 echo -e "$6\t$1\t<image-excluded>\t<no-tag>"  >> $image_list_file
 759                 # Image is excluded since the corresponding app is not used in this test
 760                 return
 761         fi
 762         tmp=${6}"\t"${1}"\t"
 763         #Create var from the input var names
 764         image="${!3}"
 765         tmptag=$4"_"$5
 766         tag="${!tmptag}"
 767
 768         optional_image_repo_target=""
 769
 770         if [ -z $image ]; then
 771                 __check_ignore_image $1
 772                 if [ $? -eq 0 ]; then
 773                         app_ds=$6
 774                         if [ -z "$6" ]; then
 775                                 app_ds="<app ignored>"
 776                         fi
 777                         echo -e "$app_ds\t$1\t<image-ignored>\t<no-tag>"  >> $image_list_file
 778                         # Image is ignored since the corresponding the images is not set in the env file
 779                         __remove_included_image $1   # Remove the image from the list of included images
 780                         return
 781                 fi
 782                 echo -e $RED"\$"$3" not set in $TEST_ENV_VAR_FILE"$ERED
 783                 ((IMAGE_ERR++))
 784                 echo ""
 785                 tmp=$tmp"<no-image>\t"
 786         else
 787
 788                 optional_image_repo_target=$image
 789
 790                 #Add repo depending on image type
 791                 if [ "$5" == "REMOTE_RELEASE" ]; then
 792                         image=$NEXUS_RELEASE_REPO$image
 793                 fi
 794                 if [ "$5" == "REMOTE" ]; then
 795                         image=$NEXUS_STAGING_REPO$image
 796                 fi
 797                 if [ "$5" == "REMOTE_SNAPSHOT" ]; then
 798                         image=$NEXUS_SNAPSHOT_REPO$image
 799                 fi
 800                 if [ "$5" == "REMOTE_PROXY" ]; then
 801                         image=$NEXUS_PROXY_REPO$image
 802                 fi
 803                 if [ "$5" == "REMOTE_RELEASE_ONAP" ]; then
 804                         image=$NEXUS_RELEASE_REPO_ONAP$image
 805                 fi
 806                 if [ "$5" == "REMOTE_RELEASE_ORAN" ]; then
 807                         image=$NEXUS_RELEASE_REPO_ORAN$image
 808                 fi
 809                 #No nexus repo added for local images, tag: LOCAL
 810                 tmp=$tmp$image"\t"
 811         fi
 812         if [ -z $tag ]; then
 813                 echo -e $RED"\$"$tmptag" not set in $TEST_ENV_VAR_FILE"$ERED
 814                 ((IMAGE_ERR++))
 815                 echo ""
 816                 tmp=$tmp"<no-tag>\t"
 817         else
 818                 tmp=$tmp$tag
 819         fi
 820         tmp=$tmp"\t"$5
 821         echo -e "$tmp" >> $image_list_file
 822         #Export the env var
 823         export "${2}"=$image":"$tag  #Note, this var may be set to the value of the target value below in __check_and_pull_image
 824
 825         remote_or_local_push=false
 826         if [ ! -z "$IMAGE_REPO_ADR" ] && [[ $5 != *"PROXY"* ]]; then
 827                 if [ $5 == "LOCAL" ]; then
 828                         remote_or_local_push=true
 829                 fi
 830                 if [[ $5 == *"REMOTE"* ]]; then
 831                         if [ "$IMAGE_REPO_POLICY" == "remote" ]; then
 832                                 remote_or_local_push=true
 833                         fi
 834                 fi
 835         fi
 836         if $remote_or_local_push; then    # Only re-tag and push images according to policy, if repo is given
 837                 export "${2}_SOURCE"=$image":"$tag  #Var to keep the actual source image
 838                 if [[ $optional_image_repo_target == *"/"* ]]; then # Replace all / with _ for images to push to external repo
 839                         optional_image_repo_target_tmp=${optional_image_repo_target//\//_}
 840                         optional_image_repo_target=$optional_image_repo_target_tmp
 841                 fi
 842                 export "${2}_TARGET"=$IMAGE_REPO_ADR"/"$optional_image_repo_target":"$tag  #Create image + tag for optional image repo - pushed later if needed
 843         else
 844                 export "${2}_SOURCE"=""
 845                 export "${2}_TARGET"=""
 846         fi
 847 }
 848
 849 # Check if app uses image included in this test run
 850 # Returns 0 if image is included, 1 if not
 851 __check_included_image() {
 852         for im in $INCLUDED_IMAGES; do
 853                 if [ "$1" == "$im" ]; then
 854                         return 0
 855                 fi
 856         done
 857         return 1
 858 }
 859
 860 # Check if app uses a project image
 861 # Returns 0 if image is included, 1 if not
 862 __check_project_image() {
 863         for im in $PROJECT_IMAGES; do
 864                 if [ "$1" == "$im" ]; then
 865                         return 0
 866                 fi
 867         done
 868         return 1
 869 }
 870
 871 # Check if app uses image built by the test script
 872 # Returns 0 if image is included, 1 if not
 873 __check_image_local_build() {
 874         for im in $LOCAL_IMAGE_BUILD; do
 875                 if [ "$1" == "$im" ]; then
 876                         return 0
 877                 fi
 878         done
 879         return 1
 880 }
 881
 882 # Check if app image is conditionally ignored in this test run
 883 # Returns 0 if image is conditionally ignored, 1 if not
 884 __check_ignore_image() {
 885         for im in $CONDITIONALLY_IGNORED_IMAGES; do
 886                 if [ "$1" == "$im" ]; then
 887                         return 0
 888                 fi
 889         done
 890         return 1
 891 }
 892
 893 # Removed image from included list of included images
 894 # Used when an image is marked as conditionally ignored
 895 __remove_included_image() {
 896         tmp_img_rem_list=""
 897         for im in $INCLUDED_IMAGES; do
 898                 if [ "$1" != "$im" ]; then
 899                         tmp_img_rem_list=$tmp_img_rem_list" "$im
 900                 fi
 901         done
 902         INCLUDED_IMAGES=$tmp_img_rem_list
 903         return 0
 904 }
 905
 906 # Check if app is included in the prestarted set of apps
 907 # Returns 0 if image is included, 1 if not
 908 __check_prestarted_image() {
 909         for im in $KUBE_PRESTARTED_IMAGES; do
 910                 if [ "$1" == "$im" ]; then
 911                         return 0
 912                 fi
 913         done
 914         return 1
 915 }
 916
 917 # Check if an app shall use a local image, based on the cmd parameters
 918 __check_image_local_override() {
 919         for im in $USE_LOCAL_IMAGES; do
 920                 if [ "$1" == "$im" ]; then
 921                         return 1
 922                 fi
 923         done
 924         return 0
 925 }
 926
 927 # Check if app uses image override
 928 # Returns the image/tag suffix LOCAL for local image or REMOTE/REMOTE_RELEASE/REMOTE_SNAPSHOT for staging/release/snapshot image
 929 __check_image_override() {
 930
 931         for im in $ORAN_IMAGES_APP_NAMES; do
 932                 if [ "$1" == "$im" ]; then
 933                         echo "REMOTE_RELEASE_ORAN"
 934                         return 0
 935                 fi
 936         done
 937
 938         for im in $ONAP_IMAGES_APP_NAMES; do
 939                 if [ "$1" == "$im" ]; then
 940                         echo "REMOTE_RELEASE_ONAP"
 941                         return 0
 942                 fi
 943         done
 944
 945         found=0
 946         for im in $PROJECT_IMAGES_APP_NAMES; do
 947                 if [ "$1" == "$im" ]; then
 948                         found=1
 949                 fi
 950         done
 951
 952         if [ $found -eq 0 ]; then
 953                 echo "REMOTE_PROXY"
 954                 return 0
 955         fi
 956
 957         suffix=""
 958         if [ $IMAGE_CATEGORY == "RELEASE" ]; then
 959                 suffix="REMOTE_RELEASE"
 960         fi
 961         if [ $IMAGE_CATEGORY == "DEV" ]; then
 962                 suffix="REMOTE"
 963         fi
 964         CTR=0
 965         for im in $USE_STAGING_IMAGES; do
 966                 if [ "$1" == "$im" ]; then
 967                         suffix="REMOTE"
 968                         ((CTR++))
 969                 fi
 970         done
 971         for im in $USE_RELEASE_IMAGES; do
 972                 if [ "$1" == "$im" ]; then
 973                         suffix="REMOTE_RELEASE"
 974                         ((CTR++))
 975                 fi
 976         done
 977         for im in $USE_SNAPSHOT_IMAGES; do
 978                 if [ "$1" == "$im" ]; then
 979                         suffix="REMOTE_SNAPSHOT"
 980                         ((CTR++))
 981                 fi
 982         done
 983         for im in $USE_LOCAL_IMAGES; do
 984                 if [ "$1" == "$im" ]; then
 985                         suffix="LOCAL"
 986                         ((CTR++))
 987                 fi
 988         done
 989         echo $suffix
 990         if [ $CTR -gt 1 ]; then
 991                 exit 1
 992         fi
 993         return 0
 994 }
 995
 996 # Function to re-tag and image and push to another image repo
 997 __retag_and_push_image() {
 998         if [ ! -z "$IMAGE_REPO_ADR" ]; then
 999                 source_image="${!1}"
1000                 trg_var_name=$1_"TARGET" # This var is created in func __check_and_create_image_var
1001                 target_image="${!trg_var_name}"
1002
1003                 if [ -z $target_image ]; then
1004                         return 0  # Image with no target shall not be pushed
1005                 fi
1006
1007                 echo -ne "  Attempt to re-tag image to: ${BOLD}${target_image}${EBOLD}${SAMELINE}"
1008                 tmp=$(docker image tag $source_image ${target_image} )
1009                 if [ $? -ne 0 ]; then
1010                         docker stop $tmp &> ./tmp/.dockererr
1011                         ((IMAGE_ERR++))
1012                         echo ""
1013                         echo -e "  Attempt to re-tag image to: ${BOLD}${target_image}${EBOLD} - ${RED}Failed${ERED}"
1014                         cat ./tmp/.dockererr
1015                         return 1
1016                 else
1017                         echo -e "  Attempt to re-tag image to: ${BOLD}${target_image}${EBOLD} - ${GREEN}OK${EGREEN}"
1018                 fi
1019                 echo -ne "  Attempt to push re-tagged image: ${BOLD}${target_image}${EBOLD}${SAMELINE}"
1020                 tmp=$(docker push ${target_image} )
1021                 if [ $? -ne 0 ]; then
1022                         docker stop $tmp &> ./tmp/.dockererr
1023                         ((IMAGE_ERR++))
1024                         echo ""
1025                         echo -e "  Attempt to push re-tagged image: ${BOLD}${target_image}${EBOLD} - ${RED}Failed${ERED}"
1026                         cat ./tmp/.dockererr
1027                         return 1
1028                 else
1029                         echo -e "  Attempt to push re-tagged image: ${BOLD}${target_image}${EBOLD} - ${GREEN}OK${EGREEN}"
1030                 fi
1031                 export "${1}"=$target_image
1032         fi
1033         return 0
1034 }
1035
1036 #Function to check if image exist and stop+remove the container+pull new images as needed
1037 #args <script-start-arg> <descriptive-image-name> <container-base-name> <image-with-tag-var-name>
1038 __check_and_pull_image() {
1039
1040         source_image="${!4}"
1041
1042         echo -e " Checking $BOLD$2$EBOLD container(s) with basename: $BOLD$3$EBOLD using image: $BOLD$source_image$EBOLD"
1043         format_string="\"{{.Repository}}\\t{{.Tag}}\\t{{.CreatedSince}}\\t{{.Size}}\""
1044         tmp_im=$(docker images --format $format_string $source_image)
1045
1046         if [ $1 == "local" ]; then
1047                 if [ -z "$tmp_im" ]; then
1048                         echo -e "  "$2" (local image): \033[1m"$source_image"\033[0m $RED does not exist in local registry, need to be built (or manually pulled)"$ERED
1049                         ((IMAGE_ERR++))
1050                         return 1
1051                 else
1052                         echo -e "  "$2" (local image): \033[1m"$source_image"\033[0m "$GREEN"OK"$EGREEN
1053                 fi
1054         elif [ $1 == "remote" ] || [ $1 == "remote-remove" ]; then
1055                 if [ $1 == "remote-remove" ]; then
1056                         if [ $RUNMODE == "DOCKER" ]; then
1057                                 echo -ne "  Attempt to stop and remove container(s), if running - ${SAMELINE}"
1058                                 tmp=$(docker ps -aq --filter name=${3} --filter network=${DOCKER_SIM_NWNAME})
1059                                 if [ $? -eq 0 ] && [ ! -z "$tmp" ]; then
1060                                         docker stop $tmp &> ./tmp/.dockererr
1061                                         if [ $? -ne 0 ]; then
1062                                                 ((IMAGE_ERR++))
1063                                                 echo ""
1064                                                 echo -e $RED"  Container(s) could not be stopped - try manual stopping the container(s)"$ERED
1065                                                 cat ./tmp/.dockererr
1066                                                 return 1
1067                                         fi
1068                                 fi
1069                                 echo -ne "  Attempt to stop and remove container(s), if running - "$GREEN"stopped"$EGREEN"${SAMELINE}"
1070                                 tmp=$(docker ps -aq --filter name=${3} --filter network=${DOCKER_SIM_NWNAME}) &> /dev/null
1071                                 if [ $? -eq 0 ] && [ ! -z "$tmp" ]; then
1072                                         docker rm $tmp &> ./tmp/.dockererr
1073                                         if [ $? -ne 0 ]; then
1074                                                 ((IMAGE_ERR++))
1075                                                 echo ""
1076                                                 echo -e $RED"  Container(s) could not be removed - try manual removal of the container(s)"$ERED
1077                                                 cat ./tmp/.dockererr
1078                                                 return 1
1079                                         fi
1080                                 fi
1081                                 echo -e "  Attempt to stop and remove container(s), if running - "$GREEN"stopped removed"$EGREEN
1082                                 tmp_im=""
1083                         else
1084                                 tmp_im=""
1085                         fi
1086                 fi
1087                 if [ -z "$tmp_im" ]; then
1088                         echo -ne "  Pulling image${SAMELINE}"
1089                         out=$(docker pull $source_image)
1090                         if [ $? -ne 0 ]; then
1091                                 echo ""
1092                                 echo -e "  Pulling image -$RED could not be pulled"$ERED
1093                                 ((IMAGE_ERR++))
1094                                 echo $out > ./tmp/.dockererr
1095                                 echo $out
1096                                 return 1
1097                         fi
1098                         echo $out > ./tmp/.dockererr
1099                         if [[ $out == *"up to date"* ]]; then
1100                                 echo -e "  Pulling image -$GREEN Image is up to date $EGREEN"
1101                         elif [[ $out == *"Downloaded newer image"* ]]; then
1102                                 echo -e "  Pulling image -$GREEN Newer image pulled $EGREEN"
1103                         else
1104                                 echo -e "  Pulling image -$GREEN Pulled $EGREEN"
1105                         fi
1106                 else
1107                         echo -e "  Pulling image -$GREEN OK $EGREEN(exists in local repository)"
1108                 fi
1109         fi
1110
1111         __retag_and_push_image $4
1112
1113         return $?
1114 }
1115
1116 setup_testenvironment() {
1117         # Check that image env setting are available
1118         echo ""
1119
1120         # Image var setup for all project images included in the test
1121         for imagename in $APP_SHORT_NAMES; do
1122                 __check_included_image $imagename
1123                 incl=$?
1124                 __check_project_image $imagename
1125                 proj=$?
1126                 if [ $incl -eq 0 ]; then
1127                         if [ $proj -eq 0 ]; then
1128                                 IMAGE_SUFFIX=$(__check_image_override $imagename)
1129                                 if [ $? -ne 0 ]; then
1130                                         echo -e $RED"Image setting from cmd line not consistent for $imagename."$ERED
1131                                         ((IMAGE_ERR++))
1132                                 fi
1133                         else
1134                                 IMAGE_SUFFIX="none"
1135                         fi
1136                         # A function name is created from the app short name
1137                         # for example app short name 'ECS' -> produce the function
1138                         # name __ECS_imagesetup
1139                         # This function is called and is expected to exist in the imported
1140                         # file for the ecs test functions
1141                         # The resulting function impl will call '__check_and_create_image_var' function
1142                         # with appropriate parameters
1143                         # If the image suffix is none, then the component decides the suffix
1144                         function_pointer="__"$imagename"_imagesetup"
1145                         $function_pointer $IMAGE_SUFFIX
1146                 fi
1147         done
1148
1149         #Errors in image setting - exit
1150         if [ $IMAGE_ERR -ne 0 ]; then
1151                 exit 1
1152         fi
1153
1154         #Print a tables of the image settings
1155         echo -e $BOLD"Images configured for start arg: "$START_ARG $EBOLD
1156         column -t -s $'\t' $image_list_file | indent1
1157
1158         echo ""
1159
1160         #Set the SIM_GROUP var
1161         echo -e $BOLD"Setting var to main dir of all container/simulator scripts"$EBOLD
1162         if [ -z "$SIM_GROUP" ]; then
1163                 SIM_GROUP=$AUTOTEST_HOME/../simulator-group
1164                 if [ ! -d  $SIM_GROUP ]; then
1165                         echo "Trying to set env var SIM_GROUP to dir 'simulator-group' in the nontrtric repo, but failed."
1166                         echo -e $RED"Please set the SIM_GROUP manually in the applicable $TEST_ENV_VAR_FILE"$ERED
1167                         exit 1
1168                 else
1169                         echo " SIM_GROUP auto set to: " $SIM_GROUP
1170                 fi
1171         elif [ $SIM_GROUP = *simulator_group ]; then
1172                 echo -e $RED"Env var SIM_GROUP does not seem to point to dir 'simulator-group' in the repo, check $TEST_ENV_VAR_FILE"$ERED
1173                 exit 1
1174         else
1175                 echo " SIM_GROUP env var already set to: " $SIM_GROUP
1176         fi
1177
1178         echo ""
1179
1180         #Temp var to check for image pull errors
1181         IMAGE_ERR=0
1182
1183         # The following sequence pull the configured images
1184
1185
1186         echo -e $BOLD"Pulling configured images, if needed"$EBOLD
1187         if [ ! -z "$IMAGE_REPO_ADR" ] && [ $IMAGE_REPO_POLICY == "local" ]; then
1188                 echo -e $YELLOW" Excluding all remote image check/pull when running with image repo: $IMAGE_REPO_ADR and image policy $IMAGE_REPO_POLICY"$EYELLOW
1189         else
1190                 for imagename in $APP_SHORT_NAMES; do
1191                         __check_included_image $imagename
1192                         incl=$?
1193                         __check_project_image $imagename
1194                         proj=$?
1195                         if [ $incl -eq 0 ]; then
1196                                 if [ $proj -eq 0 ]; then
1197                                         START_ARG_MOD=$START_ARG
1198                                         __check_image_local_override $imagename
1199                                         if [ $? -eq 1 ]; then
1200                                                 START_ARG_MOD="local"
1201                                         fi
1202                                 else
1203                                         START_ARG_MOD=$START_ARG
1204                                 fi
1205                                 __check_image_local_build $imagename
1206                                 #No pull of images built locally
1207                                 if [ $? -ne 0 ]; then
1208                                         # A function name is created from the app short name
1209                                         # for example app short name 'HTTPPROXY' -> produce the function
1210                                         # name __HTTPPROXY_imagesetup
1211                                         # This function is called and is expected to exist in the imported
1212                                         # file for the httpproxy test functions
1213                                         # The resulting function impl will call '__check_and_pull_image' function
1214                                         # with appropriate parameters
1215                                         function_pointer="__"$imagename"_imagepull"
1216                                         $function_pointer $START_ARG_MOD $START_ARG
1217                                 fi
1218                         else
1219                                 echo -e $YELLOW" Excluding $imagename image from image check/pull"$EYELLOW
1220                         fi
1221                 done
1222         fi
1223
1224         #Errors in image setting - exit
1225         if [ $IMAGE_ERR -ne 0 ]; then
1226                 echo ""
1227                 echo "#################################################################################################"
1228                 echo -e $RED"One or more images could not be pulled or containers using the images could not be stopped/removed"$ERED
1229                 echo -e $RED"Or local image, overriding remote image, does not exist"$ERED
1230                 if [ $IMAGE_CATEGORY == "DEV" ]; then
1231                         echo -e $RED"Note that SNAPSHOT images may be purged from nexus after a certain period."$ERED
1232                         echo -e $RED"In that case, switch to use a released image instead."$ERED
1233                 fi
1234                 echo "#################################################################################################"
1235                 echo ""
1236                 exit 1
1237         fi
1238
1239         echo ""
1240
1241         echo -e $BOLD"Building images needed for test"$EBOLD
1242
1243         for imagename in $APP_SHORT_NAMES; do
1244                 cd $AUTOTEST_HOME #Always reset to orig dir
1245                 __check_image_local_build $imagename
1246                 if [ $? -eq 0 ]; then
1247                         __check_included_image $imagename
1248                         if [ $? -eq 0 ]; then
1249                                 # A function name is created from the app short name
1250                                 # for example app short name 'MR' -> produce the function
1251                                 # name __MR_imagebuild
1252                                 # This function is called and is expected to exist in the imported
1253                                 # file for the mr test functions
1254                                 # The resulting function impl shall build the imagee
1255                                 function_pointer="__"$imagename"_imagebuild"
1256                                 $function_pointer
1257
1258                         else
1259                                 echo -e $YELLOW" Excluding image for app $imagename from image build"$EYELLOW
1260                         fi
1261                 fi
1262         done
1263
1264         cd $AUTOTEST_HOME # Just to make sure...
1265
1266         echo ""
1267
1268         # Create a table of the images used in the script - from local repo
1269         echo -e $BOLD"Local docker registry images used in this test script"$EBOLD
1270
1271         docker_tmp_file=./tmp/.docker-images-table
1272         format_string="{{.Repository}}\\t{{.Tag}}\\t{{.CreatedSince}}\\t{{.Size}}\\t{{.CreatedAt}}"
1273         echo -e "Application\tRepository\tTag\tCreated since\tSize\tCreated at" > $docker_tmp_file
1274
1275         for imagename in $APP_SHORT_NAMES; do
1276                 __check_included_image $imagename
1277                 if [ $? -eq 0 ]; then
1278                         # Only print image data if image repo is null, or if image repo is set and image is local
1279                         print_image_data=0
1280                         if [ -z "$IMAGE_REPO_ADR" ]; then
1281                                 print_image_data=1
1282                         else
1283                                 __check_image_local_build $imagename
1284                                 if [ $? -eq 0 ]; then
1285                                         print_image_data=1
1286                                 fi
1287                         fi
1288                         if [ $print_image_data -eq 1 ]; then
1289                                 # A function name is created from the app short name
1290                                 # for example app short name 'MR' -> produce the function
1291                                 # name __MR_imagebuild
1292                                 # This function is called and is expected to exist in the imported
1293                                 # file for the mr test functions
1294                                 # The resulting function impl shall build the imagee
1295                                 function_pointer="__"$imagename"_image_data"
1296                                 $function_pointer "$format_string" $docker_tmp_file
1297                         fi
1298                 fi
1299         done
1300
1301         column -t -s $'\t' $docker_tmp_file | indent1
1302
1303         echo ""
1304
1305         if [ ! -z "$IMAGE_REPO_ADR" ]; then
1306
1307                 # Create a table of the images used in the script - from remote repo
1308                 echo -e $BOLD"Remote repo images used in this test script"$EBOLD
1309                 echo -e $YELLOW"-- Note: These image will be pulled when the container starts. Images not managed by the test engine --"$EYELLOW
1310
1311                 docker_tmp_file=./tmp/.docker-images-table
1312                 format_string="{{.Repository}}\\t{{.Tag}}"
1313                 echo -e "Application\tRepository\tTag" > $docker_tmp_file
1314
1315                 for imagename in $APP_SHORT_NAMES; do
1316                         __check_included_image $imagename
1317                         if [ $? -eq 0 ]; then
1318                                 # Only print image data if image repo is null, or if image repo is set and image is local
1319                                 __check_image_local_build $imagename
1320                                 if [ $? -ne 0 ]; then
1321                                         # A function name is created from the app short name
1322                                         # for example app short name 'MR' -> produce the function
1323                                         # name __MR_imagebuild
1324                                         # This function is called and is expected to exist in the imported
1325                                         # file for the mr test functions
1326                                         # The resulting function impl shall build the imagee
1327                                         function_pointer="__"$imagename"_image_data"
1328                                         $function_pointer "$format_string" $docker_tmp_file
1329                                 fi
1330                         fi
1331                 done
1332
1333                 column -t -s $'\t' $docker_tmp_file | indent1
1334
1335                 echo ""
1336         fi
1337
1338         if [ $RUNMODE == "KUBE" ]; then
1339
1340                 echo "================================================================================="
1341                 echo "================================================================================="
1342
1343                 if [ -z "$IMAGE_REPO_ADR" ]; then
1344                         echo -e $YELLOW" The image pull policy is set to 'Never' - assuming a local image repo is available for all images"$EYELLOW
1345                         echo -e " This setting only works on single node clusters on the local machine"
1346                         echo -e " It does not work with multi-node clusters or remote clusters. "
1347                         export KUBE_IMAGE_PULL_POLICY="Never"
1348                 else
1349                         echo -e $YELLOW" The image pull policy is set to 'Always'"$EYELLOW
1350                         echo -e " This setting work on local clusters, multi-node clusters and remote cluster. "
1351                         echo -e " Only locally built images are managed. Remote images are always pulled from remote repos"
1352                         echo -e " Pulling remote snapshot or staging images my in some case result in pulling newer image versions outside the control of the test engine"
1353                         export KUBE_IMAGE_PULL_POLICY="Always"
1354                 fi
1355                 CLUSTER_IP=$(kubectl config view -o jsonpath={.clusters[0].cluster.server} | awk -F[/:] '{print $4}')
1356                 echo -e $YELLOW" The cluster hostname/ip is: $CLUSTER_IP"$EYELLOW
1357
1358                 echo "================================================================================="
1359                 echo "================================================================================="
1360                 echo ""
1361         fi
1362
1363         echo -e $BOLD"======================================================="$EBOLD
1364         echo -e $BOLD"== Common test setup completed -  test script begins =="$EBOLD
1365         echo -e $BOLD"======================================================="$EBOLD
1366         echo ""
1367
1368 }
1369
1370 # Function to print the test result, shall be the last cmd in a test script
1371 # args: -
1372 # (Function for test scripts)
1373 print_result() {
1374
1375         TCTEST_END=$SECONDS
1376         duration=$((TCTEST_END-TCTEST_START))
1377
1378         echo "-------------------------------------------------------------------------------------------------"
1379         echo "-------------------------------------     Test case: "$ATC
1380         echo "-------------------------------------     Ended:     "$(date)
1381         echo "-------------------------------------------------------------------------------------------------"
1382         echo "-- Description: "$TC_ONELINE_DESCR
1383         echo "-- Execution time: " $duration " seconds"
1384         echo "-- Used env file: "$TEST_ENV_VAR_FILE
1385         echo "-------------------------------------------------------------------------------------------------"
1386         echo "-------------------------------------     RESULTS"
1387         echo ""
1388
1389
1390         if [ $RES_DEVIATION -gt 0 ]; then
1391                 echo "Test case deviations"
1392                 echo "===================================="
1393                 cat $DEVIATION_FILE
1394         fi
1395         echo ""
1396         echo "Timer measurement in the test script"
1397         echo "===================================="
1398         column -t -s $'\t' $TIMER_MEASUREMENTS
1399         echo ""
1400
1401         total=$((RES_PASS+RES_FAIL))
1402         if [ $RES_TEST -eq 0 ]; then
1403                 echo -e "\033[1mNo tests seem to have been executed. Check the script....\033[0m"
1404                 echo -e "\033[31m\033[1m ___  ___ ___ ___ ___ _____   ___ _   ___ _   _   _ ___ ___ \033[0m"
1405                 echo -e "\033[31m\033[1m/ __|/ __| _ \_ _| _ \_   _| | __/_\ |_ _| | | | | | _ \ __|\033[0m"
1406                 echo -e "\033[31m\033[1m\__ \ (__|   /| ||  _/ | |   | _/ _ \ | || |_| |_| |   / _| \033[0m"
1407                 echo -e "\033[31m\033[1m|___/\___|_|_\___|_|   |_|   |_/_/ \_\___|____\___/|_|_\___|\033[0m"
1408         elif [ $total != $RES_TEST ]; then
1409                 echo -e "\033[1mTotal number of tests does not match the sum of passed and failed tests. Check the script....\033[0m"
1410                 echo -e "\033[31m\033[1m ___  ___ ___ ___ ___ _____   ___ _   ___ _   _   _ ___ ___ \033[0m"
1411                 echo -e "\033[31m\033[1m/ __|/ __| _ \_ _| _ \_   _| | __/_\ |_ _| | | | | | _ \ __|\033[0m"
1412                 echo -e "\033[31m\033[1m\__ \ (__|   /| ||  _/ | |   | _/ _ \ | || |_| |_| |   / _| \033[0m"
1413                 echo -e "\033[31m\033[1m|___/\___|_|_\___|_|   |_|   |_/_/ \_\___|____\___/|_|_\___|\033[0m"
1414         elif [ $RES_CONF_FAIL -ne 0 ]; then
1415                 echo -e "\033[1mOne or more configurations has failed. Check the script log....\033[0m"
1416                 echo -e "\033[31m\033[1m ___  ___ ___ ___ ___ _____   ___ _   ___ _   _   _ ___ ___ \033[0m"
1417                 echo -e "\033[31m\033[1m/ __|/ __| _ \_ _| _ \_   _| | __/_\ |_ _| | | | | | _ \ __|\033[0m"
1418                 echo -e "\033[31m\033[1m\__ \ (__|   /| ||  _/ | |   | _/ _ \ | || |_| |_| |   / _| \033[0m"
1419                 echo -e "\033[31m\033[1m|___/\___|_|_\___|_|   |_|   |_/_/ \_\___|____\___/|_|_\___|\033[0m"
1420         elif [ $RES_PASS = $RES_TEST ]; then
1421                 echo -e "All tests \033[32m\033[1mPASS\033[0m"
1422                 echo -e "\033[32m\033[1m  ___  _   ___ ___ \033[0m"
1423                 echo -e "\033[32m\033[1m | _ \/_\ / __/ __| \033[0m"
1424                 echo -e "\033[32m\033[1m |  _/ _ \\__ \__ \\ \033[0m"
1425                 echo -e "\033[32m\033[1m |_|/_/ \_\___/___/ \033[0m"
1426                 echo ""
1427
1428                 # Update test suite counter
1429                 if [ -f .tmp_tcsuite_pass_ctr ]; then
1430                         tmpval=$(< .tmp_tcsuite_pass_ctr)
1431                         ((tmpval++))
1432                         echo $tmpval > .tmp_tcsuite_pass_ctr
1433                 fi
1434                 if [ -f .tmp_tcsuite_pass ]; then
1435                         echo " - "$ATC " -- "$TC_ONELINE_DESCR"  Execution time: "$duration" seconds" >> .tmp_tcsuite_pass
1436                 fi
1437                 #Create file with OK exit code
1438                 echo "0" > "$AUTOTEST_HOME/.result$ATC.txt"
1439         else
1440                 echo -e "One or more tests with status  \033[31m\033[1mFAIL\033[0m "
1441                 echo -e "\033[31m\033[1m  ___ _   ___ _    \033[0m"
1442                 echo -e "\033[31m\033[1m | __/_\ |_ _| |   \033[0m"
1443                 echo -e "\033[31m\033[1m | _/ _ \ | || |__ \033[0m"
1444                 echo -e "\033[31m\033[1m |_/_/ \_\___|____|\033[0m"
1445                 echo ""
1446                 # Update test suite counter
1447                 if [ -f .tmp_tcsuite_fail_ctr ]; then
1448                         tmpval=$(< .tmp_tcsuite_fail_ctr)
1449                         ((tmpval++))
1450                         echo $tmpval > .tmp_tcsuite_fail_ctr
1451                 fi
1452                 if [ -f .tmp_tcsuite_fail ]; then
1453                         echo " - "$ATC " -- "$TC_ONELINE_DESCR"  Execution time: "$duration" seconds" >> .tmp_tcsuite_fail
1454                 fi
1455         fi
1456
1457         echo "++++ Number of tests:          "$RES_TEST
1458         echo "++++ Number of passed tests:   "$RES_PASS
1459         echo "++++ Number of failed tests:   "$RES_FAIL
1460         echo ""
1461         echo "++++ Number of failed configs: "$RES_CONF_FAIL
1462         echo ""
1463         echo "++++ Number of test case deviations: "$RES_DEVIATION
1464         echo ""
1465         echo "-------------------------------------     Test case complete    ---------------------------------"
1466         echo "-------------------------------------------------------------------------------------------------"
1467         echo ""
1468 }
1469
1470 #####################################################################
1471 ###### Functions for start, configuring, stoping, cleaning etc ######
1472 #####################################################################
1473
1474 # Start timer for time measurement
1475 # args - (any args will be printed though)
1476 start_timer() {
1477         echo -e $BOLD"INFO(${BASH_LINENO[0]}): "${FUNCNAME[0]}"," $@ $EBOLD
1478         TC_TIMER=$SECONDS
1479         echo " Timer started: $(date)"
1480 }
1481
1482 # Print the value of the time (in seconds)
1483 # args - <timer message to print>  -  timer value and message will be printed both on screen
1484 #                                     and in the timer measurement report
1485 print_timer() {
1486         echo -e $BOLD"INFO(${BASH_LINENO[0]}): "${FUNCNAME[0]}"," $@ $EBOLD
1487         if [ $# -lt 1 ]; then
1488                 ((RES_CONF_FAIL++))
1489         __print_err "need 1 or more args,  <timer message to print>" $@
1490                 exit 1
1491         fi
1492         duration=$(($SECONDS-$TC_TIMER))
1493         if [ $duration -eq 0 ]; then
1494                 duration="<1 second"
1495         else
1496                 duration=$duration" seconds"
1497         fi
1498         echo " Timer duration :" $duration
1499
1500         echo -e "${@:1} \t $duration" >> $TIMER_MEASUREMENTS
1501 }
1502
1503 # Print the value of the time (in seconds) and reset the timer
1504 # args - <timer message to print>  -  timer value and message will be printed both on screen
1505 #                                     and in the timer measurement report
1506 print_and_reset_timer() {
1507         echo -e $BOLD"INFO(${BASH_LINENO[0]}): "${FUNCNAME[0]}"," $@ $EBOLD
1508         if [ $# -lt 1 ]; then
1509                 ((RES_CONF_FAIL++))
1510         __print_err "need 1 or more args,  <timer message to print>" $@
1511                 exit 1
1512         fi
1513         duration=$(($SECONDS-$TC_TIMER))" seconds"
1514         if [ $duration -eq 0 ]; then
1515                 duration="<1 second"
1516         else
1517                 duration=$duration" seconds"
1518         fi
1519         echo " Timer duration :" $duration
1520         TC_TIMER=$SECONDS
1521         echo " Timer reset"
1522
1523         echo -e "${@:1} \t $duration" >> $TIMER_MEASUREMENTS
1524
1525 }
1526 # Print info about a deviations from intended tests
1527 # Each deviation counted is also printed in the testreport
1528 # args <deviation message to print>
1529 deviation() {
1530         echo -e $BOLD"DEVIATION(${BASH_LINENO[0]}): "${FUNCNAME[0]} $EBOLD
1531         if [ $# -lt 1 ]; then
1532                 ((RES_CONF_FAIL++))
1533                 __print_err "need 1 or more args,  <deviation message to print>" $@
1534                 exit 1
1535         fi
1536         ((RES_DEVIATION++))
1537         echo -e $BOLD$YELLOW" Test case deviation: ${@:1}"$EYELLOW$EBOLD
1538         echo "Line: ${BASH_LINENO[0]} - ${@:1}" >> $DEVIATION_FILE
1539         __print_current_stats
1540         echo ""
1541 }
1542
1543 # Stop at first FAIL test case and take all logs - only for debugging/trouble shooting
1544 __check_stop_at_error() {
1545         if [ $STOP_AT_ERROR -eq 1 ]; then
1546                 echo -e $RED"Test script configured to stop at first FAIL, taking all logs and stops"$ERED
1547                 store_logs "STOP_AT_ERROR"
1548                 exit 1
1549         fi
1550         return 0
1551 }
1552
1553 # Stop and remove all containers
1554 # args: -
1555 # (Not for test scripts)
1556 __clean_containers() {
1557
1558         echo -e $BOLD"Docker clean and stopping and removing all running containers, by container name"$EBOLD
1559
1560         #Create empty file
1561         running_contr_file="./tmp/running_contr.txt"
1562         > $running_contr_file
1563
1564         # Get list of all containers started by the test script
1565         for imagename in $APP_SHORT_NAMES; do
1566                 docker ps -a --filter "label=nrttest_app=$imagename"  --filter "network=$DOCKER_SIM_NWNAME" --format ' {{.Label "nrttest_dp"}}\n{{.Label "nrttest_app"}}\n{{.Names}}' >> $running_contr_file
1567         done
1568
1569         tab_heading1="App display name"
1570         tab_heading2="App short name"
1571         tab_heading3="Container name"
1572
1573         tab_heading1_len=${#tab_heading1}
1574         tab_heading2_len=${#tab_heading2}
1575         tab_heading3_len=${#tab_heading3}
1576         cntr=0
1577         #Calc field lengths of each item in the list of containers
1578         while read p; do
1579                 if (( $cntr % 3 == 0 ));then
1580                         if [ ${#p} -gt $tab_heading1_len ]; then
1581                                 tab_heading1_len=${#p}
1582                         fi
1583                 fi
1584                 if (( $cntr % 3 == 1));then
1585                         if [ ${#p} -gt $tab_heading2_len ]; then
1586                                 tab_heading2_len=${#p}
1587                         fi
1588                 fi
1589                 if (( $cntr % 3 == 2));then
1590                         if [ ${#p} -gt $tab_heading3_len ]; then
1591                                 tab_heading3_len=${#p}
1592                         fi
1593                 fi
1594                 let cntr=cntr+1
1595         done <$running_contr_file
1596
1597         let tab_heading1_len=tab_heading1_len+2
1598         while (( ${#tab_heading1} < $tab_heading1_len)); do
1599                 tab_heading1="$tab_heading1"" "
1600         done
1601
1602         let tab_heading2_len=tab_heading2_len+2
1603         while (( ${#tab_heading2} < $tab_heading2_len)); do
1604                 tab_heading2="$tab_heading2"" "
1605         done
1606
1607         let tab_heading3_len=tab_heading3_len+2
1608         while (( ${#tab_heading3} < $tab_heading3_len)); do
1609                 tab_heading3="$tab_heading3"" "
1610         done
1611
1612         echo " $tab_heading1$tab_heading2$tab_heading3"" Actions"
1613         cntr=0
1614         while read p; do
1615                 if (( $cntr % 3 == 0 ));then
1616                         row=""
1617                         heading=$p
1618                         heading_len=$tab_heading1_len
1619                 fi
1620                 if (( $cntr % 3 == 1));then
1621                         heading=$p
1622                         heading_len=$tab_heading2_len
1623                 fi
1624                 if (( $cntr % 3 == 2));then
1625                         contr=$p
1626                         heading=$p
1627                         heading_len=$tab_heading3_len
1628                 fi
1629                 while (( ${#heading} < $heading_len)); do
1630                         heading="$heading"" "
1631                 done
1632                 row=$row$heading
1633                 if (( $cntr % 3 == 2));then
1634                         echo -ne $row$SAMELINE
1635                         echo -ne " $row ${GREEN}stopping...${EGREEN}${SAMELINE}"
1636                         docker stop $(docker ps -qa --filter name=${contr} --filter network=$DOCKER_SIM_NWNAME) &> /dev/null
1637                         echo -ne " $row ${GREEN}stopped removing...${EGREEN}${SAMELINE}"
1638                         docker rm --force $(docker ps -qa --filter name=${contr} --filter network=$DOCKER_SIM_NWNAME) &> /dev/null
1639                         echo -e  " $row ${GREEN}stopped removed     ${EGREEN}"
1640                 fi
1641                 let cntr=cntr+1
1642         done <$running_contr_file
1643
1644         echo ""
1645
1646         echo -e $BOLD" Removing docker network"$EBOLD
1647         TMP=$(docker network ls -q --filter name=$DOCKER_SIM_NWNAME)
1648         if [ "$TMP" ==  $DOCKER_SIM_NWNAME ]; then
1649                 docker network rm $DOCKER_SIM_NWNAME | indent2
1650                 if [ $? -ne 0 ];  then
1651                         echo -e $RED" Cannot remove docker network. Manually remove or disconnect containers from $DOCKER_SIM_NWNAME"$ERED
1652                         exit 1
1653                 fi
1654         fi
1655         echo -e "$GREEN  Done$EGREEN"
1656
1657         echo -e $BOLD" Removing all unused docker neworks"$EBOLD
1658         docker network prune --force | indent2
1659         echo -e "$GREEN  Done$EGREEN"
1660
1661         echo -e $BOLD" Removing all unused docker volumes"$EBOLD
1662         docker volume prune --force | indent2
1663         echo -e "$GREEN  Done$EGREEN"
1664
1665         echo -e $BOLD" Removing all dangling/untagged docker images"$EBOLD
1666     docker rmi --force $(docker images -q -f dangling=true) &> /dev/null
1667         echo -e "$GREEN  Done$EGREEN"
1668         echo ""
1669
1670         CONTRS=$(docker ps | awk '$1 != "CONTAINER" { n++ }; END { print n+0 }')
1671         if [ $? -eq 0 ]; then
1672                 if [ $CONTRS -ne 0 ]; then
1673                         echo -e $RED"Containers running, may cause distubance to the test case"$ERED
1674                         docker ps -a | indent1
1675                         echo ""
1676                 fi
1677         fi
1678 }
1679
1680 ###################################
1681 ### Functions for kube management
1682 ###################################
1683
1684 # Get resource type for scaling
1685 # args: <resource-name> <namespace>
1686 __kube_get_resource_type() {
1687         kubectl get deployment $1 -n $2 1> /dev/null 2> ./tmp/kubeerr
1688         if [ $? -eq 0 ]; then
1689                 echo "deployment"
1690                 return 0
1691         fi
1692         kubectl get sts $1 -n $2 1> /dev/null 2> ./tmp/kubeerr
1693         if [ $? -eq 0 ]; then
1694                 echo "sts"
1695                 return 0
1696         fi
1697         echo "unknown-resource-type"
1698         return 1
1699 }
1700
1701 # Scale a kube resource to a specific count
1702 # args: <resource-type> <resource-name> <namespace> <target-count>
1703 # (Not for test scripts)
1704 __kube_scale() {
1705         echo -ne "  Setting $1 $2 replicas=$4 in namespace $3"$SAMELINE
1706         kubectl scale  $1 $2  -n $3 --replicas=$4 1> /dev/null 2> ./tmp/kubeerr
1707         if [ $? -ne 0 ]; then
1708                 echo -e "  Setting $1 $2 replicas=$4 in namespace $3 $RED Failed $ERED"
1709                 ((RES_CONF_FAIL++))
1710                 echo "  Message: $(<./tmp/kubeerr)"
1711                 return 1
1712         else
1713                 echo -e "  Setting $1 $2 replicas=$4 in namespace $3 $GREEN OK $EGREEN"
1714         fi
1715
1716         TSTART=$SECONDS
1717
1718         for i in {1..500}; do
1719                 count=$(kubectl get $1/$2  -n $3 -o jsonpath='{.status.replicas}' 2> /dev/null)
1720                 retcode=$?
1721                 if [ -z "$count" ]; then
1722                         #No value is sometimes returned for some reason, in case the resource has replica 0
1723                         count=0
1724                 fi
1725                 if [ $retcode -ne 0 ]; then
1726                         echo -e "$RED  Cannot fetch current replica count for $1 $2 in namespace $3 $ERED"
1727                         ((RES_CONF_FAIL++))
1728                         return 1
1729                 fi
1730                 #echo ""
1731                 if [ $count -ne $4 ]; then
1732                         echo -ne "  Waiting for $1 $2 replicas=$4 in namespace $3. Replicas=$count after $(($SECONDS-$TSTART)) seconds $SAMELINE"
1733                         sleep $i
1734                 else
1735                         echo -e "  Waiting for $1 $2 replicas=$4 in namespace $3. Replicas=$count after $(($SECONDS-$TSTART)) seconds"
1736                         echo -e "  Replicas=$4 after $(($SECONDS-$TSTART)) seconds $GREEN OK $EGREEN"
1737                         echo ""
1738                         return 0
1739                 fi
1740         done
1741         echo ""
1742         echo -e "$RED  Replica count did not reach target replicas=$4. Failed with replicas=$count $ERED"
1743         ((RES_CONF_FAIL++))
1744         return 0
1745 }
1746
1747 # Scale all kube resource sets to 0 in a namespace for resources having a certain lable and label-id
1748 # This function does not wait for the resource to reach 0
1749 # args: <namespace> <label-name> <label-id>
1750 # (Not for test scripts)
1751 __kube_scale_all_resources() {
1752         namespace=$1
1753         labelname=$2
1754         labelid=$3
1755         resources="deployment replicaset statefulset"
1756         for restype in $resources; do
1757                 result=$(kubectl get $restype -n $namespace -o jsonpath='{.items[?(@.metadata.labels.'$labelname'=="'$labelid'")].metadata.name}')
1758                 if [ $? -eq 0 ] && [ ! -z "$result" ]; then
1759                         deleted_resourcetypes=$deleted_resourcetypes" "$restype
1760                         for resid in $result; do
1761                                 echo -ne "  Ordered caling $restype $resid from namespace $namespace with label $labelname=$labelid to 0"$SAMELINE
1762                                 kubectl scale  $restype $resid  -n $namespace --replicas=0 1> /dev/null 2> ./tmp/kubeerr
1763                                 echo -e "  Ordered scaling $restype $resid from namespace $namespace with label $labelname=$labelid to 0 $GREEN OK $EGREEN"
1764                         done
1765                 fi
1766         done
1767 }
1768
1769 # Scale all kube resource sets to 0 in a namespace for resources having a certain lable and label-id
1770 # This function do wait for the resource to reach 0
1771 # args: <namespace> <label-name> <label-id>
1772 # (Not for test scripts)
1773 __kube_scale_and_wait_all_resources() {
1774         namespace=$1
1775         labelname=$2
1776         labelid=$3
1777         resources="deployment replicaset statefulset"
1778         scaled_all=1
1779         while [ $scaled_all -ne 0 ]; do
1780                 scaled_all=0
1781                 for restype in $resources; do
1782                         result=$(kubectl get $restype -n $namespace -o jsonpath='{.items[?(@.metadata.labels.'$labelname'=="'$labelid'")].metadata.name}')
1783                         if [ $? -eq 0 ] && [ ! -z "$result" ]; then
1784                                 for resid in $result; do
1785                                         echo -e "  Ordered scaling $restype $resid from namespace $namespace with label $labelname=$labelid to 0"
1786                                         kubectl scale  $restype $resid  -n $namespace --replicas=0 1> /dev/null 2> ./tmp/kubeerr
1787                                         count=1
1788                                         T_START=$SECONDS
1789                                         while [ $count -ne 0 ]; do
1790                                                 count=$(kubectl get $restype $resid  -n $namespace -o jsonpath='{.status.replicas}' 2> /dev/null)
1791                                                 echo -ne "  Scaling $restype $resid from namespace $namespace with label $labelname=$labelid to 0,count=$count"$SAMELINE
1792                                                 if [ $? -eq 0 ] && [ ! -z "$count" ]; then
1793                                                         sleep 0.5
1794                                                 else
1795                                                         count=0
1796                                                 fi
1797                                                 duration=$(($SECONDS-$T_START))
1798                                                 if [ $duration -gt 100 ]; then
1799                                                         #Forcring count 0, to avoid hanging for failed scaling
1800                                                         scaled_all=1
1801                                                         count=0
1802                                                 fi
1803                                         done
1804                                         echo -e "  Scaled $restype $resid from namespace $namespace with label $labelname=$labelid to 0,count=$count $GREEN OK $EGREEN"
1805                                 done
1806                         fi
1807                 done
1808         done
1809 }
1810
1811 # Remove all kube resources in a namespace for resources having a certain label and label-id
1812 # This function wait until the resources are gone. Scaling to 0 must have been ordered previously
1813 # args: <namespace> <label-name> <label-id>
1814 # (Not for test scripts)
1815 __kube_delete_all_resources() {
1816         namespace=$1
1817         labelname=$2
1818         labelid=$3
1819         resources="deployments replicaset statefulset services pods configmaps persistentvolumeclaims persistentvolumes"
1820         deleted_resourcetypes=""
1821         for restype in $resources; do
1822                 result=$(kubectl get $restype -n $namespace -o jsonpath='{.items[?(@.metadata.labels.'$labelname'=="'$labelid'")].metadata.name}')
1823                 if [ $? -eq 0 ] && [ ! -z "$result" ]; then
1824                         deleted_resourcetypes=$deleted_resourcetypes" "$restype
1825                         for resid in $result; do
1826                                 if [ $restype == "replicaset" ] || [ $restype == "statefulset" ]; then
1827                                         count=1
1828                                         while [ $count -ne 0 ]; do
1829                                                 count=$(kubectl get $restype $resid  -n $namespace -o jsonpath='{.status.replicas}' 2> /dev/null)
1830                                                 echo -ne "  Scaling $restype $resid from namespace $namespace with label $labelname=$labelid to 0,count=$count"$SAMELINE
1831                                                 if [ $? -eq 0 ] && [ ! -z "$count" ]; then
1832                                                         sleep 0.5
1833                                                 else
1834                                                         count=0
1835                                                 fi
1836                                         done
1837                                         echo -e "  Scaled $restype $resid from namespace $namespace with label $labelname=$labelid to 0,count=$count $GREEN OK $EGREEN"
1838                                 fi
1839                                 echo -ne "  Deleting $restype $resid from namespace $namespace with label $labelname=$labelid "$SAMELINE
1840                                 kubectl delete $restype $resid -n $namespace 1> /dev/null 2> ./tmp/kubeerr
1841                                 if [ $? -eq 0 ]; then
1842                                         echo -e "  Deleted $restype $resid from namespace $namespace with label $labelname=$labelid $GREEN OK $EGREEN"
1843                                 else
1844                                         echo -e "  Deleted $restype $resid from namespace $namespace with label $labelname=$labelid $GREEN Does not exist - OK $EGREEN"
1845                                 fi
1846                                 #fi
1847                         done
1848                 fi
1849         done
1850         if [ ! -z "$deleted_resourcetypes" ]; then
1851                 for restype in $deleted_resources; do
1852                         echo -ne "  Waiting for $restype in namespace $namespace with label $labelname=$labelid to be deleted..."$SAMELINE
1853                         T_START=$SECONDS
1854                         result="dummy"
1855                         while [ ! -z "$result" ]; do
1856                                 sleep 0.5
1857                                 result=$(kubectl get $restype -n $namespace -o jsonpath='{.items[?(@.metadata.labels.'$labelname'=="'$labelid'")].metadata.name}')
1858                                 echo -ne "  Waiting for $restype in namespace $namespace with label $labelname=$labelid to be deleted...$(($SECONDS-$T_START)) seconds "$SAMELINE
1859                                 if [ -z "$result" ]; then
1860                                         echo -e " Waiting for $restype in namespace $namespace with label $labelname=$labelid to be deleted...$(($SECONDS-$T_START)) seconds $GREEN OK $EGREEN"
1861                                 elif [ $(($SECONDS-$T_START)) -gt 300 ]; then
1862                                         echo -e " Waiting for $restype in namespace $namespace with label $labelname=$labelid to be deleted...$(($SECONDS-$T_START)) seconds $RED Failed $ERED"
1863                                         result=""
1864                                 fi
1865                         done
1866                 done
1867         fi
1868 }
1869
1870 # Creates a namespace if it does not exists
1871 # args: <namespace>
1872 # (Not for test scripts)
1873 __kube_create_namespace() {
1874
1875         #Check if test namespace exists, if not create it
1876         kubectl get namespace $1 1> /dev/null 2> ./tmp/kubeerr
1877         if [ $? -ne 0 ]; then
1878                 echo -ne " Creating namespace "$1 $SAMELINE
1879                 kubectl create namespace $1 1> /dev/null 2> ./tmp/kubeerr
1880                 if [ $? -ne 0 ]; then
1881                         echo -e " Creating namespace $1 $RED$BOLD FAILED $EBOLD$ERED"
1882                         ((RES_CONF_FAIL++))
1883                         echo "  Message: $(<./tmp/kubeerr)"
1884                         return 1
1885                 else
1886                         echo -e " Creating namespace $1 $GREEN$BOLD OK $EBOLD$EGREEN"
1887                 fi
1888         else
1889                 echo -e " Creating namespace $1 $GREEN$BOLD Already exists, OK $EBOLD$EGREEN"
1890         fi
1891         return 0
1892 }
1893
1894 # Find the host ip of an app (using the service resource)
1895 # args: <app-name> <namespace>
1896 # (Not for test scripts)
1897 __kube_get_service_host() {
1898         if [ $# -ne 2 ]; then
1899                 ((RES_CONF_FAIL++))
1900         __print_err "need 2 args, <app-name> <namespace>" $@
1901                 exit 1
1902         fi
1903         for timeout in {1..60}; do
1904                 host=$(kubectl get svc $1  -n $2 -o jsonpath='{.spec.clusterIP}')
1905                 if [ $? -eq 0 ]; then
1906                         if [ ! -z "$host" ]; then
1907                                 echo $host
1908                                 return 0
1909                         fi
1910                 fi
1911                 sleep 0.5
1912         done
1913         ((RES_CONF_FAIL++))
1914         echo "host-not-found-fatal-error"
1915         return 1
1916 }
1917
1918 # Find the named port to an app (using the service resource)
1919 # args: <app-name> <namespace> <port-name>
1920 # (Not for test scripts)
1921 __kube_get_service_port() {
1922         if [ $# -ne 3 ]; then
1923                 ((RES_CONF_FAIL++))
1924         __print_err "need 3 args, <app-name> <namespace> <port-name>" $@
1925                 exit 1
1926         fi
1927
1928         for timeout in {1..60}; do
1929                 port=$(kubectl get svc $1  -n $2 -o jsonpath='{...ports[?(@.name=="'$3'")].port}')
1930                 if [ $? -eq 0 ]; then
1931                         if [ ! -z "$port" ]; then
1932                                 echo $port
1933                                 return 0
1934                         fi
1935                 fi
1936                 sleep 0.5
1937         done
1938         ((RES_CONF_FAIL++))
1939         echo "0"
1940         return 1
1941 }
1942
1943 # Find the named node port to an app (using the service resource)
1944 # args: <app-name> <namespace> <port-name>
1945 # (Not for test scripts)
1946 __kube_get_service_nodeport() {
1947         if [ $# -ne 3 ]; then
1948                 ((RES_CONF_FAIL++))
1949         __print_err "need 3 args, <app-name> <namespace> <port-name>" $@
1950                 exit 1
1951         fi
1952
1953         for timeout in {1..60}; do
1954                 port=$(kubectl get svc $1  -n $2 -o jsonpath='{...ports[?(@.name=="'$3'")].nodePort}')
1955                 if [ $? -eq 0 ]; then
1956                         if [ ! -z "$port" ]; then
1957                                 echo $port
1958                                 return 0
1959                         fi
1960                 fi
1961                 sleep 0.5
1962         done
1963         ((RES_CONF_FAIL++))
1964         echo "0"
1965         return 1
1966 }
1967
1968 # Create a kube resource from a yaml template
1969 # args: <resource-type> <resource-name> <template-yaml> <output-yaml>
1970 # (Not for test scripts)
1971 __kube_create_instance() {
1972         echo -ne " Creating $1 $2"$SAMELINE
1973         envsubst < $3 > $4
1974         kubectl apply -f $4 1> /dev/null 2> ./tmp/kubeerr
1975         if [ $? -ne 0 ]; then
1976                 ((RES_CONF_FAIL++))
1977                 echo -e " Creating $1 $2 $RED Failed $ERED"
1978                 echo "  Message: $(<./tmp/kubeerr)"
1979                 return 1
1980         else
1981                 echo -e " Creating $1 $2 $GREEN OK $EGREEN"
1982         fi
1983 }
1984
1985 # Function to create a configmap in kubernetes
1986 # args: <configmap-name> <namespace> <labelname> <labelid> <path-to-data-file> <path-to-output-yaml>
1987 # (Not for test scripts)
1988 __kube_create_configmap() {
1989         echo -ne " Creating configmap $1 "$SAMELINE
1990         envsubst < $5 > $5"_tmp"
1991         cp $5"_tmp" $5  #Need to copy back to orig file name since create configmap neeed the original file name
1992         kubectl create configmap $1  -n $2 --from-file=$5 --dry-run=client -o yaml > $6
1993         if [ $? -ne 0 ]; then
1994                 echo -e " Creating configmap $1 $RED Failed $ERED"
1995                 ((RES_CONF_FAIL++))
1996                 return 1
1997         fi
1998
1999         kubectl apply -f $6 1> /dev/null 2> ./tmp/kubeerr
2000         if [ $? -ne 0 ]; then
2001                 echo -e " Creating configmap $1 $RED Apply failed $ERED"
2002                 echo "  Message: $(<./tmp/kubeerr)"
2003                 ((RES_CONF_FAIL++))
2004                 return 1
2005         fi
2006         kubectl label configmap $1 -n $2 $3"="$4 --overwrite 1> /dev/null 2> ./tmp/kubeerr
2007         if [ $? -ne 0 ]; then
2008                 echo -e " Creating configmap $1 $RED Labeling failed $ERED"
2009                 echo "  Message: $(<./tmp/kubeerr)"
2010                 ((RES_CONF_FAIL++))
2011                 return 1
2012         fi
2013         # Log the resulting map
2014         kubectl get configmap $1 -n $2 -o yaml > $6
2015
2016         echo -e " Creating configmap $1 $GREEN OK $EGREEN"
2017         return 0
2018 }
2019
2020 # This function runs a kubectl cmd where a single output value is expected, for example get ip with jsonpath filter.
2021 # The function retries up to the timeout given in the cmd flag '--cluster-timeout'
2022 # args: <full kubectl cmd with parameters>
2023 # (Not for test scripts)
2024 __kube_cmd_with_timeout() {
2025         TS_TMP=$(($SECONDS+$CLUSTER_TIME_OUT))
2026
2027         while true; do
2028                 kube_cmd_result=$($@)
2029                 if [ $? -ne 0 ]; then
2030                         kube_cmd_result=""
2031                 fi
2032                 if [ $SECONDS -ge $TS_TMP ] || [ ! -z "$kube_cmd_result" ] ; then
2033                         echo $kube_cmd_result
2034                         return 0
2035                 fi
2036                 sleep 1
2037         done
2038 }
2039
2040 # This function starts a pod that cleans a the contents of a path mounted as a pvc
2041 # After this action the pod should terminate
2042 # This should only be executed when the pod owning the pvc is not running
2043 # args: <appname> <namespace> <pvc-name> <path-to remove>
2044 # (Not for test scripts)
2045 __kube_clean_pvc() {
2046
2047         export PVC_CLEANER_NAMESPACE=$2
2048         export PVC_CLEANER_CLAIMNAME=$3
2049         export PVC_CLEANER_RM_PATH=$4
2050         input_yaml=$SIM_GROUP"/pvc-cleaner/"pvc-cleaner.yaml
2051         output_yaml=$PWD/tmp/$2-pvc-cleaner.yaml
2052
2053         envsubst < $input_yaml > $output_yaml
2054
2055         kubectl delete -f $output_yaml 1> /dev/null 2> /dev/null   # Delete the previous terminated pod - if existing
2056
2057         __kube_create_instance pod pvc-cleaner $input_yaml $output_yaml
2058         if [ $? -ne 0 ]; then
2059                 echo $YELLOW" Could not clean pvc for app: $1 - persistent storage not clean - tests may not work"
2060                 return 1
2061         fi
2062
2063         term_ts=$(($SECONDS+30))
2064         while [ $term_ts -gt $SECONDS ]; do
2065                 pod_status=$(kubectl get pod pvc-cleaner -n $PVC_CLEANER_NAMESPACE --no-headers -o custom-columns=":status.phase")
2066                 if [ "$pod_status" == "Succeeded" ]; then
2067                         return 0
2068                 fi
2069         done
2070         return 1
2071 }
2072
2073 # This function scales or deletes all resources for app selected by the testcase.
2074 # args: -
2075 # (Not for test scripts)
2076 __clean_kube() {
2077         echo -e $BOLD"Initialize kube services//pods/statefulsets/replicaset to initial state"$EBOLD
2078
2079         # Scale prestarted or managed apps
2080         for imagename in $APP_SHORT_NAMES; do
2081                 # A function name is created from the app short name
2082                 # for example app short name 'RICMSIM' -> produce the function
2083                 # name __RICSIM_kube_scale_zero or __RICSIM_kube_scale_zero_and_wait
2084                 # This function is called and is expected to exist in the imported
2085                 # file for the ricsim test functions
2086                 # The resulting function impl shall scale the resources to 0
2087                 # For prestarted apps, the function waits until the resources are 0
2088                 # For included (not prestated) apps, the scaling is just ordered
2089                 __check_prestarted_image $imagename
2090                 if [ $? -eq 0 ]; then
2091                         function_pointer="__"$imagename"_kube_scale_zero_and_wait"
2092                         echo -e " Scaling all kube resources for app $BOLD $imagename $EBOLD to 0"
2093                         $function_pointer
2094                 else
2095                         __check_included_image $imagename
2096                         if [ $? -eq 0 ]; then
2097                                 function_pointer="__"$imagename"_kube_scale_zero"
2098                                 echo -e " Scaling all kube resources for app $BOLD $imagename $EBOLD to 0"
2099                                 $function_pointer
2100                         fi
2101                 fi
2102         done
2103
2104         # Delete managed apps
2105         for imagename in $APP_SHORT_NAMES; do
2106                 __check_included_image $imagename
2107                 if [ $? -eq 0 ]; then
2108                         __check_prestarted_image $imagename
2109                         if [ $? -ne 0 ]; then
2110                                 # A function name is created from the app short name
2111                                 # for example app short name 'RICMSIM' -> produce the function
2112                                 # name __RICSIM__kube_delete_all
2113                                 # This function is called and is expected to exist in the imported
2114                                 # file for the ricsim test functions
2115                                 # The resulting function impl shall delete all its resources
2116                                 function_pointer="__"$imagename"_kube_delete_all"
2117                                 echo -e " Deleting all kube resources for app $BOLD $imagename $EBOLD"
2118                                 $function_pointer
2119                         fi
2120                 fi
2121         done
2122
2123         echo ""
2124 }
2125
2126 # Function stop and remove all containers (docker) and services/deployments etc(kube)
2127 # args: -
2128 # Function for test script
2129 clean_environment() {
2130         if [ $RUNMODE == "KUBE" ]; then
2131                 __clean_kube
2132         else
2133                 __clean_containers
2134         fi
2135 }
2136
2137 # Function stop and remove all containers (docker) and services/deployments etc(kube) in the end of the test script, if the arg 'auto-clean' is given at test script start
2138 # args: -
2139 # (Function for test scripts)
2140 auto_clean_environment() {
2141         echo
2142         if [ "$AUTO_CLEAN" == "auto" ]; then
2143                 echo -e $BOLD"Initiating automatic cleaning of environment"$EBOLD
2144                 clean_environment
2145         fi
2146 }
2147
2148 # Function to sleep a test case for a numner of seconds. Prints the optional text args as info
2149 # args: <sleep-time-in-sec> [any-text-in-quotes-to-be-printed]
2150 # (Function for test scripts)
2151 sleep_wait() {
2152
2153         echo -e $BOLD"INFO(${BASH_LINENO[0]}): "${FUNCNAME[0]}"," $@ $EBOLD
2154         if [ $# -lt 1 ]; then
2155                 ((RES_CONF_FAIL++))
2156                 __print_err "need at least one arg, <sleep-time-in-sec> [any-text-to-printed]" $@
2157                 exit 1
2158         fi
2159         #echo "---- Sleep for " $1 " seconds ---- "$2
2160         start=$SECONDS
2161         duration=$((SECONDS-start))
2162         while [ $duration -lt $1 ]; do
2163                 echo -ne "  Slept for ${duration} seconds${SAMELINE}"
2164                 sleep 1
2165                 duration=$((SECONDS-start))
2166         done
2167         echo -ne "  Slept for ${duration} seconds${SAMELINE}"
2168         echo ""
2169 }
2170
2171 # Print error info for the call in the parent script (test case). Arg: <error-message-to-print>
2172 # Not to be called from the test script itself.
2173 __print_err() {
2174     echo -e $RED ${FUNCNAME[1]} " "$1" " ${BASH_SOURCE[2]} " line" ${BASH_LINENO[1]} $ERED
2175         if [ $# -gt 1 ]; then
2176                 echo -e $RED" Got: "${FUNCNAME[1]} ${@:2} $ERED
2177         fi
2178         ((RES_CONF_FAIL++))
2179 }
2180
2181 # Function to create the docker network for the test
2182 # Not to be called from the test script itself.
2183 __create_docker_network() {
2184         tmp=$(docker network ls --format={{.Name}} --filter name=$DOCKER_SIM_NWNAME)
2185         if [ $? -ne 0 ]; then
2186                 echo -e $RED" Could not check if docker network $DOCKER_SIM_NWNAME exists"$ERED
2187                 return 1
2188         fi
2189         if [ "$tmp" != $DOCKER_SIM_NWNAME ]; then
2190                 echo -e " Creating docker network:$BOLD $DOCKER_SIM_NWNAME $EBOLD"
2191                 docker network create $DOCKER_SIM_NWNAME | indent2
2192                 if [ $? -ne 0 ]; then
2193                         echo -e $RED" Could not create docker network $DOCKER_SIM_NWNAME"$ERED
2194                         return 1
2195                 else
2196                         echo -e "$GREEN  Done$EGREEN"
2197                 fi
2198         else
2199                 echo -e " Docker network $DOCKER_SIM_NWNAME already exists$GREEN OK $EGREEN"
2200         fi
2201 }
2202
2203 # Function to start container with docker-compose and wait until all are in state running.
2204 # If the <docker-compose-file> is empty, the default 'docker-compose.yml' is assumed.
2205 #args: <docker-compose-dir> <docker-compose-file> <docker-compose-arg>|NODOCKERARGS <count> <app-name>+
2206 # (Not for test scripts)
2207 __start_container() {
2208
2209         if [ $# -lt 5 ]; then
2210                 ((RES_CONF_FAIL++))
2211         __print_err "need 5 or more args, <docker-compose-dir> <docker-compose-file> <docker-compose-arg>|NODOCKERARGS <count> <app-name>+" $@
2212                 exit 1
2213         fi
2214
2215         __create_docker_network
2216
2217         curdir=$PWD
2218         cd $SIM_GROUP
2219         compose_dir=$1
2220         cd $1
2221         shift
2222         compose_file=$1
2223         if [ -z "$compose_file" ]; then
2224                 compose_file="docker-compose.yml"
2225         fi
2226         shift
2227         compose_args=$1
2228         shift
2229         appcount=$1
2230         shift
2231
2232         if [ "$compose_args" == "NODOCKERARGS" ]; then
2233                 docker-compose -f $compose_file up -d &> .dockererr
2234                 if [ $? -ne 0 ]; then
2235                         echo -e $RED"Problem to launch container(s) with docker-compose"$ERED
2236                         cat .dockererr
2237                         echo -e $RED"Stopping script...."$ERED
2238                         exit 1
2239                 fi
2240         else
2241                 docker-compose -f $compose_file up -d $compose_args &> .dockererr
2242                 if [ $? -ne 0 ]; then
2243                         echo -e $RED"Problem to launch container(s) with docker-compose"$ERED
2244                         cat .dockererr
2245                         echo -e $RED"Stopping script...."$ERED
2246                         exit 1
2247                 fi
2248         fi
2249
2250         cd $curdir
2251
2252         appindex=0
2253         while [ $appindex -lt $appcount ]; do
2254                 appname=$1
2255                 shift
2256                 app_started=0
2257                 for i in {1..10}; do
2258                         if [ "$(docker inspect --format '{{ .State.Running }}' $appname)" == "true" ]; then
2259                                         echo -e " Container $BOLD${appname}$EBOLD$GREEN running$EGREEN on$BOLD image $(docker inspect --format '{{ .Config.Image }}' ${appname}) $EBOLD"
2260                                         app_started=1
2261                                         break
2262                                 else
2263                                         sleep $i
2264                         fi
2265                 done
2266                 if [ $app_started -eq 0 ]; then
2267                         ((RES_CONF_FAIL++))
2268                         echo ""
2269                         echo -e $RED" Container $BOLD${appname}$EBOLD could not be started"$ERED
2270                         echo -e $RED" Stopping script..."$ERED
2271                         exit 1
2272                 fi
2273                 let appindex=appindex+1
2274         done
2275         return 0
2276 }
2277
2278 # Function to check if container/service is responding to http/https
2279 # args: <container-name>|<service-name> url
2280 # (Not for test scripts)
2281 __check_service_start() {
2282
2283         if [ $# -ne 2 ]; then
2284                 ((RES_CONF_FAIL++))
2285                 __print_err "need 2 args, <container-name>|<service-name> url" $@
2286                 return 1
2287         fi
2288
2289         if [ $RUNMODE == "KUBE" ]; then
2290                 ENTITY="service/set/deployment"
2291         else
2292                 ENTITY="container"
2293         fi
2294         appname=$1
2295         url=$2
2296         echo -ne " Container $BOLD${appname}$EBOLD starting${SAMELINE}"
2297
2298
2299         pa_st=false
2300         echo -ne " Waiting for ${ENTITY} ${appname} service status...${SAMELINE}"
2301         TSTART=$SECONDS
2302         loop_ctr=0
2303         while (( $TSTART+600 > $SECONDS )); do
2304                 result="$(__do_curl -m 10 $url)"
2305                 if [ $? -eq 0 ]; then
2306                         if [ ${#result} -gt 15 ]; then
2307                                 #If response is too long, truncate
2308                                 result="...response text too long, omitted"
2309                         fi
2310                         echo -ne " Waiting for {ENTITY} $BOLD${appname}$EBOLD service status on ${3}, result: $result${SAMELINE}"
2311                         echo -ne " The ${ENTITY} $BOLD${appname}$EBOLD$GREEN is alive$EGREEN, responds to service status:$GREEN $result $EGREEN on ${url} after $(($SECONDS-$TSTART)) seconds"
2312                         pa_st=true
2313                         break
2314                 else
2315                         TS_TMP=$SECONDS
2316                         TS_OFFSET=$loop_ctr
2317                         if (( $TS_OFFSET > 5 )); then
2318                                 TS_OFFSET=5
2319                         fi
2320                         while [ $(($TS_TMP+$TS_OFFSET)) -gt $SECONDS ]; do
2321                                 echo -ne " Waiting for ${ENTITY} ${appname} service status on ${url}...$(($SECONDS-$TSTART)) seconds, retrying in $(($TS_TMP+$TS_OFFSET-$SECONDS)) seconds   ${SAMELINE}"
2322                                 sleep 1
2323                         done
2324                 fi
2325                 let loop_ctr=loop_ctr+1
2326         done
2327
2328         if [ "$pa_st" = "false"  ]; then
2329                 ((RES_CONF_FAIL++))
2330                 echo -e $RED" The ${ENTITY} ${appname} did not respond to service status on ${url} in $(($SECONDS-$TSTART)) seconds"$ERED
2331                 return 1
2332         fi
2333
2334         echo ""
2335         return 0
2336 }
2337
2338
2339 #################
2340 ### Log functions
2341 #################
2342
2343 __check_container_logs() {
2344
2345         dispname=$1
2346         appname=$2
2347         logpath=$3
2348         warning=$4
2349         error=$5
2350
2351         echo -e $BOLD"Checking $dispname container $appname log ($logpath) for WARNINGs and ERRORs"$EBOLD
2352
2353         if [ $RUNMODE == "KUBE" ]; then
2354                 echo -e $YELLOW" Internal log for $dispname not checked in kube"$EYELLOW
2355                 return
2356         fi
2357
2358         #tmp=$(docker ps | grep $appname)
2359         tmp=$(docker ps -q --filter name=$appname) #get the container id
2360         if [ -z "$tmp" ]; then  #Only check logs for running Policy Agent apps
2361                 echo " "$dispname" is not running, no check made"
2362                 return
2363         fi
2364         foundentries="$(docker exec -t $tmp grep $warning $logpath | wc -l)"
2365         if [ $? -ne  0 ];then
2366                 echo "  Problem to search $appname log $logpath"
2367         else
2368                 if [ $foundentries -eq 0 ]; then
2369                         echo "  No WARN entries found in $appname log $logpath"
2370                 else
2371                         echo -e "  Found \033[1m"$foundentries"\033[0m WARN entries in $appname log $logpath"
2372                 fi
2373         fi
2374         foundentries="$(docker exec -t $tmp grep $error $logpath | wc -l)"
2375         if [ $? -ne  0 ];then
2376                 echo "  Problem to search $appname log $logpath"
2377         else
2378                 if [ $foundentries -eq 0 ]; then
2379                         echo "  No ERR entries found in $appname log $logpath"
2380                 else
2381                         echo -e $RED"  Found \033[1m"$foundentries"\033[0m"$RED" ERR entries in $appname log $logpath"$ERED
2382                 fi
2383         fi
2384         echo ""
2385 }
2386
2387 # Store all container logs and other logs in the log dir for the script
2388 # Logs are stored with a prefix in case logs should be stored several times during a test
2389 # args: <logfile-prefix>
2390 # (Function for test scripts)
2391 store_logs() {
2392         if [ $# != 1 ]; then
2393                 ((RES_CONF_FAIL++))
2394         __print_err "need one arg, <file-prefix>" $@
2395                 exit 1
2396         fi
2397         echo -e $BOLD"Storing all docker/kube container logs and other test logs in $TESTLOGS/$ATC using prefix: "$1$EBOLD
2398
2399         docker stats --no-stream > $TESTLOGS/$ATC/$1_docker_stats.log 2>&1
2400
2401         docker ps -a  > $TESTLOGS/$ATC/$1_docker_ps.log 2>&1
2402
2403         cp .httplog_${ATC}.txt $TESTLOGS/$ATC/$1_httplog_${ATC}.txt 2>&1
2404
2405         if [ $RUNMODE == "DOCKER" ]; then
2406
2407                 # Store docker logs for all container
2408                 for imagename in $APP_SHORT_NAMES; do
2409                         __check_included_image $imagename
2410                         if [ $? -eq 0 ]; then
2411                                 # A function name is created from the app short name
2412                                 # for example app short name 'RICMSIM' -> produce the function
2413                                 # name __RICSIM__store_docker_logs
2414                                 # This function is called and is expected to exist in the imported
2415                                 # file for the ricsim test functions
2416                                 # The resulting function impl shall store the docker logs for each container
2417                                 function_pointer="__"$imagename"_store_docker_logs"
2418                                 $function_pointer "$TESTLOGS/$ATC/" $1
2419                         fi
2420                 done
2421         fi
2422         if [ $RUNMODE == "KUBE" ]; then
2423                 namespaces=$(kubectl  get namespaces -o jsonpath='{.items[?(@.metadata.name)].metadata.name}')
2424                 for nsid in $namespaces; do
2425                         pods=$(kubectl get pods -n $nsid -o jsonpath='{.items[?(@.metadata.labels.autotest)].metadata.name}')
2426                         for podid in $pods; do
2427                                 kubectl logs -n $nsid $podid > $TESTLOGS/$ATC/$1_${podid}.log
2428                         done
2429                 done
2430         fi
2431         echo ""
2432 }
2433
2434 ###############
2435 ## Generic curl
2436 ###############
2437 # Generic curl function, assumes all 200-codes are ok
2438 # args: <valid-curl-args-including full url>
2439 # returns: <returned response (without respose code)>  or "<no-response-from-server>" or "<not found, <http-code>>""
2440 # returns: The return code is 0 for ok and 1 for not ok
2441 __do_curl() {
2442         echo ${FUNCNAME[1]} "line: "${BASH_LINENO[1]} >> $HTTPLOG
2443         proxyflag=""
2444         if [ $RUNMODE == "KUBE" ]; then
2445                 if [ ! -z "$KUBE_PROXY_PATH" ]; then
2446                         if [ $KUBE_PROXY_HTTPX == "http" ]; then
2447                                 proxyflag=" --proxy $KUBE_PROXY_PATH"
2448                         else
2449                                 proxyflag=" --proxy-insecure --proxy $KUBE_PROXY_PATH"
2450                         fi
2451                 fi
2452         fi
2453         curlString="curl -skw %{http_code} $proxyflag $@"
2454         echo " CMD: $curlString" >> $HTTPLOG
2455         res=$($curlString)
2456         retcode=$?
2457         echo " RESP: $res" >> $HTTPLOG
2458         echo " RETCODE: $retcode" >> $HTTPLOG
2459         if [ $retcode -ne 0 ]; then
2460                 echo "<no-response-from-server>"
2461                 return 1
2462         fi
2463         http_code="${res:${#res}-3}"
2464         if [ ${#res} -eq 3 ]; then
2465                 if [ $http_code -lt 200 ] || [ $http_code -gt 299 ]; then
2466                         echo "<no-response-from-server>"
2467                         return 1
2468                 else
2469                         return 0
2470                 fi
2471         else
2472                 if [ $http_code -lt 200 ] || [ $http_code -gt 299 ]; then
2473                         echo "<not found, resp:${http_code}>"
2474                         return 1
2475                 fi
2476                 if [ $# -eq 2 ]; then
2477                         echo "${res:0:${#res}-3}" | xargs
2478                 else
2479                         echo "${res:0:${#res}-3}"
2480                 fi
2481
2482                 return 0
2483         fi
2484 }
2485
2486 #######################################
2487 ### Basic helper function for test cases
2488 #######################################
2489
2490 # Test a simulator container variable value towards target value using an condition operator with an optional timeout.
2491 # Arg: <simulator-name> <host> <variable-name> <condition-operator> <target-value>  - This test is done
2492 # immediately and sets pass or fail depending on the result of comparing variable and target using the operator.
2493 # Arg: <simulator-name> <host> <variable-name> <condition-operator> <target-value> <timeout>  - This test waits up to the timeout
2494 # before setting pass or fail depending on the result of comparing variable and target using the operator.
2495 # If the <variable-name> has the 'json:' prefix, the the variable will be used as url and the <target-value> will be compared towards the length of the json array in the response.
2496 # Not to be called from test script.
2497
2498 __var_test() {
2499         checkjsonarraycount=0
2500
2501         if [ $# -eq 6 ]; then
2502                 if [[ $3 == "json:"* ]]; then
2503                         checkjsonarraycount=1
2504                 fi
2505
2506                 echo -e $BOLD"TEST $TEST_SEQUENCE_NR (${BASH_LINENO[1]}): ${1}, ${3} ${4} ${5} within ${6} seconds"$EBOLD
2507                 ((RES_TEST++))
2508                 ((TEST_SEQUENCE_NR++))
2509                 start=$SECONDS
2510                 ctr=0
2511                 for (( ; ; )); do
2512                         if [ $checkjsonarraycount -eq 0 ]; then
2513                                 result="$(__do_curl $2$3)"
2514                                 retcode=$?
2515                                 result=${result//[[:blank:]]/} #Strip blanks
2516                         else
2517                                 path=${3:5}
2518                                 result="$(__do_curl $2$path)"
2519                                 retcode=$?
2520                                 echo "$result" > ./tmp/.tmp.curl.json
2521                                 result=$(python3 ../common/count_json_elements.py "./tmp/.tmp.curl.json")
2522                         fi
2523                         duration=$((SECONDS-start))
2524                         echo -ne " Result=${result} after ${duration} seconds${SAMELINE}"
2525                         let ctr=ctr+1
2526                         if [ $retcode -ne 0 ]; then
2527                                 if [ $duration -gt $6 ]; then
2528                                         ((RES_FAIL++))
2529                                         echo -e $RED" FAIL${ERED} - ${3} ${4} ${5} not reached in ${6} seconds, result = ${result}"
2530                                         __print_current_stats
2531                                         __check_stop_at_error
2532                                         return
2533                                 fi
2534                         elif [ $4 = "=" ] && [ "$result" -eq $5 ]; then
2535                                 ((RES_PASS++))
2536                                 echo -e " Result=${result} after ${duration} seconds${SAMELINE}"
2537                                 echo -e $GREEN" PASS${EGREEN} - Result=${result} after ${duration} seconds"
2538                                 __print_current_stats
2539                                 return
2540                         elif [ $4 = ">" ] && [ "$result" -gt $5 ]; then
2541                                 ((RES_PASS++))
2542                                 echo -e " Result=${result} after ${duration} seconds${SAMELINE}"
2543                                 echo -e $GREEN" PASS${EGREEN} - Result=${result} after ${duration} seconds"
2544                                 __print_current_stats
2545                                 return
2546                         elif [ $4 = "<" ] && [ "$result" -lt $5 ]; then
2547                                 ((RES_PASS++))
2548                                 echo -e " Result=${result} after ${duration} seconds${SAMELINE}"
2549                                 echo -e $GREEN" PASS${EGREEN} - Result=${result} after ${duration} seconds"
2550                                 __print_current_stats
2551                                 return
2552                         elif [ $4 = "contain_str" ] && [[ $result =~ $5 ]]; then
2553                                 ((RES_PASS++))
2554                                 echo -e " Result=${result} after ${duration} seconds${SAMELINE}"
2555                                 echo -e $GREEN" PASS${EGREEN} - Result=${result} after ${duration} seconds"
2556                                 __print_current_stats
2557                                 return
2558                         else
2559                                 if [ $duration -gt $6 ]; then
2560                                         ((RES_FAIL++))
2561                                         echo -e $RED" FAIL${ERED} - ${3} ${4} ${5} not reached in ${6} seconds, result = ${result}"
2562                                         __print_current_stats
2563                                         __check_stop_at_error
2564                                         return
2565                                 fi
2566                         fi
2567                         sleep 1
2568                 done
2569         elif [ $# -eq 5 ]; then
2570                 if [[ $3 == "json:"* ]]; then
2571                         checkjsonarraycount=1
2572                 fi
2573
2574                 echo -e $BOLD"TEST $TEST_SEQUENCE_NR (${BASH_LINENO[1]}): ${1}, ${3} ${4} ${5}"$EBOLD
2575                 ((RES_TEST++))
2576                 ((TEST_SEQUENCE_NR++))
2577                 if [ $checkjsonarraycount -eq 0 ]; then
2578                         result="$(__do_curl $2$3)"
2579                         retcode=$?
2580                         result=${result//[[:blank:]]/} #Strip blanks
2581                 else
2582                         path=${3:5}
2583                         result="$(__do_curl $2$path)"
2584                         retcode=$?
2585                         echo "$result" > ./tmp/.tmp.curl.json
2586                         result=$(python3 ../common/count_json_elements.py "./tmp/.tmp.curl.json")
2587                 fi
2588                 if [ $retcode -ne 0 ]; then
2589                         ((RES_FAIL++))
2590                         echo -e $RED" FAIL ${ERED}- ${3} ${4} ${5} not reached, result = ${result}"
2591                         __print_current_stats
2592                         __check_stop_at_error
2593                 elif [ $4 = "=" ] && [ "$result" -eq $5 ]; then
2594                         ((RES_PASS++))
2595                         echo -e $GREEN" PASS${EGREEN} - Result=${result}"
2596                         __print_current_stats
2597                 elif [ $4 = ">" ] && [ "$result" -gt $5 ]; then
2598                         ((RES_PASS++))
2599                         echo -e $GREEN" PASS${EGREEN} - Result=${result}"
2600                         __print_current_stats
2601                 elif [ $4 = "<" ] && [ "$result" -lt $5 ]; then
2602                         ((RES_PASS++))
2603                         echo -e $GREEN" PASS${EGREEN} - Result=${result}"
2604                         __print_current_stats
2605                 elif [ $4 = "contain_str" ] && [[ $result =~ $5 ]]; then
2606                         ((RES_PASS++))
2607                         echo -e $GREEN" PASS${EGREEN} - Result=${result}"
2608                         __print_current_stats
2609                 else
2610                         ((RES_FAIL++))
2611                         echo -e $RED" FAIL${ERED} - ${3} ${4} ${5} not reached, result = ${result}"
2612                         __print_current_stats
2613                         __check_stop_at_error
2614                 fi
2615         else
2616                 echo "Wrong args to __var_test, needs five or six args: <simulator-name> <host> <variable-name> <condition-operator> <target-value> [ <timeout> ]"
2617                 echo "Got:" $@
2618                 exit 1
2619         fi
2620 }