111 lines
2.6 KiB
Python
111 lines
2.6 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
from fastapi import HTTPException, UploadFile, status
|
|
from icecream import ic
|
|
|
|
from app.controllers.musicians import MusicianController
|
|
from app.models.musician import Musician
|
|
|
|
mock_queries = MagicMock()
|
|
mc = MusicianController(musician_queries=mock_queries)
|
|
|
|
sample_data = [
|
|
{
|
|
"id": 1,
|
|
"name": "John Doe",
|
|
"bio": "A musician",
|
|
"headshot_id": "headshot123",
|
|
},
|
|
{
|
|
"id": 2,
|
|
"name": "Jane Doe",
|
|
"bio": "Another musician",
|
|
"headshot_id": "headshotABC",
|
|
},
|
|
]
|
|
|
|
bad_data = [
|
|
# no bio
|
|
{
|
|
"id": "three",
|
|
"name": "Jack Doe",
|
|
"headshot_id": "headshot456",
|
|
}
|
|
]
|
|
|
|
|
|
def mock_select_all_series():
|
|
return sample_data
|
|
|
|
|
|
def mock_select_all_series_sad():
|
|
return bad_data
|
|
|
|
|
|
def mock_select_one_by_id(musician_id: int):
|
|
for musician in sample_data:
|
|
if musician.get("id") == musician_id:
|
|
return musician
|
|
return None
|
|
|
|
|
|
mock_queries.select_all = mock_select_all_series
|
|
mock_queries.select_one_by_id = mock_select_one_by_id
|
|
|
|
|
|
def test_type():
|
|
assert isinstance(mc, MusicianController)
|
|
|
|
|
|
"""
|
|
TODO: write tests for following methods:
|
|
|
|
- _update_musician_headshot
|
|
- _update_musician_bio
|
|
- _upload_headshot
|
|
"""
|
|
|
|
|
|
def test_happy_get_musicians():
|
|
musicians = mc.get_musicians()
|
|
assert isinstance(musicians, list)
|
|
assert len(musicians) == 2
|
|
for musician in musicians:
|
|
assert isinstance(musician, Musician)
|
|
m1, m2 = musicians
|
|
assert m1.id == 1
|
|
assert m1.name == "John Doe"
|
|
assert m1.bio == "A musician"
|
|
assert m1.headshot_id == "headshot123"
|
|
assert m2.id == 2
|
|
assert m2.name == "Jane Doe"
|
|
assert m2.bio == "Another musician"
|
|
assert m2.headshot_id == "headshotABC"
|
|
|
|
|
|
def test_sad_get_musicians():
|
|
mock_queries.select_all = mock_select_all_series_sad
|
|
with pytest.raises(HTTPException) as e:
|
|
mc.get_musicians()
|
|
mock_queries.select_all = mock_select_all_series
|
|
assert isinstance(e.value, HTTPException)
|
|
assert e.value.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
|
|
|
|
|
|
def test_happy_get_musician():
|
|
musician = mc.get_musician(1)
|
|
assert isinstance(musician, Musician)
|
|
assert musician.id == 1
|
|
assert musician.name == "John Doe"
|
|
assert musician.bio == "A musician"
|
|
assert musician.headshot_id == "headshot123"
|
|
|
|
|
|
def test_musician_not_found():
|
|
with pytest.raises(HTTPException) as e:
|
|
mc.get_musician(3)
|
|
assert isinstance(e.value, HTTPException)
|
|
assert e.value.status_code == status.HTTP_404_NOT_FOUND
|
|
assert e.value.detail == "Musician not found"
|