rycus86/prometheus_flask_exporter

#!/bin/bash

cd "$(dirname "$0")"

_fail() {
    docker logs flask-multi-processes
    docker rm -f flask-multi-processes > /dev/null 2>&1
    exit 1
}

docker build -f Dockerfile -t flask-multi-processes ../../. > /dev/null || _fail
docker run -d --name flask-multi-processes -p 4000:4000 flask-multi-processes > /dev/null || _fail

echo 'Waiting for the server to start...'

for _ in $(seq 1 10); do
    if curl -fs http://localhost:4000/metrics > /dev/null; then
        break
    else
        sleep 0.2
    fi
done

echo 'Starting the tests...'

for _ in $(seq 1 10); do
    curl -s http://localhost:4000/test > /dev/null
    if [ "$?" != "0" ]; then
        echo 'Failed to request the test endpoint'
        _fail
    fi
done

curl -s http://localhost:4000/metrics \
  | grep 'flask_http_request_duration_seconds_count{method="GET",path="/test",status="200"} 10.0' \
  > /dev/null

curl -s http://localhost:4000/metrics \
  | grep 'flask_http_request_total{method="GET",status="200"} 10.0' \
  > /dev/null

if [ "$?" != "0" ]; then
    echo 'The expected metrics are not found'
    _fail
fi

docker rm -f flask-multi-processes > /dev/null
echo 'OK, all done'