run e2e tests in parallel

This commit is contained in:
Adil Hafeez 2025-12-25 17:41:32 -08:00
parent 88d14a205b
commit d594526a60
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
5 changed files with 302 additions and 0 deletions

View file

@ -0,0 +1,52 @@
#!/bin/bash
# Run model alias routing e2e tests
# if any of the commands fail, the script will exit
set -e
. ./common_scripts.sh
print_disk_usage
mkdir -p ~/plano_logs
touch ~/plano_logs/modelserver.log
print_debug() {
log "Received signal to stop"
log "Printing debug logs for docker"
log "===================================="
tail -n 100 ../build.log
planoai logs --debug | tail -n 100
}
trap 'print_debug' INT TERM ERR
log starting > ../build.log
log building and installing plano cli
log ==================================
cd ../../cli
poetry install
cd -
log building docker image for arch gateway
log ======================================
cd ../../
planoai build
cd -
# Once we build plano we have to install the dependencies again to a new virtual environment.
poetry install
log startup arch gateway with model alias routing demo
cd ../../
planoai down
planoai up demos/use_cases/model_alias_routing/config_with_aliases.yaml
cd -
log running e2e tests for model alias routing
log ========================================
poetry run pytest test_model_alias_routing.py
log shutting down the arch gateway service
log =======================================
planoai down

View file

@ -0,0 +1,52 @@
#!/bin/bash
# Run OpenAI responses API e2e tests
# if any of the commands fail, the script will exit
set -e
. ./common_scripts.sh
print_disk_usage
mkdir -p ~/plano_logs
touch ~/plano_logs/modelserver.log
print_debug() {
log "Received signal to stop"
log "Printing debug logs for docker"
log "===================================="
tail -n 100 ../build.log
planoai logs --debug | tail -n 100
}
trap 'print_debug' INT TERM ERR
log starting > ../build.log
log building and installing plano cli
log ==================================
cd ../../cli
poetry install
cd -
log building docker image for arch gateway
log ======================================
cd ../../
planoai build
cd -
# Once we build plano we have to install the dependencies again to a new virtual environment.
poetry install
log startup arch gateway with model alias routing demo
cd ../../
planoai down
planoai up demos/use_cases/model_alias_routing/config_with_aliases.yaml
cd -
log running e2e tests for openai responses api client
log ================================================
poetry run pytest test_openai_responses_api_client.py
log shutting down the arch gateway service
log =======================================
planoai down

View file

@ -0,0 +1,52 @@
#!/bin/bash
# Run OpenAI responses API with state storage e2e tests
# if any of the commands fail, the script will exit
set -e
. ./common_scripts.sh
print_disk_usage
mkdir -p ~/plano_logs
touch ~/plano_logs/modelserver.log
print_debug() {
log "Received signal to stop"
log "Printing debug logs for docker"
log "===================================="
tail -n 100 ../build.log
planoai logs --debug | tail -n 100
}
trap 'print_debug' INT TERM ERR
log starting > ../build.log
log building and installing plano cli
log ==================================
cd ../../cli
poetry install
cd -
log building docker image for arch gateway
log ======================================
cd ../../
planoai build
cd -
# Once we build plano we have to install the dependencies again to a new virtual environment.
poetry install
log startup arch gateway with state storage for openai responses api client demo
cd ../../
planoai down
planoai up tests/e2e/config_memory_state_v1_responses.yaml
cd -
log running e2e tests for openai responses api client with state
log ============================================================
poetry run pytest test_openai_responses_api_client_with_state.py
log shutting down the arch gateway service
log =======================================
planoai down

View file

@ -0,0 +1,64 @@
#!/bin/bash
# Run prompt gateway e2e tests
# if any of the commands fail, the script will exit
set -e
. ./common_scripts.sh
print_disk_usage
mkdir -p ~/plano_logs
touch ~/plano_logs/modelserver.log
print_debug() {
log "Received signal to stop"
log "Printing debug logs for docker"
log "===================================="
tail -n 100 ../build.log
planoai logs --debug | tail -n 100
}
trap 'print_debug' INT TERM ERR
log starting > ../build.log
log building and running function_calling demo
log ===========================================
cd ../../demos/samples_python/weather_forecast/
docker compose up weather_forecast_service --build -d
cd -
log building and installing plano cli
log ==================================
cd ../../cli
poetry install
cd -
log building docker image for arch gateway
log ======================================
cd ../../
planoai build
cd -
# Once we build plano we have to install the dependencies again to a new virtual environment.
poetry install
log startup arch gateway with function calling demo
cd ../../
planoai down
planoai up demos/samples_python/weather_forecast/config.yaml
cd -
log running e2e tests for prompt gateway
log ====================================
poetry run pytest test_prompt_gateway.py
log shutting down the arch gateway service for prompt_gateway demo
log ===============================================================
planoai down
log shutting down the weather_forecast demo
log =======================================
cd ../../demos/samples_python/weather_forecast
docker compose down
cd -