split Test_IStream

comtypes/test/test_istream.py

@@ -8,27 +8,41 @@
 from comtypes.gen.PortableDeviceApiLib import IStream
 
 
-class Test_IStream(ut.TestCase):
-    def test_istream(self):
-        # Create an IStream
-        stream: IStream = POINTER(IStream)()  # type: ignore
-        comtypes._ole32.CreateStreamOnHGlobal(None, c_bool(True), byref(stream))
+def _create_stream() -> IStream:
+    # Create an IStream
+    stream = POINTER(IStream)()  # type: ignore
+    comtypes._ole32.CreateStreamOnHGlobal(None, c_bool(True), byref(stream))
+    return stream  # type: ignore
 
+
+class Test_RemoteWrite(ut.TestCase):
+    def test_RemoteWrite(self):
+        stream = _create_stream()
         test_data = "Some data".encode("utf-8")
         pv = (c_ubyte * len(test_data)).from_buffer(bytearray(test_data))
 
         written = stream.RemoteWrite(pv, len(test_data))
+
+        # Verification
         self.assertEqual(written, len(test_data))
-        # make sure the data actually gets written before trying to read back
-        stream.Commit(0)
 
+
+class Test_RemoteRead(ut.TestCase):
+    def test_RemoteRead(self):
+        stream = _create_stream()
+        test_data = "Some data".encode("utf-8")
+        pv = (c_ubyte * len(test_data)).from_buffer(bytearray(test_data))
+        stream.RemoteWrite(pv, len(test_data))
+
+        # Make sure the data actually gets written before trying to read back
+        stream.Commit(0)
         # Move the stream back to the beginning
         STREAM_SEEK_SET = 0
         stream.RemoteSeek(0, STREAM_SEEK_SET)
 
-        read_buffer_size = 1024
+        buffer_size = 1024
 
-        read_buffer, data_read = stream.RemoteRead(read_buffer_size)
+        read_buffer, data_read = stream.RemoteRead(buffer_size)
 
         # Verification
         self.assertEqual(data_read, len(test_data))