2021-09-09 08:51:29 -08:00
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
|
2021-12-18 19:04:36 -09:00
|
|
|
from tests import data as test_data
|
|
|
|
from tests.utils.fixture_schemas import TestUser
|
|
|
|
|
2021-10-07 09:39:47 -08:00
|
|
|
|
2021-12-18 19:04:36 -09:00
|
|
|
class Routes:
|
|
|
|
def get_user_image(user_id: str, file_name: str = "profile.webp") -> str:
|
|
|
|
return f"/api/media/users/{user_id}/{file_name}"
|
2021-09-09 08:51:29 -08:00
|
|
|
|
2021-12-18 19:04:36 -09:00
|
|
|
def user_image(user_id: str) -> str:
|
|
|
|
return f"/api/users/{user_id}/image"
|
2021-09-09 08:51:29 -08:00
|
|
|
|
|
|
|
|
2021-12-18 19:04:36 -09:00
|
|
|
def test_user_get_image(api_client: TestClient, unique_user: TestUser):
|
|
|
|
# Get the user's image
|
|
|
|
response = api_client.get(Routes.get_user_image(str(unique_user.user_id)))
|
2021-09-09 08:51:29 -08:00
|
|
|
assert response.status_code == 200
|
|
|
|
|
2021-12-18 19:04:36 -09:00
|
|
|
# Ensure that the returned value is a valid image
|
|
|
|
assert response.headers["Content-Type"] == "image/webp"
|
2021-09-09 08:51:29 -08:00
|
|
|
|
|
|
|
|
2021-12-18 19:04:36 -09:00
|
|
|
def test_user_update_image(api_client: TestClient, unique_user: TestUser):
|
|
|
|
image = {"profile": test_data.images_test_image_1.read_bytes()}
|
2021-09-09 08:51:29 -08:00
|
|
|
|
2021-12-18 19:04:36 -09:00
|
|
|
# Update the user's image
|
|
|
|
response = api_client.post(Routes.user_image(str(unique_user.user_id)), files=image, headers=unique_user.token)
|
|
|
|
assert response.status_code == 200
|
|
|
|
|
|
|
|
# Request the image again
|
|
|
|
response = api_client.get(Routes.get_user_image(str(unique_user.user_id)))
|
|
|
|
assert response.status_code == 200
|