Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add AsyncIO HTTP compression test #6975

Merged
merged 2 commits into from
Mar 13, 2024
Merged

Add AsyncIO HTTP compression test #6975

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
1aea16e
Add AsyncIO HTTP compression test
kthui Mar 11, 2024
1d86996
Improve command line option handling
kthui Mar 12, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
108 changes: 46 additions & 62 deletions qa/L0_data_compression/test.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,37 @@ fi
set -e

# End-to-end testing with simple model
function run_data_compression_infer_client() {
local client_path=$1
local request_algorithm=$2
local response_algorithm=$3
local log_path=$4

local python_or_cpp=`echo -n "$client_path" | tail -c 3`
fpetrini15 marked this conversation as resolved.
Show resolved Hide resolved
if [ "$python_or_cpp" == ".py" ]; then
local infer_client="python $client_path"
local request_cmd_option="--request-compression-algorithm $request_algorithm"
local response_cmd_option="--response-compression-algorithm $response_algorithm"
else # C++ if not end with ".py"
local infer_client=$client_path
local request_cmd_option="-i $request_algorithm"
local response_cmd_option="-o $response_algorithm"
fi

local cmd_options="-v"
if [ "$request_algorithm" != "" ]; then
cmd_options+=" $request_cmd_option"
fi
if [ "$response_algorithm" != "" ]; then
cmd_options+=" $response_cmd_option"
fi

$infer_client $cmd_options >> $log_path 2>&1
return $?
}

SIMPLE_INFER_CLIENT_PY=../clients/simple_http_infer_client.py
SIMPLE_AIO_INFER_CLIENT_PY=../clients/simple_http_aio_infer_client.py
SIMPLE_INFER_CLIENT=../clients/simple_http_infer_client

CLIENT_LOG=`pwd`/client.log
Expand All @@ -85,68 +115,22 @@ if [ "$SERVER_PID" == "0" ]; then
exit 1
fi

set +e
# Test various combinations
python $SIMPLE_INFER_CLIENT_PY -v --request-compression-algorithm deflate >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

python $SIMPLE_INFER_CLIENT_PY -v --request-compression-algorithm gzip >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

python $SIMPLE_INFER_CLIENT_PY -v --response-compression-algorithm deflate >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

python $SIMPLE_INFER_CLIENT_PY -v --response-compression-algorithm gzip >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

python $SIMPLE_INFER_CLIENT_PY -v --request-compression-algorithm deflate --response-compression-algorithm gzip >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

python $SIMPLE_INFER_CLIENT_PY -v --request-compression-algorithm gzip --response-compression-algorithm deflate >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

$SIMPLE_INFER_CLIENT -v -i deflate >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

$SIMPLE_INFER_CLIENT -v -i gzip >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

$SIMPLE_INFER_CLIENT -v -o deflate >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

$SIMPLE_INFER_CLIENT -v -o gzip >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

$SIMPLE_INFER_CLIENT -v -i deflate -o gzip >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi

$SIMPLE_INFER_CLIENT -v -i gzip -o deflate >> "${CLIENT_LOG}" 2>&1
if [ $? -ne 0 ]; then
RET=1
fi
set -e
for INFER_CLIENT in "$SIMPLE_INFER_CLIENT_PY" "$SIMPLE_AIO_INFER_CLIENT_PY" "$SIMPLE_INFER_CLIENT"; do
for REQUEST_ALGORITHM in "deflate" "gzip" ""; do
for RESPONSE_ALGORITHM in "deflate" "gzip" ""; do
if [ "$REQUEST_ALGORITHM" == "$RESPONSE_ALGORITHM" ]; then
continue
fi

set +e
run_data_compression_infer_client "$INFER_CLIENT" "$REQUEST_ALGORITHM" "$RESPONSE_ALGORITHM" "$CLIENT_LOG"
if [ $? -ne 0 ]; then
RET=1
fi
set -e
done
done
done

kill $SERVER_PID
wait $SERVER_PID
Expand Down
Loading