Skip to content

Commit 7eac58c

Browse files
committed
test: stabilize admin self-ban coverage
Signed-off-by: Grant Ramsay <[email protected]>
1 parent 33780a1 commit 7eac58c

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

tests/integration/test_user_routes.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -259,19 +259,23 @@ async def test_admin_cant_ban_self(
259259
self, client: AsyncClient, test_db: AsyncSession
260260
) -> None:
261261
"""Ensure an admin can ban a user."""
262-
admin_user = self.get_test_user(admin=True)
263-
test_db.add(User(**admin_user))
264-
token = AuthManager.encode_token(User(id=1))
262+
admin_user_data = self.get_test_user(admin=True)
263+
admin_user = User(**admin_user_data)
264+
test_db.add(admin_user)
265265

266+
await test_db.flush()
267+
admin_user_id = admin_user.id
266268
await test_db.commit()
267269

270+
token = AuthManager.encode_token(User(id=admin_user_id))
271+
268272
response = await client.post(
269-
"/users/1/ban",
273+
f"/users/{admin_user_id}/ban",
270274
headers={"Authorization": f"Bearer {token}"},
271275
)
272276

273277
check_not_banned = await client.get(
274-
"/users/?user_id=1",
278+
f"/users/?user_id={admin_user_id}",
275279
headers={"Authorization": f"Bearer {token}"},
276280
)
277281

0 commit comments

Comments
 (0)