From 365684f13738b28720352b099ab0a018a723405b Mon Sep 17 00:00:00 2001 From: Timo Stollenwerk Date: Mon, 18 Jan 2010 12:20:45 +0000 Subject: [PATCH] test stop iteration for get_replies method. svn path=/plone.app.discussion/trunk/; revision=33218 --- plone/app/discussion/tests/test_comments_viewlet.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/plone/app/discussion/tests/test_comments_viewlet.py b/plone/app/discussion/tests/test_comments_viewlet.py index 4a4ea1a..5d0b0b3 100644 --- a/plone/app/discussion/tests/test_comments_viewlet.py +++ b/plone/app/discussion/tests/test_comments_viewlet.py @@ -78,7 +78,12 @@ class TestCommentsViewlet(PloneTestCase): conversation = IConversation(self.portal.doc1) conversation.addComment(comment) conversation.addComment(comment) - self.assertEquals(sum(1 for w in self.viewlet.get_replies()), 2) + replies = self.viewlet.get_replies() + self.assertEquals(sum(1 for w in replies), 2) + replies = self.viewlet.get_replies() + replies.next() + replies.next() + self.assertRaises(StopIteration, replies.next) def test_get_replies_with_workflow_actions(self): self.failIf(self.viewlet.get_replies(workflow_actions=True))