upated db

Author: TechWithTy

backend/apps/users/views/tests/test_main_view.py

@@ -15,7 +15,7 @@ class TestMainViews:
     """Integration tests for main script execution endpoints using real Supabase"""
 
     @pytest.fixture(autouse=True)
-    def setup(self, test_django_user, test_admin_django_user, django_db_blocker):
+    def setup(self, test_django_user, test_admin_django_user, django_db_blocker, ensure_test_tables):
         # Default script parameters
         self.script_params = {
             "parameters": {
@@ -49,19 +49,24 @@ def setup(self, test_django_user, test_admin_django_user, django_db_blocker):
         yield
 
         with django_db_blocker.unblock():
-            # Clean up after tests
-            low_credit_auth_user = self.low_credit_user.user
-            self.low_credit_user.delete()
-            low_credit_auth_user.delete()
-            CreditTransaction.objects.filter(user=low_credit_auth_user).delete()
-            
-            # Reset credits for test users
-            if hasattr(self.test_user, 'credits_balance'):
-                self.test_user.credits_balance = 1000
-                self.test_user.save()
-            if hasattr(self.admin_user, 'credits_balance'):
-                self.admin_user.credits_balance = 1000
-                self.admin_user.save()
+            # Clean up after tests - use try/except to handle missing tables
+            try:
+                # Clear transactions first to avoid foreign key constraint errors
+                from apps.credits.models import CreditTransaction
+                CreditTransaction.objects.filter(user=self.low_credit_user.user).delete()
+                
+                # Then delete the users
+                low_credit_auth_user = self.low_credit_user.user
+                self.low_credit_user.delete()
+                low_credit_auth_user.delete()
+                
+                # Reset credits for test users
+                if hasattr(self.test_user, 'credits_balance'):
+                    self.test_user.credits_balance = 1000
+                    self.test_user.save()
+            except Exception as e:
+                # Log the error but don't fail the test
+                print(f"Cleanup error: {str(e)}")
 
     def test_main_script_execution_success(self, monkeypatch, django_db_blocker):
         """Test successful script execution with real Supabase auth"""

backend/conftest.py

@@ -3,6 +3,7 @@
 import os
 import sys
 import django
+import pytest
 
 # Add the project root to the Python path
 sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
@@ -33,3 +34,28 @@ def pytest_configure(config):
 
     # Also filter out RuntimeWarnings about coroutines not being awaited
     warnings.filterwarnings("ignore", message="coroutine .* was never awaited")
+
+# Add fixture for properly creating tables in test database
+@pytest.fixture(scope='function')
+def ensure_test_tables(django_db_setup, django_db_blocker):
+    """Ensure all required tables exist in the test database before tests run"""
+    with django_db_blocker.unblock():
+        from django.core.management import call_command
+        # Migrate specific apps that might cause issues
+        call_command('migrate', 'credits', verbosity=0)
+        call_command('migrate', 'users', verbosity=0)
+
+# Override default databases to include supabase in test isolation
+@pytest.fixture(scope='function')
+def django_db_setup(request, django_db_setup, django_db_blocker):
+    """Custom database setup that adds supabase to available databases"""
+    from django.test import override_settings
+    
+    # Get the existing databases
+    from django.conf import settings
+    databases = list(settings.DATABASES.keys())
+    if 'supabase' not in databases:
+        databases.append('supabase')
+    
+    with override_settings(DATABASE_ROUTERS=[]):
+        yield