Commit 8ecee728 authored by Thong Kuah's avatar Thong Kuah
Browse files

get_replicas and happy case test

parent 25163ff7

.gitlab-ci.yml

+14 −0
Original line number Diff line number Diff line
@@ -163,6 +163,20 @@ test-deploy-name: 
        exit 1

      fi



test-get-replicas:

  stage: test

  image: "$BUILD_IMAGE_NAME"

  variables:

    CI_ENVIRONMENT_SLUG: production

  script:

    - source /build/deploy-helpers.sh

    - replicas=$(get_replicas "stable" "100")

    - |

      if [[ $replicas != 1 ]]; then

        echo "$replicas should equal 1"

        exit 1

      fi



before_script:

  - |

    function expected_error() {

src/deploy-helpers.sh

+32 −0
Original line number Diff line number Diff line
@@ -88,3 +88,35 @@ function deploy_name() { 


  echo $name

}



function get_replicas() {

  track="${1:-stable}"

  percentage="${2:-100}"



  env_track=$( echo $track | tr -s  '[:lower:]'  '[:upper:]' )

  env_slug=$( echo ${CI_ENVIRONMENT_SLUG//-/_} | tr -s  '[:lower:]'  '[:upper:]' )



  if [[ "$track" == "stable" ]] || [[ "$track" == "rollout" ]]; then

    # for stable track get number of replicas from `PRODUCTION_REPLICAS`

    eval new_replicas=\$${env_slug}_REPLICAS

    if [[ -z "$new_replicas" ]]; then

      new_replicas=$REPLICAS

    fi

  else

    # for all tracks get number of replicas from `CANARY_PRODUCTION_REPLICAS`

    eval new_replicas=\$${env_track}_${env_slug}_REPLICAS

    if [[ -z "$new_replicas" ]]; then

      eval new_replicas=\${env_track}_REPLICAS

    fi

  fi



  replicas="${new_replicas:-1}"

  replicas="$(($replicas * $percentage / 100))"



  # always return at least one replicas

  if [[ $replicas -gt 0 ]]; then

    echo "$replicas"

  else

    echo 1

  fi

}