ducklet
32bbfe881b
The new format is much closer to the format used by /groups/ratings. Also allows to filter based on Unwind's ID.
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
from starlette.testclient import TestClient
|
|
import pytest
|
|
|
|
from unwind import create_app
|
|
from unwind import db, models, imdb
|
|
|
|
pytestmark = pytest.mark.asyncio
|
|
|
|
app = create_app()
|
|
|
|
|
|
async def test_app():
|
|
await db.open_connection_pool()
|
|
conn = db.shared_connection()
|
|
|
|
async with conn.transaction(force_rollback=True):
|
|
|
|
client = TestClient(app)
|
|
response = client.get("/api/v1/movies")
|
|
assert response.status_code == 403
|
|
|
|
client.auth = "user1", "secret1"
|
|
|
|
response = client.get("/api/v1/movies")
|
|
assert response.status_code == 200
|
|
assert response.json() == []
|
|
|
|
m = models.Movie(
|
|
title="test movie",
|
|
release_year=2013,
|
|
media_type="Movie",
|
|
imdb_id="tt12345678",
|
|
genres={"genre-1"},
|
|
)
|
|
await db.add(m)
|
|
|
|
response = client.get("/api/v1/movies", params={"include_unrated": 1})
|
|
assert response.status_code == 200
|
|
assert response.json() == [{**db.asplain(m), "user_scores": []}]
|
|
|
|
m_plain = {
|
|
"unwind_id": m.id,
|
|
"canonical_title": m.title,
|
|
"imdb_score": m.imdb_score,
|
|
"imdb_votes": m.imdb_votes,
|
|
"link": imdb.movie_url(m.imdb_id),
|
|
"media_type": m.media_type,
|
|
"original_title": m.original_title,
|
|
"user_scores": [],
|
|
"year": m.release_year,
|
|
}
|
|
|
|
response = client.get("/api/v1/movies", params={"imdb_id": m.imdb_id})
|
|
assert response.status_code == 200
|
|
assert response.json() == [m_plain]
|
|
|
|
response = client.get("/api/v1/movies", params={"unwind_id": str(m.id)})
|
|
assert response.status_code == 200
|
|
assert response.json() == [m_plain]
|
|
|
|
await db.close_connection_pool()
|