Small changes to #598

1 files changed, 5 insertions, 12 deletions

diff --git a/t/integration/transport.py b/t/integration/transport.py
index ebb3accf..5a24f699 100644
--- a/t/integration/transport.py
+++ b/t/integration/transport.py
@@ -13,22 +13,17 @@ from case.skip import SkipTest
 
 from kombu import Connection
 from kombu import Exchange, Queue
-from kombu.five import range
+from kombu.five import buffer_t, range
+from kombu.utils.encoding import str_to_bytes
 
 if sys.version_info >= (2, 5):
     from hashlib import sha256 as _digest
 else:
     from sha import new as _digest  # noqa
 
-if not hasattr(string, 'letters'):
-    string.letters = string.ascii_letters
-
 
 def _nobuf(x):
-    if 'buffer' in locals():
-        return [str(i) if isinstance(i, buffer) else i for i in x]
-    else:
-        return [str(i) for i in x]
+    return [str(i) if isinstance(i, buffer_t) else i for i in x]
 
 
 def consumeN(conn, consumer, n=1, timeout=30):
@@ -167,13 +162,11 @@ class TransportCase(unittest.TestCase):
                 purged += self.purge_consumer(consumer)
 
     def _digest(self, data):
-        if isinstance(data, type(u'')):
-            data = data.encode()
-        return _digest(data).hexdigest()
+        return _digest(str_to_bytes(data)).hexdigest()
 
     def test_produce__consume_large_messages(
             self, bytes=1048576, n=10,
-            charset=string.punctuation + string.letters + string.digits):
+            charset=string.punctuation + string.ascii_letters + string.digits):
         if not self.verify_alive():
             return
         bytes = min(x for x in [bytes, self.message_size_limit] if x)