opencode-a2a/scripts/smoke_test_built_cli.sh at main · Intelligent-Internet/opencode-a2a · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
#!/usr/bin/env bash
# Validate that a locally built distribution artifact can be installed as a uv tool and serves /health.
set -euo pipefail

if [[ "$#" -gt 1 ]]; then
  echo "Expected at most one built artifact path argument" >&2
  exit 1
fi

if ! command -v uv >/dev/null 2>&1; then
  echo "uv not found in PATH" >&2
  exit 1
fi

if ! command -v curl >/dev/null 2>&1; then
  echo "curl not found in PATH" >&2
  exit 1
fi

python_bin="${PYTHON_BIN:-}"
if [[ -z "${python_bin}" ]]; then
  if command -v python3 >/dev/null 2>&1; then
    python_bin="python3"
  elif command -v python >/dev/null 2>&1; then
    python_bin="python"
  else
    echo "python3 or python not found in PATH" >&2
    exit 1
  fi
fi

artifact_path="${1:-${SMOKE_TEST_ARTIFACT_PATH:-${SMOKE_TEST_WHEEL_PATH:-}}}"

if [[ -z "${artifact_path}" ]]; then
  shopt -s nullglob
  wheel_paths=(dist/opencode_a2a-*.whl)
  shopt -u nullglob

  if [[ "${#wheel_paths[@]}" -eq 0 ]]; then
    echo "No built wheel found in dist/" >&2
    exit 1
  fi

  if [[ "${#wheel_paths[@]}" -gt 1 ]]; then
    echo "Multiple built wheels found; pass an explicit artifact path or set SMOKE_TEST_ARTIFACT_PATH." >&2
    printf ' - %s\n' "${wheel_paths[@]}" >&2
    exit 1
  fi

  artifact_path="${wheel_paths[0]}"
fi

if [[ ! -f "${artifact_path}" ]]; then
  echo "Artifact path does not exist: ${artifact_path}" >&2
  exit 1
fi
tmpdir="$(mktemp -d)"
tool_dir="${tmpdir}/tools"
tool_bin_dir="${tmpdir}/bin"
server_log="${tmpdir}/server.log"

cleanup() {
  local exit_code="$1"
  if [[ -n "${server_pid:-}" ]] && kill -0 "${server_pid}" >/dev/null 2>&1; then
    kill "${server_pid}" >/dev/null 2>&1 || true
    wait "${server_pid}" >/dev/null 2>&1 || true
  fi
  rm -rf "${tmpdir}"
  exit "${exit_code}"
}

trap 'cleanup $?' EXIT

mkdir -p "${tool_dir}" "${tool_bin_dir}"

UV_TOOL_DIR="${tool_dir}" \
UV_TOOL_BIN_DIR="${tool_bin_dir}" \
uv tool install "${artifact_path}" --python "${python_bin}"

mapfile -t installed_python_paths < <(
  find "${tool_dir}" \( -type f -o -type l \) -path '*/bin/python' | sort
)
if [[ "${#installed_python_paths[@]}" -ne 1 ]]; then
  echo "Expected exactly one installed tool python, found ${#installed_python_paths[@]}" >&2
  printf ' - %s\n' "${installed_python_paths[@]}" >&2 || true
  exit 1
fi
installed_python="${installed_python_paths[0]}"

"${installed_python}" -c "import opencode_a2a; print(opencode_a2a.__version__)" >/dev/null

port="$(
  "${python_bin}" - <<'PY'
import socket

with socket.socket() as sock:
    sock.bind(("127.0.0.1", 0))
    print(sock.getsockname()[1])
PY
)"

bearer_token="smoke-test-token"

A2A_STATIC_AUTH_CREDENTIALS="[{\"scheme\":\"bearer\",\"token\":\"${bearer_token}\",\"principal\":\"automation\"}]" \
A2A_PORT="${port}" \
A2A_HOST="127.0.0.1" \
"${tool_bin_dir}/opencode-a2a" serve >"${server_log}" 2>&1 &
server_pid="$!"

health_url="http://127.0.0.1:${port}/health"
wait_for_health() {
  local attempt
  for attempt in $(seq 1 50); do
    if ! kill -0 "${server_pid}" >/dev/null 2>&1; then
      echo "CLI smoke test failed; server exited before becoming healthy at ${health_url}" >&2
      cat "${server_log}" >&2
      return 1
    fi

    if curl --silent --fail --output /dev/null \
      -H "Authorization: Bearer ${bearer_token}" \
      "${health_url}" 2>/dev/null; then
      return 0
    fi

    sleep 0.2
  done

  echo "CLI smoke test failed; server did not become healthy at ${health_url}" >&2
  cat "${server_log}" >&2
  return 1
}

wait_for_health