56d1346789
It turns out this has to do with some additions to the fixture adding of media entries. Since the new fake_upload field is True by default, adding multiple entries that have media_files at once meant that multiple additions of FileKeynames would be added at once if they were not saved at the same time. Tricky! Thankfully, Elrond helped us figure this one out. Thanks, Elrond :) And thanks also to Aapo Rantalainen for sponsoring this commit!
94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
# GNU MediaGoblin -- federated, autonomous media hosting
|
|
# Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from mediagoblin.db.base import Session
|
|
from mediagoblin.db.models import User, MediaEntry, MediaComment
|
|
from mediagoblin.tests.tools import fixture_add_user, fixture_media_entry
|
|
|
|
|
|
def test_404_for_non_existent(test_app):
|
|
res = test_app.get('/does-not-exist/', expect_errors=True)
|
|
assert res.status_int == 404
|
|
|
|
|
|
def test_user_deletes_other_comments(test_app):
|
|
user_a = fixture_add_user(u"chris_a")
|
|
user_b = fixture_add_user(u"chris_b")
|
|
|
|
media_a = fixture_media_entry(uploader=user_a.id, save=False,
|
|
expunge=False, fake_upload=False)
|
|
media_b = fixture_media_entry(uploader=user_b.id, save=False,
|
|
expunge=False, fake_upload=False)
|
|
Session.add(media_a)
|
|
Session.add(media_b)
|
|
Session.flush()
|
|
|
|
# Create all 4 possible comments:
|
|
for u_id in (user_a.id, user_b.id):
|
|
for m_id in (media_a.id, media_b.id):
|
|
cmt = MediaComment()
|
|
cmt.media_entry = m_id
|
|
cmt.author = u_id
|
|
cmt.content = u"Some Comment"
|
|
Session.add(cmt)
|
|
|
|
Session.flush()
|
|
|
|
usr_cnt1 = User.query.count()
|
|
med_cnt1 = MediaEntry.query.count()
|
|
cmt_cnt1 = MediaComment.query.count()
|
|
|
|
User.query.get(user_a.id).delete(commit=False)
|
|
|
|
usr_cnt2 = User.query.count()
|
|
med_cnt2 = MediaEntry.query.count()
|
|
cmt_cnt2 = MediaComment.query.count()
|
|
|
|
# One user deleted
|
|
assert usr_cnt2 == usr_cnt1 - 1
|
|
# One media gone
|
|
assert med_cnt2 == med_cnt1 - 1
|
|
# Three of four comments gone.
|
|
assert cmt_cnt2 == cmt_cnt1 - 3
|
|
|
|
User.query.get(user_b.id).delete()
|
|
|
|
usr_cnt2 = User.query.count()
|
|
med_cnt2 = MediaEntry.query.count()
|
|
cmt_cnt2 = MediaComment.query.count()
|
|
|
|
# All users gone
|
|
assert usr_cnt2 == usr_cnt1 - 2
|
|
# All media gone
|
|
assert med_cnt2 == med_cnt1 - 2
|
|
# All comments gone
|
|
assert cmt_cnt2 == cmt_cnt1 - 4
|
|
|
|
|
|
def test_media_deletes_broken_attachment(test_app):
|
|
user_a = fixture_add_user(u"chris_a")
|
|
|
|
media = fixture_media_entry(uploader=user_a.id, save=False, expunge=False)
|
|
media.attachment_files.append(dict(
|
|
name=u"some name",
|
|
filepath=[u"does", u"not", u"exist"],
|
|
))
|
|
Session.add(media)
|
|
Session.flush()
|
|
|
|
MediaEntry.query.get(media.id).delete()
|
|
User.query.get(user_a.id).delete()
|