Added test database connection script

FredMagas · Aug 8, 2024 · c6fd116 · c6fd116
1 parent 8589038
commit c6fd116
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -24,6 +24,13 @@ jobs:
         source venv/bin/activate
         pip install -r requirements.txt
 
+    - name: Test database connection
+      env:
+        DATABASE_URL: ${{ secrets.DATABASE_URL }}
+      run: |
+        source venv/bin/activate
+        python test_db_connection.py
+
     - name: Make retry.sh executable
       run: chmod +x retry.sh
 

diff --git a/test_db_connection.py b/test_db_connection.py
@@ -0,0 +1,11 @@
+import psycopg2
+import os
+
+DATABASE_URL = os.getenv('DATABASE_URL')
+
+try:
+    conn = psycopg2.connect(DATABASE_URL)
+    print("Connection successful")
+except Exception as e:
+    print(f"Connection failed: {e}")
+    exit(1)