snuffster/szurubooru
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

server/tests: update func.posts tests

Browse source
This commit is contained in:
rr- 2016-08-14 11:48:19 +02:00
parent 264f9ee70b
commit d102c9bdba
2 changed files with 49 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

77
server/szurubooru/tests/func/test_posts.py
View file

@ -1,7 +1,7 @@
import os
import datetime
import unittest.mock
import pytest
from datetime import datetime
from szurubooru import db
from szurubooru.func import posts, users, comments, snapshots, tags, images
@ -61,7 +61,7 @@ def test_serialize_note():
'text': '...'
}
def test_serialize_empty_post():
def test_serialize_post_when_empty():
assert posts.serialize_post(None, None) is None
def test_serialize_post(
@ -80,8 +80,8 @@ def test_serialize_post(
auth_user = user_factory(name='auth user')
post = db.Post()
post.post_id = 1
post.creation_time = datetime.datetime(1997, 1, 1)
post.last_edit_time = datetime.datetime(1998, 1, 1)
post.creation_time = datetime(1997, 1, 1)
post.last_edit_time = datetime(1998, 1, 1)
post.tags = [
tag_factory(names=['tag1', 'tag2']),
tag_factory(names=['tag3'])
@ -105,26 +105,26 @@ def test_serialize_post(
db.PostFavorite(
post=post,
user=user_factory(name='fav1'),
time=datetime.datetime(1800, 1, 1)),
time=datetime(1800, 1, 1)),
db.PostFeature(
post=post,
user=user_factory(),
time=datetime.datetime(1999, 1, 1)),
time=datetime(1999, 1, 1)),
db.PostScore(
post=post,
user=auth_user,
score=-1,
time=datetime.datetime(1800, 1, 1)),
time=datetime(1800, 1, 1)),
db.PostScore(
post=post,
user=user_factory(),
score=1,
time=datetime.datetime(1800, 1, 1)),
time=datetime(1800, 1, 1)),
db.PostScore(
post=post,
user=user_factory(),
score=1,
time=datetime.datetime(1800, 1, 1))])
time=datetime(1800, 1, 1))])
db.session.flush()
result = posts.serialize_post(post, auth_user)
@ -133,8 +133,8 @@ def test_serialize_post(
assert result == {
'id': 1,
'version': 1,
'creationTime': datetime.datetime(1997, 1, 1),
'lastEditTime': datetime.datetime(1998, 1, 1),
'creationTime': datetime(1997, 1, 1),
'lastEditTime': datetime(1998, 1, 1),
'safety': 'safe',
'source': '4gag',
'type': 'image',
@ -158,7 +158,7 @@ def test_serialize_post(
'noteCount': 0,
'featureCount': 1,
'relationCount': 0,
'lastFeatureTime': datetime.datetime(1999, 1, 1),
'lastFeatureTime': datetime(1999, 1, 1),
'favoritedBy': ['fav1'],
'hasCustomThumbnail': True,
'mimeType': 'image/jpeg',
@ -166,6 +166,18 @@ def test_serialize_post(
'comments': ['commenter1', 'commenter2'],
}
def test_serialize_micro_post(post_factory, user_factory):
with unittest.mock.patch('szurubooru.func.posts.get_post_thumbnail_url'):
posts.get_post_thumbnail_url.return_value = 'https://example.com/thumb.png'
auth_user = user_factory()
post = post_factory()
db.session.add(post)
db.session.flush()
assert posts.serialize_micro_post(post, auth_user) == {
'id': post.post_id,
'thumbnailUrl': 'https://example.com/thumb.png',
}
def test_get_post_count(post_factory):
previous_count = posts.get_post_count()
db.session.add_all([post_factory(), post_factory()])
@ -179,6 +191,8 @@ def test_try_get_post_by_id(post_factory):
db.session.flush()
assert posts.try_get_post_by_id(post.post_id) == post
assert posts.try_get_post_by_id(post.post_id + 1) is None
with pytest.raises(posts.InvalidPostIdError):
posts.get_post_by_id('-')
def test_get_post_by_id(post_factory):
post = post_factory()
@ -187,14 +201,16 @@ def test_get_post_by_id(post_factory):
assert posts.get_post_by_id(post.post_id) == post
with pytest.raises(posts.PostNotFoundError):
posts.get_post_by_id(post.post_id + 1)
with pytest.raises(posts.InvalidPostIdError):
posts.get_post_by_id('-')
def test_create_post(user_factory, fake_datetime):
with unittest.mock.patch('szurubooru.func.posts.update_post_content'), \
unittest.mock.patch('szurubooru.func.posts.update_post_tags'), \
fake_datetime('1997-01-01'):
unittest.mock.patch('szurubooru.func.posts.update_post_tags'), \
fake_datetime('1997-01-01'):
auth_user = user_factory()
post, new_tags = posts.create_post('content', ['tag'], auth_user)
assert post.creation_time == datetime.datetime(1997, 1, 1)
assert post.creation_time == datetime(1997, 1, 1)
assert post.last_edit_time is None
posts.update_post_tags.assert_called_once_with(post, ['tag'])
posts.update_post_content.assert_called_once_with(post, 'content')
@ -209,7 +225,7 @@ def test_update_post_safety(input_safety, expected_safety):
posts.update_post_safety(post, input_safety)
assert post.safety == expected_safety
def test_update_post_invalid_safety():
def test_update_post_safety_with_invalid_string():
post = db.Post()
with pytest.raises(posts.InvalidPostSafetyError):
posts.update_post_safety(post, 'bad')
@ -219,7 +235,7 @@ def test_update_post_source():
posts.update_post_source(post, 'x')
assert post.source == 'x'
def test_update_post_invalid_source():
def test_update_post_source_with_too_long_string():
post = db.Post()
with pytest.raises(posts.InvalidPostSourceError):
posts.update_post_source(post, 'x' * 1000)
@ -277,7 +293,7 @@ def test_update_post_content_to_existing_content(
with pytest.raises(posts.PostAlreadyUploadedError):
posts.update_post_content(another_post, read_asset('png.png'))
def test_update_post_content_broken_content(
def test_update_post_content_with_broken_content(
tmpdir, config_injector, post_factory, read_asset):
# the rationale behind this behavior is to salvage user upload even if the
# server software thinks it's broken. chances are the server is wrong,
@ -298,7 +314,7 @@ def test_update_post_content_broken_content(
assert post.canvas_height is None
@pytest.mark.parametrize('input_content', [None, b'not a media file'])
def test_update_post_invalid_content(input_content):
def test_update_post_content_with_invalid_content(input_content):
post = db.Post()
with pytest.raises(posts.InvalidPostContentError):
posts.update_post_content(post, input_content)
@ -340,7 +356,7 @@ def test_update_post_thumbnail_to_default(
assert not os.path.exists(str(tmpdir) + '/data/posts/custom-thumbnails/1.dat')
assert os.path.exists(str(tmpdir) + '/data/generated-thumbnails/1.jpg')
def test_update_post_thumbnail_broken_thumbnail(
def test_update_post_thumbnail_with_broken_thumbnail(
tmpdir, config_injector, read_asset, post_factory):
config_injector({
'data_dir': str(tmpdir.mkdir('data')),
@ -363,7 +379,7 @@ def test_update_post_thumbnail_broken_thumbnail(
assert image.width == 1
assert image.height == 1
def test_update_post_content_leaves_custom_thumbnail(
def test_update_post_content_leaving_custom_thumbnail(
tmpdir, config_injector, read_asset, post_factory):
config_injector({
'data_dir': str(tmpdir.mkdir('data')),
@ -403,7 +419,7 @@ def test_update_post_relations(post_factory):
assert post.relations[0].post_id == relation1.post_id
assert post.relations[1].post_id == relation2.post_id
def test_relation_bidirectionality(post_factory):
def test_update_post_relations_bidirectionality(post_factory):
relation1 = post_factory()
relation2 = post_factory()
db.session.add_all([relation1, relation2])
@ -414,7 +430,7 @@ def test_relation_bidirectionality(post_factory):
assert len(post.relations) == 1
assert post.relations[0].post_id == relation2.post_id
def test_update_post_non_existing_relations():
def test_update_post_relations_with_nonexisting_posts():
post = db.Post()
with pytest.raises(posts.InvalidPostRelationError):
posts.update_post_relations(post, [100])
@ -452,7 +468,7 @@ def test_update_post_notes():
[{'polygon': [[0, 0], [0, 0], [0, 1]], 'text': None}],
[{'polygon': [[0, 0], [0, 0], [0, 1]]}],
])
def test_update_post_invalid_notes(input):
def test_update_post_notes_with_invalid_content(input):
post = db.Post()
with pytest.raises(posts.InvalidPostNoteError):
posts.update_post_notes(post, input)
@ -462,18 +478,23 @@ def test_update_post_flags():
posts.update_post_flags(post, ['loop'])
assert post.flags == ['loop']
def test_update_post_invalid_flags():
def test_update_post_flags_with_invalid_content():
post = db.Post()
with pytest.raises(posts.InvalidPostFlagError):
posts.update_post_flags(post, ['invalid'])
def test_featuring_post(post_factory, user_factory):
def test_feature_post(post_factory, user_factory):
post = post_factory()
user = user_factory()
previous_featured_post = posts.try_get_featured_post()
posts.feature_post(post, user)
new_featured_post = posts.try_get_featured_post()
assert previous_featured_post is None
assert new_featured_post == post
def test_delete(post_factory):
post = post_factory()
db.session.add(post)
assert posts.get_post_count() == 1
posts.delete(post)
assert posts.get_post_count() == 0

2
server/szurubooru/tests/search/configs/test_post_search_config.py
View file

@ -22,8 +22,6 @@ def score_factory(user_factory):
@pytest.fixture
def note_factory():
def factory(post=None):
if post:
return db.PostNote(polygon='...', text='...', post=post)
return db.PostNote(polygon='...', text='...')
return factory