From b0470e8b9f9f93fb9100b772aa8c8359e6874942 Mon Sep 17 00:00:00 2001
From: Jimmy Ma <jimmy@airbyte.io>
Date: Thu, 29 Sep 2022 15:47:35 -0700
Subject: [PATCH] Add some services start validation to acceptance_tests.sh

---
 tools/bin/acceptance_test.sh | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/tools/bin/acceptance_test.sh b/tools/bin/acceptance_test.sh
index aa831d4c0c50ba..3b3e192bec4944 100755
--- a/tools/bin/acceptance_test.sh
+++ b/tools/bin/acceptance_test.sh
@@ -6,6 +6,18 @@ set -e
 
 assert_root
 
+## Helper functions
+
+get_epoch_time() {
+  date +'%s'
+}
+
+check_success() {
+  docker-compose ps | grep "^$1" | grep 'Exit 0' >/dev/null || (echo "$1 didn't run successfully"; exit 1)
+}
+
+##
+
 echo "Starting app..."
 
 # Detach so we can run subsequent commands
@@ -17,7 +29,26 @@ shutdown_cmd="docker-compose down -v || docker kill \$(docker ps -a -f volume=ai
 # trap "echo 'docker-compose logs:' && docker-compose logs -t --tail 1000 && $shutdown_cmd" EXIT
 
 echo "Waiting for services to begin"
-while [[ "$(curl -s -o /dev/null -w ''%{http_code}'' localhost:8000/api/v1/health)" != "200" ]]; do echo "Waiting for docker deployment.."; sleep 5; done
+starttime=`get_epoch_time`
+maxtime=300
+while [[ "$(curl -s -o /dev/null -w ''%{http_code}'' localhost:8000/api/v1/health)" != "200" ]];
+do
+  echo "Waiting for docker deployment.."
+  currenttime=`get_epoch_time`
+  if [[ $(( $currenttime - $starttime )) -gt $maxtime ]]; then
+    docker-compose ps
+    echo "Platform is taking more than ${maxtime}s to start. Aborting..."
+    exit 1
+  fi
+  sleep 5
+done
+
+# Getting a snapshot of the docker compose state
+docker-compose ps
+
+# Make sure init containers ran successfully
+check_success 'init'
+check_success 'airbyte-bootloader'
 
 echo "Running e2e tests via gradle"
 SUB_BUILD=PLATFORM USE_EXTERNAL_DEPLOYMENT=true ./gradlew :airbyte-tests:acceptanceTests --rerun-tasks --scan