From 751e5d415e7c749344ffa81dea634b05c19fd6e7 Mon Sep 17 00:00:00 2001
From: Dilawar Singh <dilawar@subcom.tech>
Date: Fri, 7 Oct 2022 15:03:15 +0530
Subject: [PATCH] chore: fix tests
stream can return `None`
---
tests/test_sanity.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/tests/test_sanity.py b/tests/test_sanity.py
index 0caa677..aab985e 100644
--- a/tests/test_sanity.py
+++ b/tests/test_sanity.py
@@ -10,9 +10,11 @@ def test_sanity():
def test_run_simple():
for line in bitia.__main__.run_pipeline("ls -ltr /"):
- print(111, line.decode().rstrip())
+ if line:
+ print(111, line.decode().rstrip())
for line in bitia.__main__.run_pipeline("ls -ltr /", rerun=True):
- print(222, line.decode().rstrip())
+ if line:
+ print(222, line.decode().rstrip())
if __name__ == "__main__":
--
GitLab