65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
from datetime import datetime
|
|
|
|
import pytest
|
|
|
|
from orm.author import Author
|
|
from orm.reaction import Reaction, ReactionKind
|
|
from orm.shout import Shout
|
|
|
|
|
|
@pytest.fixture
|
|
def test_setup(db_session):
|
|
"""Set up test data."""
|
|
now = int(datetime.now().timestamp())
|
|
author = Author(name="Test Author", slug="test-author", user="test-user-id")
|
|
db_session.add(author)
|
|
db_session.flush()
|
|
|
|
shout = Shout(
|
|
title="Test Shout",
|
|
slug="test-shout",
|
|
created_by=author.id,
|
|
body="This is a test shout",
|
|
layout="article",
|
|
lang="ru",
|
|
community=1,
|
|
created_at=now,
|
|
updated_at=now,
|
|
)
|
|
db_session.add_all([author, shout])
|
|
db_session.commit()
|
|
return {"author": author, "shout": shout}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_reaction(test_client, db_session, test_setup):
|
|
"""Test creating a reaction on a shout."""
|
|
response = test_client.post(
|
|
"/",
|
|
json={
|
|
"query": """
|
|
mutation CreateReaction($reaction: ReactionInput!) {
|
|
create_reaction(reaction: $reaction) {
|
|
error
|
|
reaction {
|
|
id
|
|
kind
|
|
body
|
|
created_by {
|
|
name
|
|
}
|
|
}
|
|
}
|
|
}
|
|
""",
|
|
"variables": {
|
|
"reaction": {"shout": test_setup["shout"].id, "kind": ReactionKind.LIKE.value, "body": "Great post!"}
|
|
},
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert "error" not in data
|
|
assert data["data"]["create_reaction"]["reaction"]["kind"] == ReactionKind.LIKE.value
|