fix tests broken by taking lint advice

splunk · Dec 3, 2020 · 139d907 · 139d907
1 parent efaa7e1
commit 139d907
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tests/searchcommands/chunked_data_stream.py b/tests/searchcommands/chunked_data_stream.py
@@ -42,11 +42,11 @@ def __init__(self, stream):
     def read_chunk(self):
         header = self.stream.readline()
 
-        while header > 0 and header.strip() == b'':
+        while len(header) > 0 and header.strip() == b'':
             header = self.stream.readline()  # Skip empty lines
-
-        if not header == 0:
+        if len(header) == 0:
             raise EOFError
+
         version, meta, data = header.rstrip().split(b',')
         metabytes = self.stream.read(int(meta))
         databytes = self.stream.read(int(data))

diff --git a/tests/searchcommands/test_generator_command.py b/tests/searchcommands/test_generator_command.py
@@ -34,6 +34,9 @@ def generate(self):
         finished_seen = chunk.meta.get("finished", False)
         for row in chunk.data:
             seen.add(row["event_index"])
+    print(out_stream.getvalue())
+    print(expected)
+    print(seen)
     assert expected.issubset(seen)
     assert finished_seen