diff options
| -rw-r--r-- | mediagoblin/tests/test_submission.py | 15 | ||||
| -rw-r--r-- | mediagoblin/user_pages/views.py | 4 |
2 files changed, 19 insertions, 0 deletions
diff --git a/mediagoblin/tests/test_submission.py b/mediagoblin/tests/test_submission.py index 2f11bdfb..1f56779e 100644 --- a/mediagoblin/tests/test_submission.py +++ b/mediagoblin/tests/test_submission.py @@ -180,6 +180,18 @@ class TestSubmission: # Does media entry exist? assert_true(media) + # Add a comment, so we can test for its deletion later. + get_comments = lambda: list( + request.db.MediaComment.find({'media_entry': media._id})) + assert_false(get_comments()) + response = self.test_app.post( + request.urlgen('mediagoblin.user_pages.media_post_comment', + user=self.test_user.username, + media=media._id), + {'comment_content': 'i love this test'}) + response.follow() + assert_true(get_comments()) + # Do not confirm deletion # --------------------------------------------------- response = self.test_app.post( @@ -219,6 +231,9 @@ class TestSubmission: request.db.MediaEntry.find( {'_id': media._id}).count()) + # How about the comment? + assert_false(get_comments()) + def test_malicious_uploads(self): # Test non-suppoerted file with non-supported extension # ----------------------------------------------------- diff --git a/mediagoblin/user_pages/views.py b/mediagoblin/user_pages/views.py index 530dea64..6eff684c 100644 --- a/mediagoblin/user_pages/views.py +++ b/mediagoblin/user_pages/views.py @@ -173,6 +173,10 @@ def media_confirm_delete(request, media): if form.confirm.data is True: username = media.get_uploader.username + # Delete all the associated comments + for comment in media.get_comments(): + comment.delete() + # Delete all files on the public storage delete_media_files(media) |