1
0
mirror of https://github.com/bec-project/bec_widgets.git synced 2026-05-02 04:52:32 +02:00
Files
bec_widgets/.github/scripts/setup_benchmark_env.sh
T
2026-04-20 16:25:51 +02:00

70 lines
2.1 KiB
Bash
Executable File

#!/usr/bin/env bash
set -euo pipefail
bec_core_branch="${BEC_CORE_BRANCH:-main}"
ophyd_devices_branch="${OPHYD_DEVICES_BRANCH:-main}"
plugin_repo_branch="${PLUGIN_REPO_BRANCH:-main}"
python_version="${PYTHON_VERSION:-3.11}"
if command -v conda >/dev/null 2>&1; then
conda_base="$(conda info --base)"
source "$conda_base/etc/profile.d/conda.sh"
fi
echo "Using branch ${bec_core_branch} of BEC CORE"
git clone --branch "$bec_core_branch" https://github.com/bec-project/bec.git
echo "Using branch ${ophyd_devices_branch} of OPHYD_DEVICES"
git clone --branch "$ophyd_devices_branch" https://github.com/bec-project/ophyd_devices.git
echo "Using branch ${plugin_repo_branch} of bec_testing_plugin"
git clone --branch "$plugin_repo_branch" https://github.com/bec-project/bec_testing_plugin.git
conda create -q -n test-environment "python=${python_version}"
conda activate test-environment
cd bec
source ./bin/install_bec_dev.sh -t
cd ..
python -m pip install -e ./ophyd_devices -e .[dev,pyside6] -e ./bec_testing_plugin
benchmark_tmp_dir="$(mktemp -d)"
export BEC_SERVICE_CONFIG="$benchmark_tmp_dir/services_config.yaml"
ready_file="$benchmark_tmp_dir/ready"
supervisor_log="$benchmark_tmp_dir/bec-benchmark-services.log"
python .github/scripts/start_bec_benchmark_services.py \
--files-path "$benchmark_tmp_dir" \
--services-config "$BEC_SERVICE_CONFIG" \
--ready-file "$ready_file" \
> "$supervisor_log" 2>&1 &
supervisor_pid=$!
cleanup_benchmark_services() {
if kill -0 "$supervisor_pid" >/dev/null 2>&1; then
kill "$supervisor_pid"
wait "$supervisor_pid" || true
fi
rm -rf "$benchmark_tmp_dir"
}
trap cleanup_benchmark_services EXIT
deadline=$((SECONDS + 30))
while [ ! -f "$ready_file" ]; do
if ! kill -0 "$supervisor_pid" >/dev/null 2>&1; then
cat "$supervisor_log" >&2 || true
echo "BEC benchmark service supervisor exited before becoming ready" >&2
exit 1
fi
if [ "$SECONDS" -ge "$deadline" ]; then
cat "$supervisor_log" >&2 || true
echo "Timed out waiting for BEC benchmark services" >&2
exit 1
fi
sleep 0.2
done
cat "$supervisor_log"
echo "BEC_SERVICE_CONFIG=$BEC_SERVICE_CONFIG" >> "$GITHUB_ENV"