#!/usr/bin/env bash
# Restart test: remove the container, start a new one on the same volume, read the count.
# Then a counter-test: a container WITHOUT the volume. Run from the folder with common.py.
set -u
PY=${PY:-python}
wait_up() { until curl -sf "localhost:$1/api/v2/heartbeat" >/dev/null; do sleep 0.5; done; }

echo "== before"
$PY probe.py 8010

echo "== docker rm -f kb-chroma; ./run_chroma.sh (same volume kb-chroma-data)"
docker rm -f kb-chroma
./run_chroma.sh
wait_up 8010
$PY probe.py 8010

echo "== counter-test: new container, no volume, port 8011"
docker run -d --name kb-chroma-novol -p 127.0.0.1:8011:8000 -e ANONYMIZED_TELEMETRY=False chromadb/chroma:1.5.9
wait_up 8011
$PY probe.py 8011
docker rm -f kb-chroma-novol

echo "== volume size"
docker run --rm -v kb-chroma-data:/data alpine du -sh /data
docker run --rm -v kb-chroma-data:/data alpine sh -c 'ls -la /data; du -ak /data | sort -rn | head -6'
docker system df -v | grep -E "VOLUME NAME|kb-chroma"