open-law/tests/test_upvote.py

from flask import current_app as Response
from flask.testing import FlaskClient

from app import models as m
from tests.utils import login


def test_upvote_interpretation(client: FlaskClient):
    _, user = login(client)

    response: Response = client.post(
        "/vote/interpretation/999",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=True,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 404
    assert response.json["message"] == "Interpretation not found"

    interpretation = m.Interpretation(
        text="Test Interpretation 1 Text",
        user_id=user.id,
    ).save()

    assert interpretation.vote_count == 0

    response: Response = client.post(
        f"/vote/interpretation/{interpretation.id}",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=True,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 200
    json = response.json
    assert json
    assert "vote_count" in json
    assert json["vote_count"] == 1
    assert "current_user_vote" in json
    assert json["current_user_vote"]
    assert interpretation.vote_count == 1

    response: Response = client.post(
        f"/vote/interpretation/{interpretation.id}",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=True,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 200
    json = response.json
    assert json
    assert "vote_count" in json
    assert json["vote_count"] == 0
    assert "current_user_vote" in json
    assert json["current_user_vote"] is None
    assert interpretation.vote_count == 0

    response: Response = client.post(
        f"/vote/interpretation/{interpretation.id}",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=False,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 200
    json = response.json
    assert json
    assert "vote_count" in json
    assert json["vote_count"] == -1
    assert "current_user_vote" in json
    assert not json["current_user_vote"]
    assert interpretation.vote_count == -1

    response: Response = client.post(
        f"/vote/interpretation/{interpretation.id}",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=False,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 200
    json = response.json
    assert json
    assert "vote_count" in json
    assert json["vote_count"] == 0
    assert "current_user_vote" in json
    assert json["current_user_vote"] is None
    assert interpretation.vote_count == 0


def test_upvote_comment(client: FlaskClient):
    _, user = login(client)

    response: Response = client.post(
        "/vote/comment/999",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=True,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 404
    assert response.json["message"] == "Comment not found"

    comment = m.Comment(
        text="Test Comment 1 Text",
        user_id=user.id,
    ).save()

    assert comment.vote_count == 0

    response: Response = client.post(
        f"/vote/comment/{comment.id}",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=True,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 200
    json = response.json
    assert json
    assert "vote_count" in json
    assert json["vote_count"] == 1
    assert "current_user_vote" in json
    assert json["current_user_vote"]
    assert comment.vote_count == 1

    response: Response = client.post(
        f"/vote/comment/{comment.id}",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=True,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 200
    json = response.json
    assert json
    assert "vote_count" in json
    assert json["vote_count"] == 0
    assert "current_user_vote" in json
    assert json["current_user_vote"] is None
    assert comment.vote_count == 0

    response: Response = client.post(
        f"/vote/comment/{comment.id}",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=False,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 200
    json = response.json
    assert json
    assert "vote_count" in json
    assert json["vote_count"] == -1
    assert "current_user_vote" in json
    assert not json["current_user_vote"]
    assert comment.vote_count == -1

    response: Response = client.post(
        f"/vote/comment/{comment.id}",
        headers={"Content-Type": "application/json"},
        json=dict(
            positive=False,
        ),
        follow_redirects=True,
    )

    assert response
    assert response.status_code == 200
    json = response.json
    assert json
    assert "vote_count" in json
    assert json["vote_count"] == 0
    assert "current_user_vote" in json
    assert json["current_user_vote"] is None
    assert comment.vote_count == 0