splunk
diff --git a/‎splunklib/searchcommands/search_command.py
Lines changed: 2 additions & 1 deletion b/‎splunklib/searchcommands/search_command.py
Lines changed: 2 additions & 1 deletion
diff --git a/‎tests/searchcommands/chunked_data_stream.py b/‎tests/searchcommands/chunked_data_stream.py
diff --git a/‎tests/searchcommands/test_generator_command.py b/‎tests/searchcommands/test_generator_command.py
diff --git a/‎tests/searchcommands/test_reporting_command.py b/‎tests/searchcommands/test_reporting_command.py
diff --git a/‎tests/searchcommands/test_streaming_command.py
Lines changed: 19 additions & 0 deletions b/‎tests/searchcommands/test_streaming_command.py
Lines changed: 19 additions & 0 deletions
@@ -851,7 +851,8 @@ def _execute(self, ifile, process):
 
     @staticmethod
     def _as_binary_stream(ifile):
-        if six.PY2:
+        naught = ifile.read(0)
+        if isinstance(naught, bytes):
             return ifile
 
         try:
 
@@ -0,0 +1,19 @@
+import io
+from splunklib.searchcommands import StreamingCommand, Configuration
+
+
+def test_simple_streaming_command():
+    @Configuration()
+    class TestStreamingCommand(StreamingCommand):
+        count = 0
+
+        def stream(self, records):
+            for record in records:
+                record["idx"] = self.count
+                self.count += 1
+                yield record
+
+    cmd = TestStreamingCommand()
+    ifile = io.BytesIO()
+    ofile = io.BytesIO()
+    cmd._process_protocol_v2([], ifile, ofile)