test_unbounded_response: to_unicode → custom six code

This commit is contained in:
Adrián Chaves 2022-03-01 15:50:19 +01:00
parent efd72b08f9
commit d2589c74c5
1 changed files with 6 additions and 1 deletions

View File

@ -1,6 +1,7 @@
import json
import logging
import six
from testfixtures import LogCapture
from twisted.internet import defer
from twisted.trial.unittest import TestCase
@ -161,7 +162,11 @@ with multiples lines
crawler = self.runner.create_crawler(SimpleSpider)
with LogCapture() as l:
yield crawler.crawl(self.mockserver.url("/raw?{0}".format(query)), mockserver=self.mockserver)
self.assertEqual(to_unicode(l).count("Got response 200"), 1)
if six.PY2:
message = unicode(l)
else:
message = str(l)
self.assertEqual(message.count("Got response 200"), 1)
@defer.inlineCallbacks
def test_retry_conn_lost(self):