move test_partial_send

mambocab · mambocab · commit 3230a7de45f4 · 2018-01-19T15:20:45.000-05:00
diff --git a/tests/unit/io/test_asyncorereactor.py b/tests/unit/io/test_asyncorereactor.py
@@ -58,23 +58,6 @@ def setUp(self):
         if is_monkey_patched():
             raise unittest.SkipTest("Can't test asyncore with monkey patching")
 
-    def test_partial_send(self):
-        c = self.make_connection()
-
-        # only write the first four bytes of the OptionsMessage
-        write_size = 4
-        c.socket.send.side_effect = None
-        c.socket.send.return_value = write_size
-        c.handle_write()
-
-        msg_size = 9  # v3+ frame header
-        expected_writes = int(math.ceil(float(msg_size) / write_size))
-        size_mod = msg_size % write_size
-        last_write_size = size_mod if size_mod else write_size
-        self.assertFalse(c.is_defunct)
-        self.assertEqual(expected_writes, c.socket.send.call_count)
-        self.assertEqual(last_write_size, len(c.socket.send.call_args[0][0]))
-
     def test_socket_error_on_read(self):
         c = self.make_connection()
 
diff --git a/tests/unit/io/test_libevreactor.py b/tests/unit/io/test_libevreactor.py
@@ -62,23 +62,6 @@ def setUp(self):
         for p in patchers:
             p.start()
 
-    def test_partial_send(self):
-        c = self.make_connection()
-
-        # only write the first four bytes of the OptionsMessage
-        write_size = 4
-        c._socket.send.side_effect = None
-        c._socket.send.return_value = write_size
-        c.handle_write(None, 0)
-
-        msg_size = 9  # v3+ frame header
-        expected_writes = int(math.ceil(float(msg_size) / write_size))
-        size_mod = msg_size % write_size
-        last_write_size = size_mod if size_mod else write_size
-        self.assertFalse(c.is_defunct)
-        self.assertEqual(expected_writes, c._socket.send.call_count)
-        self.assertEqual(last_write_size, len(c._socket.send.call_args[0][0]))
-
     def test_socket_error_on_read(self):
         c = self.make_connection()
 
diff --git a/tests/unit/io/utils.py b/tests/unit/io/utils.py
@@ -23,6 +23,7 @@
 from mock import Mock
 
 import errno
+import math
 import os
 from socket import error as socket_error
 
@@ -321,3 +322,21 @@ def test_blocking_on_write(self):
         c.handle_write(*self.null_handle_function_args)
         self.assertFalse(c.is_defunct)
         self.assertTrue(self.get_socket(c).send.call_args is not None)
+
+    def test_partial_send(self):
+        c = self.make_connection()
+
+        # only write the first four bytes of the OptionsMessage
+        write_size = 4
+        self.get_socket(c).send.side_effect = None
+        self.get_socket(c).send.return_value = write_size
+        c.handle_write(*self.null_handle_function_args)
+
+        msg_size = 9  # v3+ frame header
+        expected_writes = int(math.ceil(float(msg_size) / write_size))
+        size_mod = msg_size % write_size
+        last_write_size = size_mod if size_mod else write_size
+        self.assertFalse(c.is_defunct)
+        self.assertEqual(expected_writes, self.get_socket(c).send.call_count)
+        self.assertEqual(last_write_size,
+                         len(self.get_socket(c).send.call_args[0][0]))
