Fix test_asyncore after merge. It needs to use bytes.

diff --git a/Lib/asyncore.py b/Lib/asyncore.py
index fc110c9ef29c29651357eb1f07be6d2292cd03cf..32f579035fa4af554d65a26132a1a6c2b45c1e45 100644 (file)
--- a/Lib/asyncore.py
+++ b/Lib/asyncore.py
@@ -344,14 +344,14 @@ class dispatcher:
                 # a closed connection is indicated by signaling
                 # a read condition, and having recv() return 0.
                 self.handle_close()
-                return ''
+                return b''
             else:
                 return data
         except socket.error as why:
             # winsock sometimes throws ENOTCONN
             if why[0] in [ECONNRESET, ENOTCONN, ESHUTDOWN]:
                 self.handle_close()
-                return ''
+                return b''
             else:
                 raise
 
@@ -447,7 +447,7 @@ class dispatcher_with_send(dispatcher):
 
     def __init__(self, sock=None, map=None):
         dispatcher.__init__(self, sock, map)
-        self.out_buffer = ''
+        self.out_buffer = b''
 
     def initiate_send(self):
         num_sent = 0
@@ -461,6 +461,7 @@ class dispatcher_with_send(dispatcher):
         return (not self.connected) or len(self.out_buffer)
 
     def send(self, data):
+        assert isinstance(data, bytes)
         if self.debug:
             self.log_info('sending %s' % repr(data))
         self.out_buffer = self.out_buffer + data

diff --git a/Lib/test/test_asyncore.py b/Lib/test/test_asyncore.py
index 7602b9fca66e69d7515de5dacb6e56fa197527f7..fca5b297db597251cfc3d4f5ff0e777566ff9052 100644 (file)
--- a/Lib/test/test_asyncore.py
+++ b/Lib/test/test_asyncore.py
@@ -9,7 +9,7 @@ import time
 
 from test import test_support
 from test.test_support import TESTFN, run_unittest, unlink
-from StringIO import StringIO
+from io import StringIO, BytesIO
 
 HOST = "127.0.0.1"
 PORT = 54329
@@ -66,9 +66,10 @@ def capture_server(evt, buf):
         n = 200
         while n > 0:
             data = conn.recv(10)
+            assert isinstance(data, bytes)
             # keep everything except for the newline terminator
-            buf.write(data.replace('\n', ''))
-            if '\n' in data:
+            buf.write(data.replace(b'\n', b''))
+            if b'\n' in data:
                 break
             n -= 1
             time.sleep(0.01)
@@ -338,16 +339,16 @@ class DispatcherWithSendTests(unittest.TestCase):
 
     def test_send(self):
         self.evt = threading.Event()
-        cap = StringIO()
-        threading.Thread(target=capture_server, args=(self.evt,cap)).start()
+        cap = BytesIO()
+        threading.Thread(target=capture_server, args=(self.evt, cap)).start()
         time.sleep(1) # Give server time to initialize
 
-        data = "Suppose there isn't a 16-ton weight?"*5
+        data = b"Suppose there isn't a 16-ton weight?"*5
         d = dispatcherwithsend_noread()
         d.create_socket(socket.AF_INET, socket.SOCK_STREAM)
         d.connect((HOST, PORT))
         d.send(data)
-        d.send('\n')
+        d.send(b'\n')
 
         n = 1000
         while d.out_buffer and n > 0:
@@ -366,7 +367,7 @@ if hasattr(asyncore, 'file_wrapper'):
     class FileWrapperTest(unittest.TestCase):
         def setUp(self):
             self.d = "It's not dead, it's sleeping!"
-            file(TESTFN, 'w').write(self.d)
+            open(TESTFN, 'w').write(self.d)
 
         def tearDown(self):
             unlink(TESTFN)
@@ -377,8 +378,8 @@ if hasattr(asyncore, 'file_wrapper'):
 
             self.assertEqual(w.fd, fd)
             self.assertEqual(w.fileno(), fd)
-            self.assertEqual(w.recv(13), "It's not dead")
-            self.assertEqual(w.read(6), ", it's")
+            self.assertEqual(w.recv(13), b"It's not dead")
+            self.assertEqual(w.read(6), b", it's")
             w.close()
             self.assertRaises(OSError, w.read, 1)
 
@@ -391,7 +392,7 @@ if hasattr(asyncore, 'file_wrapper'):
             w.write(d1)
             w.send(d2)
             w.close()
-            self.assertEqual(file(TESTFN).read(), self.d + d1 + d2)
+            self.assertEqual(open(TESTFN).read(), self.d + d1 + d2)
 
 
 def test_main():