jebin2 commited on
Commit
08ca14b
·
1 Parent(s): 260bc17
Files changed (1) hide show
  1. tests/test_models.py +3 -2
tests/test_models.py CHANGED
@@ -294,6 +294,7 @@ class TestGeminiJobModel:
294
 
295
  # Query by priority
296
  result = await db_session.execute(
 
297
  select(GeminiJob).where(GeminiJob.priority == "fast")
298
  )
299
  jobs = result.scalars().all()
@@ -511,8 +512,8 @@ class TestModelRelationships:
511
  db_session.add(user)
512
  await db_session.commit()
513
 
514
- job1 = GeminiJob(job_id="job_fast", user_id=user.id, job_type="video")
515
- job2 = GeminiJob(job_id="job_medium", user_id=user.id, job_type="image")
516
 
517
  db_session.add_all([job1, job2])
518
  await db_session.commit()
 
294
 
295
  # Query by priority
296
  result = await db_session.execute(
297
+ GeminiJob.user_id == user.id # Filter by this user only
298
  select(GeminiJob).where(GeminiJob.priority == "fast")
299
  )
300
  jobs = result.scalars().all()
 
512
  db_session.add(user)
513
  await db_session.commit()
514
 
515
+ job1 = GeminiJob(job_id="job_rel_1", user_id=user.id, job_type="video")
516
+ job2 = GeminiJob(job_id="job_rel_2", user_id=user.id, job_type="image")
517
 
518
  db_session.add_all([job1, job2])
519
  await db_session.commit()