diff options
author | Jessica Tallon <tsyesika@tsyesika.se> | 2015-10-20 13:02:15 +0000 |
---|---|---|
committer | Jessica Tallon <tsyesika@tsyesika.se> | 2015-10-20 13:02:15 +0000 |
commit | 03bb1b7907d2d0fa522bf4705e4891494839da09 (patch) | |
tree | f60658eb937886dfd1cc349cab605c13a02942e5 /mediagoblin/tests/test_submission.py | |
parent | fd703bb4d0665958d853b89f6069eefd8a8c8113 (diff) | |
parent | 64a456a4e50b03e4fa2b33ceb208e88d2e02fce7 (diff) | |
download | mediagoblin-03bb1b7907d2d0fa522bf4705e4891494839da09.tar.lz mediagoblin-03bb1b7907d2d0fa522bf4705e4891494839da09.tar.xz mediagoblin-03bb1b7907d2d0fa522bf4705e4891494839da09.zip |
Merge branch 'comments'
Diffstat (limited to 'mediagoblin/tests/test_submission.py')
-rw-r--r-- | mediagoblin/tests/test_submission.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/mediagoblin/tests/test_submission.py b/mediagoblin/tests/test_submission.py index c0c3d3cf..f9031d37 100644 --- a/mediagoblin/tests/test_submission.py +++ b/mediagoblin/tests/test_submission.py @@ -99,8 +99,14 @@ class TestSubmission: return {'upload_files': [('file', filename)]} def check_comments(self, request, media_id, count): - comments = request.db.MediaComment.query.filter_by(media_entry=media_id) - assert count == len(list(comments)) + gmr = request.db.GenericModelReference.query.filter_by( + obj_pk=media_id, + model_type=request.db.MediaEntry.__tablename__ + ).first() + if gmr is None and count <= 0: + return # Yerp it's fine. + comments = request.db.Comment.query.filter_by(target_id=gmr.id) + assert count == comments.count() def test_missing_fields(self): # Test blank form |