Spaces:
Sleeping
Sleeping
test
Browse files- 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="
|
| 515 |
-
job2 = GeminiJob(job_id="
|
| 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()
|