updated search_command.py file

Author: akaila-splunk

splunklib/searchcommands/search_command.py

@@ -178,7 +178,7 @@ def add_field(self, current_record, field_name, field_value):
         current_record[field_name] = field_value
 
     def gen_record(self, **record):
-        self._record_writer.custom_fields |= record.keys()
+        self._record_writer.custom_fields |= set(record.keys())
         return record
 
     record = Option(doc='''

tests/searchcommands/test_generator_command.py

@@ -82,5 +82,5 @@ def generate(self):
     fieldnames_actual = set()
     for chunk in ds:
         for row in chunk.data:
-            fieldnames_actual |= row.keys()
+            fieldnames_actual |= set(row.keys())
     assert fieldnames_expected.issubset(fieldnames_actual)