Skip to content

Commit f05a720

Browse files
committed
Update test_core.py
1 parent e19613a commit f05a720

File tree

1 file changed

+4
-21
lines changed

1 file changed

+4
-21
lines changed

test/e2e/test_core.py

Lines changed: 4 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -672,25 +672,8 @@ def test_get_jobs_as_user_flag_for_staff(mock_post, mock_get_all_jobs, mock_get_
672672

673673
mock_get_experiment_numbers_for_user_number.return_value = [1820497]
674674

675-
mock_get_all_jobs.return_value = [
676-
{
677-
"id": 1234,
678-
"state": "COMPLETED",
679-
"inputs": {},
680-
"outputs": None,
681-
"start": None,
682-
"end": None,
683-
"type": "JobType.AUTOREDUCTION",
684-
},
685-
{
686-
"id": 5678,
687-
"state": "FAILED",
688-
"inputs": {},
689-
"outputs": None,
690-
"start": None,
691-
"end": None,
692-
"type": "JobType.AUTOREDUCTION",
693-
},
694-
]
675+
response_as_user_true = client.get("/jobs?as_user=true", headers={"Authorization": f"Bearer {STAFF_TOKEN}"})
676+
677+
response_as_user_false = client.get("/jobs?as_user=false", headers={"Authorization": f"Bearer {STAFF_TOKEN}"})
695678

696-
client.get("/jobs?as_user=true", headers={"Authorization": f"Bearer {STAFF_TOKEN}"})
679+
assert len(response_as_user_true.response()) == len(response_as_user_false.response())

0 commit comments

Comments
 (0)