Test for allow_empty_input in generating commands

Author: vmalaviya-splunk

tests/searchcommands/test_generator_command.py

@@ -4,6 +4,7 @@
 from . import chunked_data_stream as chunky
 
 from splunklib.searchcommands import Configuration, GeneratingCommand
+from unittest import TestCase
 
 
 def test_simple_generator():
@@ -41,4 +42,21 @@ def generate(self):
     assert finished_seen
 
 
+def test_allow_empty_input_for_generating_command():
+    """
+    Passing allow_empty_input for generating command will cause an error
+    """
+    @Configuration()
+    class GeneratorTest(GeneratingCommand):
+        def generate(self):
+            for num in range(1, 3):
+                yield {"_index": num}
+    generator = GeneratorTest()
+    in_stream = io.BytesIO()
+    out_stream = io.BytesIO()
+
+    try:
+        generator.process([], in_stream, out_stream, allow_empty_input=False)
+    except ValueError as error:
+        assert str(error) == "allow_empty_input cannot be False for Generating Commands"