test/common/dmaapadp_api_functions.sh

   1 #!/bin/bash
   2
   3 #  ============LICENSE_START===============================================
   4 #  Copyright (C) 2021 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 container/service managemnt functions test functions for the Dmaap Adatper
  21
  22
  23 ################ Test engine functions ################
  24
  25 # Create the image var used during the test
  26 # arg: <image-tag-suffix> (selects staging, snapshot, release etc)
  27 # <image-tag-suffix> is present only for images with staging, snapshot,release tags
  28 __DMAAPADP_imagesetup() {
  29         __check_and_create_image_var DMAAPADP "DMAAP_ADP_IMAGE" "DMAAP_ADP_IMAGE_BASE" "DMAAP_ADP_IMAGE_TAG" $1 "$DMAAP_ADP_DISPLAY_NAME"
  30 }
  31
  32 # Pull image from remote repo or use locally built image
  33 # arg: <pull-policy-override> <pull-policy-original>
  34 # <pull-policy-override> Shall be used for images allowing overriding. For example use a local image when test is started to use released images
  35 # <pull-policy-original> Shall be used for images that does not allow overriding
  36 # Both var may contain: 'remote', 'remote-remove' or 'local'
  37 __DMAAPADP_imagepull() {
  38         __check_and_pull_image $1 "$DMAAP_ADP_DISPLAY_NAME" $DMAAP_ADP_APP_NAME DMAAP_ADP_IMAGE
  39 }
  40
  41 # Build image (only for simulator or interfaces stubs owned by the test environment)
  42 # arg: <image-tag-suffix> (selects staging, snapshot, release etc)
  43 # <image-tag-suffix> is present only for images with staging, snapshot,release tags
  44 __DMAAPADP_imagebuild() {
  45         echo -e $RED" Image for app DMAAPADP shall never be built"$ERED
  46 }
  47
  48 # Generate a string for each included image using the app display name and a docker images format string
  49 # If a custom image repo is used then also the source image from the local repo is listed
  50 # arg: <docker-images-format-string> <file-to-append>
  51 __DMAAPADP_image_data() {
  52         echo -e "$DMAAP_ADP_DISPLAY_NAME\t$(docker images --format $1 $DMAAP_ADP_IMAGE)" >>   $2
  53         if [ ! -z "$DMAAP_ADP_IMAGE_SOURCE" ]; then
  54                 echo -e "-- source image --\t$(docker images --format $1 $DMAAP_ADP_IMAGE_SOURCE)" >>   $2
  55         fi
  56 }
  57
  58 # Scale kubernetes resources to zero
  59 # All resources shall be ordered to be scaled to 0, if relevant. If not relevant to scale, then do no action.
  60 # This function is called for apps fully managed by the test script
  61 __DMAAPADP_kube_scale_zero() {
  62         __kube_scale_all_resources $KUBE_NONRTRIC_NAMESPACE autotest DMAAPADP
  63 }
  64
  65 # Scale kubernetes resources to zero and wait until this has been accomplished, if relevant. If not relevant to scale, then do no action.
  66 # This function is called for prestarted apps not managed by the test script.
  67 __DMAAPADP_kube_scale_zero_and_wait() {
  68         __kube_scale_and_wait_all_resources $KUBE_NONRTRIC_NAMESPACE app "$KUBE_NONRTRIC_NAMESPACE"-dmaapadapterservice
  69 }
  70
  71 # Delete all kube resouces for the app
  72 # This function is called for apps managed by the test script.
  73 __DMAAPADP_kube_delete_all() {
  74         __kube_delete_all_resources $KUBE_NONRTRIC_NAMESPACE autotest DMAAPADP
  75 }
  76
  77 # Store docker logs
  78 # This function is called for apps managed by the test script.
  79 # args: <log-dir> <file-prexix>
  80 __DMAAPADP_store_docker_logs() {
  81         if [ $RUNMODE == "KUBE" ]; then
  82                 kubectl  logs -l "autotest=DMAAPADP" -n $KUBE_NONRTRIC_NAMESPACE --tail=-1 > $1$2_dmaapadapter.log 2>&1
  83         else
  84                 docker logs $DMAAP_ADP_APP_NAME > $1$2_dmaapadapter.log 2>&1
  85         fi
  86 }
  87
  88 # Initial setup of protocol, host and ports
  89 # This function is called for apps managed by the test script.
  90 # args: -
  91 __DMAAPADP_initial_setup() {
  92         use_dmaapadp_http
  93 }
  94
  95 # Set app short-name, app name and namespace for logging runtime statistics of kubernets pods or docker containers
  96 # For docker, the namespace shall be excluded
  97 # This function is called for apps managed by the test script as well as for prestarted apps.
  98 # args: -
  99 __DMAAPADP_statisics_setup() {
 100         if [ $RUNMODE == "KUBE" ]; then
 101                 echo "DMAAPADP $DMAAP_ADP_APP_NAME $KUBE_NONRTRIC_NAMESPACE"
 102         else
 103                 echo "DMAAPADP $DMAAP_ADP_APP_NAME"
 104         fi
 105 }
 106
 107 #######################################################
 108
 109 # Set http as the protocol to use for all communication to the Dmaap adapter
 110 # args: -
 111 # (Function for test scripts)
 112 use_dmaapadp_http() {
 113         __dmaapadp_set_protocoll "http" $DMAAP_ADP_INTERNAL_PORT $DMAAP_ADP_EXTERNAL_PORT
 114 }
 115
 116 # Set https as the protocol to use for all communication to the Dmaap adapter
 117 # args: -
 118 # (Function for test scripts)
 119 use_dmaapadp_https() {
 120         __dmaapadp_set_protocoll "https" $DMAAP_ADP_INTERNAL_SECURE_PORT $DMAAP_ADP_EXTERNAL_SECURE_PORT
 121 }
 122
 123 # Setup paths to svc/container for internal and external access
 124 # args: <protocol> <internal-port> <external-port>
 125 __dmaapadp_set_protocoll() {
 126         echo -e $BOLD"$DMAAP_ADP_DISPLAY_NAME protocol setting"$EBOLD
 127         echo -e " Using $BOLD http $EBOLD towards $DMAAP_ADP_DISPLAY_NAME"
 128
 129         ## Access to Dmaap adapter
 130
 131         DMAAP_ADP_SERVICE_PATH=$1"://"$DMAAP_ADP_APP_NAME":"$2  # docker access, container->container and script->container via proxy
 132         if [ $RUNMODE == "KUBE" ]; then
 133                 DMAAP_ADP_SERVICE_PATH=$1"://"$DMAAP_ADP_APP_NAME.$KUBE_NONRTRIC_NAMESPACE":"$3 # kube access, pod->svc and script->svc via proxy
 134         fi
 135
 136         # DMAAP_ADP_ADAPTER used for switching between REST and DMAAP (only REST supported currently)
 137         DMAAP_ADP_ADAPTER_TYPE="REST"
 138         DMAAP_ADP_ADAPTER=$DMAAP_ADP_SERVICE_PATH
 139
 140         echo ""
 141 }
 142
 143 # Export env vars for config files, docker compose and kube resources
 144 # args: PROXY|NOPROXY
 145 __dmaapadp_export_vars() {
 146
 147         export DMAAP_ADP_APP_NAME
 148         export DMAAP_ADP_DISPLAY_NAME
 149
 150         export KUBE_NONRTRIC_NAMESPACE
 151         export DOCKER_SIM_NWNAME
 152
 153         export DMAAP_ADP_IMAGE
 154
 155         export DMAAP_ADP_INTERNAL_PORT
 156         export DMAAP_ADP_INTERNAL_SECURE_PORT
 157         export DMAAP_ADP_EXTERNAL_PORT
 158         export DMAAP_ADP_EXTERNAL_SECURE_PORT
 159
 160         export DMAAP_ADP_CONFIG_MOUNT_PATH
 161         export DMAAP_ADP_DATA_MOUNT_PATH
 162         export DMAAP_ADP_HOST_MNT_DIR
 163         export DMAAP_ADP_CONFIG_FILE
 164         export DMAAP_ADP_DATA_FILE
 165
 166         export DMAAP_ADP_CONFIG_CONFIGMAP_NAME=$DMAAP_ADP_APP_NAME"-config"
 167         export DMAAP_ADP_DATA_CONFIGMAP_NAME=$DMAAP_ADP_APP_NAME"-data"
 168
 169         export DMMAAP_ADP_PROXY_FLAG="false"
 170
 171         if [ $1 == "PROXY" ]; then
 172                 export DMAAP_ADP_HTTP_PROXY_CONFIG_PORT=$HTTP_PROXY_CONFIG_PORT  #Set if proxy is started
 173                 export DMAAP_ADP_HTTP_PROXY_CONFIG_HOST_NAME=$HTTP_PROXY_CONFIG_HOST_NAME #Set if proxy is started
 174                 if [ $DMAAP_ADP_HTTP_PROXY_CONFIG_PORT -eq 0 ] || [ -z "$DMAAP_ADP_HTTP_PROXY_CONFIG_HOST_NAME" ]; then
 175                         echo -e $YELLOW" Warning: HTTP PROXY will not be configured, proxy app not started"$EYELLOW
 176                 else
 177                         echo " Configured with http proxy"
 178                 fi
 179                 export DMMAAP_ADP_PROXY_FLAG="true"
 180         else
 181                 export DMAAP_ADP_HTTP_PROXY_CONFIG_PORT=0
 182                 export DMAAP_ADP_HTTP_PROXY_CONFIG_HOST_NAME=""
 183                 echo " Configured without http proxy"
 184         fi
 185
 186
 187         # paths to other components
 188         export ECS_SERVICE_PATH
 189         export DMAAP_ADP_SERVICE_PATH
 190         export MR_SERVICE_PATH
 191
 192 }
 193
 194 # Start the Dmaap Adapter
 195 # args: -
 196 # (Function for test scripts)
 197 start_dmaapadp() {
 198
 199         echo -e $BOLD"Starting $DMAAP_ADP_DISPLAY_NAME"$EBOLD
 200
 201         if [ $RUNMODE == "KUBE" ]; then
 202
 203                 # Check if app shall be fully managed by the test script
 204                 __check_included_image "DMAAPADP"
 205                 retcode_i=$?
 206
 207                 # Check if app shall only be used by the testscipt
 208                 __check_prestarted_image "DMAAPADP"
 209                 retcode_p=$?
 210
 211                 if [ $retcode_i -ne 0 ] && [ $retcode_p -ne 0 ]; then
 212                         echo -e $RED"The $DMAAP_ADP_APP_NAME app is not included as managed nor prestarted in this test script"$ERED
 213                         echo -e $RED"The $DMAAP_ADP_APP_NAME will not be started"$ERED
 214                         exit
 215                 fi
 216                 if [ $retcode_i -eq 0 ] && [ $retcode_p -eq 0 ]; then
 217                         echo -e $RED"The $DMAAP_ADP_APP_NAME app is included both as managed and prestarted in this test script"$ERED
 218                         echo -e $RED"The $DMAAP_ADP_APP_NAME will not be started"$ERED
 219                         exit
 220                 fi
 221
 222                 # Check if app shall be used - not managed - by the test script
 223                 if [ $retcode_p -eq 0 ]; then
 224                         echo -e " Using existing $DMAAP_ADP_APP_NAME deployment and service"
 225                         echo " Setting DMAAPADP replicas=1"
 226                         __kube_scale statefulset $DMAAP_ADP_APP_NAME $KUBE_NONRTRIC_NAMESPACE 1
 227                 fi
 228
 229                 if [ $retcode_i -eq 0 ]; then
 230                         echo -e " Creating $DMAAP_ADP_APP_NAME deployment and service"
 231
 232                         __kube_create_namespace $KUBE_NONRTRIC_NAMESPACE
 233
 234                         __dmaapadp_export_vars $1
 235
 236                         # Create config map for config
 237                         configfile=$PWD/tmp/$DMAAP_ADP_CONFIG_FILE
 238                         #cp $2 $configfile
 239                         envsubst < $2 > $configfile
 240                         output_yaml=$PWD/tmp/dmaapadp_cfc.yaml
 241                         __kube_create_configmap $DMAAP_ADP_CONFIG_CONFIGMAP_NAME $KUBE_NONRTRIC_NAMESPACE autotest DMAAPADP $configfile $output_yaml
 242
 243                         # Create config map for data
 244                         data_json=$PWD/tmp/$DMAAP_ADP_DATA_FILE
 245                         if [ $# -lt 3 ]; then
 246                                 #create empty dummy file
 247                                 echo "{}" > $data_json
 248                         else
 249                                 cp $3 $data_json
 250                         fi
 251                         output_yaml=$PWD/tmp/dmaapadp_cfd.yaml
 252                         __kube_create_configmap $DMAAP_ADP_DATA_CONFIGMAP_NAME $KUBE_NONRTRIC_NAMESPACE autotest DMAAPADP $data_json $output_yaml
 253
 254
 255                         # Create service
 256                         input_yaml=$SIM_GROUP"/"$DMAAP_ADP_COMPOSE_DIR"/"svc.yaml
 257                         output_yaml=$PWD/tmp/dmaapadp_svc.yaml
 258                         __kube_create_instance service $DMAAP_ADP_APP_NAME $input_yaml $output_yaml
 259
 260                         # Create app
 261                         input_yaml=$SIM_GROUP"/"$DMAAP_ADP_COMPOSE_DIR"/"app.yaml
 262                         output_yaml=$PWD/tmp/dmaapadp_app.yaml
 263                         __kube_create_instance app $DMAAP_ADP_APP_NAME $input_yaml $output_yaml
 264
 265                 fi
 266
 267                 __check_service_start $DMAAP_ADP_APP_NAME $DMAAP_ADP_SERVICE_PATH$DMAAP_ADP_ALIVE_URL
 268
 269         else
 270                 # Check if docker app shall be fully managed by the test script
 271                 __check_included_image 'DMAAPADP'
 272                 if [ $? -eq 1 ]; then
 273                         echo -e $RED"The $DMAAP_ADP_DISPLAY_NAME app is not included in this test script"$ERED
 274                         echo -e $RED"The $DMAAP_ADP_DISPLAY_NAME will not be started"$ERED
 275                         exit
 276                 fi
 277
 278                 __dmaapadp_export_vars $1
 279
 280                 dest_file=$SIM_GROUP/$DMAAP_ADP_COMPOSE_DIR/$DMAAP_ADP_HOST_MNT_DIR/$DMAAP_ADP_CONFIG_FILE
 281
 282                 envsubst < $2 > $dest_file
 283
 284                 dest_file=$SIM_GROUP/$DMAAP_ADP_COMPOSE_DIR/$DMAAP_ADP_HOST_MNT_DIR/$DMAAP_ADP_DATA_FILE
 285
 286                 if [ $# -lt 3 ]; then
 287                         #create empty dummy file
 288                         echo "{}" > $dest_file
 289                 else
 290                         envsubst < $3 > $dest_file
 291                 fi
 292
 293                 __start_container $DMAAP_ADP_COMPOSE_DIR "" NODOCKERARGS 1 $DMAAP_ADP_APP_NAME
 294
 295                 __check_service_start $DMAAP_ADP_APP_NAME $DMAAP_ADP_SERVICE_PATH$DMAAP_ADP_ALIVE_URL
 296         fi
 297         echo ""
 298 }
 299
 300 # Turn on trace level tracing
 301 # args: -
 302 # (Function for test scripts)
 303 set_dmaapadp_trace() {
 304         echo -e $BOLD"$DMAAP_ADP_DISPLAY_NAME trace logging"$EBOLD
 305         curlString="$DMAAP_ADP_SERVICE_PATH$DMAAP_ADP_ACTUATOR -X POST  -H Content-Type:application/json -d {\"configuredLevel\":\"trace\"}"
 306         result=$(__do_curl "$curlString")
 307         if [ $? -ne 0 ]; then
 308                 __print_err "could not set trace mode" $@
 309                 ((RES_CONF_FAIL++))
 310                 return 1
 311         fi
 312         echo ""
 313         return 0
 314 }