chore: import upstream snapshot with attribution

tests/python_client/deploy/test.sh

@@ -0,0 +1,242 @@
+#!/bin/bash
+set -e
+# set -x
+
+func() {
+    echo "Usage:"
+    echo "test.sh [-t Task] [-m Mode] [-r Release] [-p Password]"
+    echo "Description"
+    echo "Task, the task type of test. reinstall or upgrade"
+    echo "Mode, the mode of milvus deploy. standalone or cluster"
+    echo "Release, the release of milvus. e.g. 2.0.0-rc8"
+    echo "Password, the password of root"
+    exit -1
+}
+
+
+echo "check os env"
+platform='unknown'
+unamestr=$(uname)
+if [[ "$unamestr" == 'Linux' ]]; then
+   platform='Linux'
+elif [[ "$unamestr" == 'Darwin' ]]; then
+   platform='Mac'
+fi
+echo "platform: $platform"
+
+
+
+Task="reinstall"
+Mode="standalone"
+Release="v2.0.0"
+while getopts "hm:t:p:" OPT;
+do
+    case $OPT in
+	m) Mode="$OPTARG";;
+    t) Task="$OPTARG";;
+	p) Password="$OPTARG";;
+	h) func;;
+	?) func;;
+    esac
+done
+
+ROOT_FOLDER=$(cd "$(dirname "$0")";pwd)
+
+# to export docker compose logs before exit
+function error_exit {
+    pushd ${ROOT_FOLDER}/${Deploy_Dir}
+    echo "test failed"
+    current=$(date "+%Y-%m-%d-%H-%M-%S")
+    if [ ! -d logs  ];
+    then
+        mkdir logs
+    fi
+    docker compose ps
+    docker compose logs > ./logs/${Deploy_Dir}-${Task}-${current}.log 2>&1
+    echo "log saved to $(pwd)/logs/${Deploy_Dir}-${Task}-${current}.log"
+    popd
+    exit 1
+}
+
+function replace_image_tag {
+    image_repo=$1
+    image_tag=$2
+    if [ "$platform" == "Mac" ];
+    then
+        # for mac os
+        sed -i "" "s/milvusdb\/milvus.*/${image_repo}\:${image_tag}/g" docker-compose.yml   
+    else
+        #for linux os 
+        sed -i "s/milvusdb\/milvus.*/${image_repo}\:${image_tag}/g" docker-compose.yml
+    fi
+
+}
+       
+
+#to check containers all running and minio is healthy
+function check_healthy {
+    cnt=$(docker compose ps | grep -E "running|Running|Up|up" | wc -l)
+    healthy=$(docker compose ps | grep "healthy" | wc -l)
+    time_cnt=0
+    echo "running num $cnt expect num $Expect"
+    echo "healthy num $healthy expect num $Expect_health"
+    while [[ $cnt -ne $Expect || $healthy -ne 1 ]];
+    do
+    printf "waiting all containers get running\n"
+    sleep 5
+    let time_cnt+=5
+    # if time is greater than 300s, the condition still not satisfied, we regard it as a failure
+    if [ $time_cnt -gt 300 ];
+    then
+        printf "timeout,there are some issue with deployment!"
+        error_exit
+    fi
+    cnt=$(docker compose ps | grep -E "running|Running|Up|up" | wc -l)
+    healthy=$(docker compose ps | grep "healthy" | wc -l)
+    echo "running num $cnt expect num $Expect"
+    echo "healthy num $healthy expect num $Expect_health"
+    done
+}
+
+Deploy_Dir=$Mode
+Task=$Task
+Release=$Release
+pw=$Password
+
+echo "mode: $Mode"
+echo "task: $Task"
+echo "password: $pw"
+## if needed, install dependency 
+#echo "install dependency"
+#pip install -r scripts/requirements.txt
+
+if [ ! -d ${Deploy_Dir}  ];
+then
+    mkdir ${Deploy_Dir}
+fi
+
+echo "get tag info"
+
+python scripts/get_tag.py
+
+latest_tag=$(jq -r ".latest_tag" tag_info.json)
+latest_rc_tag=$(jq -r ".latest_rc_tag" tag_info.json)
+# release_version="v${Release}"
+echo $latest_rc_tag
+
+pushd ${Deploy_Dir}
+# download docker-compose.yml
+wget https://github.com/milvus-io/milvus/releases/download/${latest_rc_tag}/milvus-${Mode}-docker-compose.yml -O docker-compose.yml
+ls
+# clean env to deploy a fresh milvus
+docker compose down
+docker compose ps
+echo "$pw"| sudo -S rm -rf ./volumes
+
+# first deployment
+if [ "$Task" == "reinstall" ];
+then
+    printf "download latest milvus docker compose yaml file from github\n"
+    wget https://raw.githubusercontent.com/milvus-io/milvus/master/deployments/docker/${Mode}/docker-compose.yml -O docker-compose.yml
+    printf "start to deploy latest rc tag milvus\n"
+    replace_image_tag "milvusdb\/milvus" $latest_tag
+fi
+if [ "$Task" == "upgrade" ];
+then
+    printf "start to deploy previous rc tag milvus\n"
+    replace_image_tag "milvusdb\/milvus" $latest_rc_tag # replace previous rc tag
+
+fi
+cat docker-compose.yml|grep milvusdb
+Expect=$(grep "container_name" docker-compose.yml | wc -l)
+Expect_health=$(grep "healthcheck" docker-compose.yml | wc -l)
+docker compose up -d
+check_healthy
+docker compose ps
+popd
+
+# test for first deployment
+printf "test for first deployment\n"
+if [ "$Task" == "reinstall" ];
+then
+    python scripts/action_before_reinstall.py || error_exit
+fi
+if [ "$Task" == "upgrade" ];
+then
+    python scripts/action_before_upgrade.py || error_exit
+fi
+
+pushd ${Deploy_Dir}
+# uninstall milvus
+printf "start to uninstall milvus\n"
+docker compose down
+sleep 10
+printf "check all containers removed\n"
+docker compose ps
+
+# second deployment
+if [ "$Task" == "reinstall" ];
+then
+    printf "start to reinstall milvus\n"
+    #because the task is reinstall, so don't change images tag
+fi
+if [ "$Task" == "upgrade" ];
+then
+    printf "start to upgrade milvus\n"
+    # because the task is upgrade, so replace image tag to latest, like rc4-->rc5
+    printf "download latest milvus docker compose yaml file from github\n"
+    wget https://raw.githubusercontent.com/milvus-io/milvus/master/deployments/docker/${Mode}/docker-compose.yml -O docker-compose.yml
+    printf "start to deploy latest rc tag milvus\n"
+    replace_image_tag "milvusdb\/milvus" $latest_tag
+
+fi
+cat docker-compose.yml|grep milvusdb
+docker compose up -d
+check_healthy
+docker compose ps
+popd
+
+# wait for milvus ready
+sleep 120
+
+# test for second deployment
+printf "test for second deployment\n"
+if [ "$Task" == "reinstall" ];
+then
+    python scripts/action_after_reinstall.py || error_exit
+fi
+if [ "$Task" == "upgrade" ];
+then
+    python scripts/action_after_upgrade.py || error_exit
+fi
+
+
+# test for third deployment(after docker compose restart)
+pushd ${Deploy_Dir}
+printf "start to restart milvus\n"
+docker compose restart
+check_healthy
+docker compose ps
+popd
+
+# wait for milvus ready
+sleep 120
+printf "test for third deployment\n"
+if [ "$Task" == "reinstall" ];
+then
+    python scripts/action_after_reinstall.py || error_exit
+fi
+if [ "$Task" == "upgrade" ];
+then
+    python scripts/action_after_upgrade.py || error_exit
+fi
+
+
+pushd ${Deploy_Dir}
+# clean env
+docker compose ps
+docker compose down
+sleep 10
+docker compose ps
+echo "$pw"|sudo -S rm -rf ./volumes
+popd