1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
from datetime import datetime
from unittest.mock import patch
import pytest
from szurubooru import db
from szurubooru.func import comments, users
def test_serialize_user(user_factory, comment_factory):
with patch("szurubooru.func.users.get_avatar_url"):
users.get_avatar_url.return_value = "https://example.com/avatar.png"
comment = comment_factory(user=user_factory(name="dummy"))
comment.comment_id = 77
comment.creation_time = datetime(1997, 1, 1)
comment.last_edit_time = datetime(1998, 1, 1)
comment.text = "text"
db.session.add(comment)
db.session.flush()
auth_user = user_factory()
assert comments.serialize_comment(comment, auth_user) == {
"id": comment.comment_id,
"postId": comment.post.post_id,
"creationTime": datetime(1997, 1, 1, 0, 0),
"lastEditTime": datetime(1998, 1, 1, 0, 0),
"score": 0,
"ownScore": 0,
"text": "text",
"user": {
"name": "dummy",
"avatarUrl": "https://example.com/avatar.png",
},
"version": 1,
}
def test_try_get_comment(comment_factory):
comment = comment_factory()
db.session.add(comment)
db.session.flush()
assert comments.try_get_comment_by_id(comment.comment_id + 1) is None
assert comments.try_get_comment_by_id(comment.comment_id) is comment
def test_get_comment(comment_factory):
comment = comment_factory()
db.session.add(comment)
db.session.flush()
with pytest.raises(comments.CommentNotFoundError):
comments.get_comment_by_id(comment.comment_id + 1)
assert comments.get_comment_by_id(comment.comment_id) is comment
def test_create_comment(user_factory, post_factory, fake_datetime):
user = user_factory()
post = post_factory()
db.session.add_all([user, post])
with patch("szurubooru.func.comments.update_comment_text"), fake_datetime(
"1997-01-01"
):
comment = comments.create_comment(user, post, "text")
assert comment.creation_time == datetime(1997, 1, 1)
assert comment.user == user
assert comment.post == post
comments.update_comment_text.assert_called_once_with(comment, "text")
def test_update_comment_text_with_emptry_string(comment_factory):
comment = comment_factory()
with pytest.raises(comments.EmptyCommentTextError):
comments.update_comment_text(comment, None)
def test_update_comment_text(comment_factory):
comment = comment_factory()
comments.update_comment_text(comment, "text")
assert comment.text == "text"
|
