mirror of
https://github.com/azaion/detections.git
synced 2026-04-25 19:46:32 +00:00
Roman Meshko
65 lines
1.8 KiB
Bash
Executable File
65 lines
1.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
PROFILE="${E2E_PROFILE:-cpu}"
|
|
case "$PROFILE" in
|
|
cpu)
|
|
DETECTIONS_SERVICE="detections"
|
|
;;
|
|
gpu)
|
|
DETECTIONS_SERVICE="detections-gpu"
|
|
;;
|
|
jetson)
|
|
DETECTIONS_SERVICE="detections-jetson"
|
|
;;
|
|
*)
|
|
echo "ERROR: unsupported E2E_PROFILE=$PROFILE (expected: cpu, gpu, jetson)"
|
|
exit 2
|
|
;;
|
|
esac
|
|
|
|
COMPOSE="docker compose -f docker-compose.test.yml --profile $PROFILE"
|
|
|
|
usage() {
|
|
echo "Usage: $0 <test_path> [pytest_args...]"
|
|
echo ""
|
|
echo "Examples:"
|
|
echo " $0 tests/test_video.py # run all tests in file"
|
|
echo " $0 tests/test_video.py::test_ft_p_10_frame_sampling_ac1 # run single test"
|
|
echo " $0 tests/test_video.py -k 'frame_sampling' # run by keyword"
|
|
echo ""
|
|
echo "Environment:"
|
|
echo " E2E_PROFILE=cpu|gpu|jetson compose profile to run (default: cpu)"
|
|
echo ""
|
|
echo "Flags -v -x -s are always included."
|
|
exit 1
|
|
}
|
|
|
|
[[ $# -lt 1 ]] && usage
|
|
|
|
$COMPOSE up -d --build "$DETECTIONS_SERVICE"
|
|
echo "--- Waiting for detections service to become healthy..."
|
|
for i in $(seq 1 60); do
|
|
if $COMPOSE exec -T "$DETECTIONS_SERVICE" python3 -c "import urllib.request; urllib.request.urlopen('http://localhost:8080/health')" 2>/dev/null; then
|
|
echo "--- Detections service is healthy"
|
|
break
|
|
fi
|
|
if [[ "$i" == "60" ]]; then
|
|
echo "ERROR: detections service did not become healthy"
|
|
$COMPOSE logs "$DETECTIONS_SERVICE" --tail 100
|
|
exit 1
|
|
fi
|
|
sleep 2
|
|
done
|
|
|
|
echo "--- Running: pytest $* -v -x -s --csv=/results/report.csv"
|
|
set +e
|
|
$COMPOSE run --rm --build --no-deps e2e-runner pytest "$@" -v -x -s --csv=/results/report.csv
|
|
EXIT_CODE=$?
|
|
set -e
|
|
|
|
echo "--- Test finished with exit code $EXIT_CODE"
|
|
echo "--- Detections logs (last 100 lines):"
|
|
$COMPOSE logs "$DETECTIONS_SERVICE" --tail 100
|
|
exit $EXIT_CODE
|