aboutsummaryrefslogtreecommitdiffstats
path: root/mediagoblin/tests/test_submission.py
diff options
context:
space:
mode:
authorJessica Tallon <tsyesika@tsyesika.se>2015-10-20 13:02:15 +0000
committerJessica Tallon <tsyesika@tsyesika.se>2015-10-20 13:02:15 +0000
commit03bb1b7907d2d0fa522bf4705e4891494839da09 (patch)
treef60658eb937886dfd1cc349cab605c13a02942e5 /mediagoblin/tests/test_submission.py
parentfd703bb4d0665958d853b89f6069eefd8a8c8113 (diff)
parent64a456a4e50b03e4fa2b33ceb208e88d2e02fce7 (diff)
downloadmediagoblin-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.py10
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