versitygw/tests/setup.sh

#!/usr/bin/env bats

# Copyright 2024 Versity Software
# This file is licensed under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http:#www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

source ./tests/env.sh
source ./tests/report.sh
source ./tests/setup_mc.sh
source ./tests/drivers/delete_bucket/delete_bucket_rest.sh
source ./tests/util/util_object.sh
source ./tests/versity.sh

check_secrets_line() {
  if [[ $secrets_line =~ ^(USER_ID_(ADMIN|USERPLUS|USER)_[0-9])= ]]; then
    match=${BASH_REMATCH[1]}
    role=$(echo -n "${BASH_REMATCH[2]}" | tr '[:upper:]' '[:lower:]')

    if [ -z "${!match}" ]; then
      log 2 "$match secrets parameter missing"
      return 1
    fi
    username_env="${match/USER_ID/USERNAME}"
    password_env="${match/USER_ID/PASSWORD}"
    if [ -z "${!username_env}" ]; then
      log 2 "$username_env secrets parameter missing"
      return 1
    fi
    if [ -z "${!password_env}" ]; then
      log 2 "$password_env secrets parameter missing"
      return 1
    fi
    local user_exists_code=0
    user_exists "${!username_env}" || user_exists_code=$?
    if [ $user_exists_code -eq 2 ]; then
      log 2 "error checking for user existence"
      return 1
    fi
    if [ $user_exists_code -eq 1 ] && ! create_user_versitygw "${!username_env}" "${!password_env}" "$role"; then
      log 2 "error creating user"
      return 1
    fi
  fi
  return 0
}

static_user_versitygw_setup() {
  while read -r secrets_line || [ -n "$secrets_line" ]; do
    if ! check_secrets_line; then
      return 1
    fi
  done < "$SECRETS_FILE"
}

# bats setup function
setup() {
  TEST_ID=$(date +"%Y%m%d-%H%M%S").$(basename "$BATS_TEST_FILENAME").$BATS_TEST_NAME
  export TEST_ID

  source_config_file

  if ! setup_test_log_file; then
    log 1 "error creating test log file"
    return 1
  fi

  if [ -n "$TEST_LOG_FILE" ]; then
    printf "\n%s\n\n" "**** $TEST_ID ****" >> "${TEST_LOG_FILE}.${TEST_ID}"
  fi

  if ! base_setup; then
    log 2 "error checking env vars or running versitygw"
    return 1
  fi

  if [ -n "$TEST_LOG_FILE" ]; then
    printf "\n%s\n\n" "**** $TEST_ID ****" >> "$TEST_LOG_FILE.$TEST_ID"
  fi

  if [ "$RUN_USERS" == "true" ] && [ "$DIRECT" != "true" ] && [ "$CREATE_STATIC_USERS_IF_NONEXISTENT" == "true" ]; then
    if ! static_user_versitygw_setup; then
      log 2 "error setting up static versitygw users"
      return 1
    fi
  fi

  log 4 "Running test $BATS_TEST_NAME"
  if [[ $LOG_LEVEL -ge 5 ]] || [[ -n "$TIME_LOG" ]]; then
    START_TIME=$(date +%s)
    export START_TIME
  fi

  if [[ $RUN_S3CMD == true ]]; then
    S3CMD_OPTS=()
    S3CMD_OPTS+=(-c "$S3CMD_CONFIG")
    S3CMD_OPTS+=(--access_key="$AWS_ACCESS_KEY_ID")
    S3CMD_OPTS+=(--secret_key="$AWS_SECRET_ACCESS_KEY")
    S3CMD_OPTS+=(--region="$AWS_REGION")
    export S3CMD_CONFIG S3CMD_OPTS
  fi

  if [[ $RUN_MC == true ]] && ! check_add_mc_alias; then
    log 1 "error checking for or adding mc alias"
    return 1
  fi

  export AWS_PROFILE
  log 4 "********** END SETUP **********"
  return 0
}

# bats teardown function
teardown() {
  local response proc_status_one proc_status_two

  response=$(check_versity_process_status)
  read -r proc_status_one proc_status_two <<< "$response"

  if [ "$proc_status_one" == "none" ] || [[ ( "$proc_status_one" == "running" ) && (( "$proc_status_two" == "none") || ( "$proc_status_two" == "running")) ]]; then
    bucket_and_user_cleanup
  fi
  if [ "$proc_status_one" == "running" ]; then
    stop_versity_process "$VERSITYGW_PID_1"
  fi
  if [ "$proc_status_two" == "running" ]; then
    stop_versity_process "$VERSITYGW_PID_2"
  fi
  remove_test_file_folder_if_desired
  teardown_logs
}

bucket_and_user_cleanup() {
  log 4 "********** BEGIN TEARDOWN **********"
  if [ "$DELETE_BUCKETS_AFTER_TEST" != "false" ] && ! cleanup_buckets; then
    log 3 "error cleaning up buckets after test"
  fi
  if [ "$SKIP_USERS_TESTS" != "true" ]; then
    if [ -n "$USERNAME_ONE" ]; then
      if user_exists "$USERNAME_ONE" && ! delete_user "$USERNAME_ONE"; then
        log 3 "error deleting user $USERNAME_ONE"
      fi
    fi
    if [ -n "$USERNAME_TWO" ]; then
      if user_exists "$USERNAME_TWO" && ! delete_user "$USERNAME_TWO"; then
        log 3 "error deleting user $USERNAME_TWO"
      fi
    fi
  fi
  if [ "$AUTOGENERATE_USERS" == "true" ] && ! delete_autogenerated_users; then
    log 3 "error deleting autocreated users"
  fi
  return 0
}