From a0a7f87f625ca29ccddc123eb6540aa15a779eb7 Mon Sep 17 00:00:00 2001 From: Brett Smith Date: Sat, 17 Mar 2012 17:53:00 -0400 Subject: When media is deleted, delete associated comments too. The actual code is just a simple for loop; there might be a better implementation but this is a fine start. I also extended test_delete to check this too. --- mediagoblin/tests/test_submission.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'mediagoblin/tests/test_submission.py') 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 # ----------------------------------------------------- -- cgit v1.2.3