From df8d1771e51e082bec43151c6832f5f3b16acaf8 Mon Sep 17 00:00:00 2001
From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
Date: Mon, 8 Dec 2025 09:02:16 -0600
Subject: [PATCH] fix tests

Fix migration 030 by using raw sql to select usernames (avoid ORM selecting nonexistent columns)
---
 migrations/030_create_user_review_status.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/migrations/030_create_user_review_status.py b/migrations/030_create_user_review_status.py
index 17f2b36b9..38937f7f9 100644
--- a/migrations/030_create_user_review_status.py
+++ b/migrations/030_create_user_review_status.py
@@ -54,7 +54,9 @@ def migrate(migrator, database, fake=False, **kwargs):
 
     # Migrate existing has_been_reviewed data to UserReviewStatus for all users
     def migrate_data():
-        all_users = list(User.select())
+        # Use raw SQL to avoid ORM issues with columns that don't exist yet
+        cursor = database.execute_sql('SELECT "username" FROM "user"')
+        all_users = cursor.fetchall()
         if not all_users:
             return
 
@@ -63,7 +65,7 @@ def migrate(migrator, database, fake=False, **kwargs):
         )
         reviewed_segment_ids = [row[0] for row in cursor.fetchall()]
         # also migrate for anonymous (unauthenticated users)
-        usernames = [user.username for user in all_users] + ["anonymous"]
+        usernames = [user[0] for user in all_users] + ["anonymous"]
 
         for segment_id in reviewed_segment_ids:
             for username in usernames: