chore(testinfra): more postgrest tests & add a TODO

Author: soedirgo

docker/all-in-one/healthcheck.sh

@@ -38,6 +38,7 @@ fail2ban-client status
 curl -sSfI "http://localhost:$PGEXPORTER_PORT/metrics"
 
 # vector is up (if starting logflare)
+# TODO: make this non-conditional once we set up local logflare for testinfra
 if [ -n "${LOGFLARE_API_KEY:-}" ]; then
   curl -sSfI "http://localhost:$VECTOR_API_PORT/health"
 fi

testinfra/README.md

@@ -3,7 +3,7 @@
 ## Prerequisites
 
 ```sh
-pip3 install docker pytest pytest-testinfra
+pip3 install docker pytest pytest-testinfra requests
 ```
 
 ## Running locally

testinfra/test_all_in_one.py

@@ -3,15 +3,16 @@
 from typing import cast
 import docker
 import pytest
+import requests
 import subprocess
 import testinfra
 
 all_in_one_image_tag = "supabase/all-in-one:testinfra"
 all_in_one_envs = {
     "POSTGRES_PASSWORD": "postgres",
     "JWT_SECRET": "super-secret-jwt-token-with-at-least-32-characters-long",
-    "ANON_KEY": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyAgCiAgICAicm9sZSI6ICJhbm9uIiwKICAgICJpc3MiOiAic3VwYWJhc2UtZGVtbyIsCiAgICAiaWF0IjogMTY0MTc2OTIwMCwKICAgICJleHAiOiAxNzk5NTM1NjAwCn0.dc_X5iR_VP_qT0zsiyj_I_OZ2T9FtRU2BBNWN8Bu4GE",
-    "SERVICE_ROLE_KEY": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyAgCiAgICAicm9sZSI6ICJzZXJ2aWNlX3JvbGUiLAogICAgImlzcyI6ICJzdXBhYmFzZS1kZW1vIiwKICAgICJpYXQiOiAxNjQxNzY5MjAwLAogICAgImV4cCI6IDE3OTk1MzU2MDAKfQ.DaYlNEoUrrEn2Ig7tqibS-PHK5vgusbcbo7X36XVt4Q",
+    "ANON_KEY": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJyb2xlIjoiYW5vbiIsImlzcyI6InN1cGFiYXNlLWRlbW8iLCJpYXQiOjE2NDE3NjkyMDAsImV4cCI6MTc5OTUzNTYwMH0.F_rDxRTPE8OU83L_CNgEGXfmirMXmMMugT29Cvc8ygQ",
+    "SERVICE_ROLE_KEY": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJyb2xlIjoic2VydmljZV9yb2xlIiwiaXNzIjoic3VwYWJhc2UtZGVtbyIsImlhdCI6MTY0MTc2OTIwMCwiZXhwIjoxNzk5NTM1NjAwfQ.5z-pJI1qwZg1LE5yavGLqum65WOnnaaI5eZ3V00pLww",
     "ADMIN_API_KEY": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJyb2xlIjoic3VwYWJhc2VfYWRtaW4iLCJpc3MiOiJzdXBhYmFzZS1kZW1vIiwiaWF0IjoxNjQxNzY5MjAwLCJleHAiOjE3OTk1MzU2MDB9.Y9mSNVuTw2TdfryoaqM5wySvwQemGGWfSe9ixcklVfM",
     "DATA_VOLUME_MOUNTPOINT": "/data",
     "MACHINE_TYPE": "shared_cpu_1x_512m",
@@ -74,6 +75,29 @@ def get_health(container: Container) -> str:
     container.remove(v=True, force=True)
 
 
-def test_postgrest_service(host):
+def test_postgrest_is_running(host):
     postgrest = host.supervisor("services:postgrest")
     assert postgrest.is_running
+
+
+def test_postgrest_responds_to_requests():
+    res = requests.get(
+        "http://localhost:8000/rest/v1/",
+        headers={
+            "apikey": all_in_one_envs["ANON_KEY"],
+            "authorization": f"Bearer {all_in_one_envs['ANON_KEY']}",
+        },
+    )
+    assert res.ok
+
+
+def test_postgrest_can_connect_to_db():
+    res = requests.get(
+        "http://localhost:8000/rest/v1/buckets",
+        headers={
+            "apikey": all_in_one_envs["SERVICE_ROLE_KEY"],
+            "authorization": f"Bearer {all_in_one_envs['SERVICE_ROLE_KEY']}",
+            "accept-profile": "storage",
+        },
+    )
+    assert res.ok