🐛 [-bug] Addressed issues with test stream capabilities

davidbrownell · May 30, 2024 · 1395f57 · 1395f57
1 parent ced9263
commit 1395f57
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/dbrownell_Common/TestHelpers/StreamTestHelpers.py b/src/dbrownell_Common/TestHelpers/StreamTestHelpers.py
@@ -48,7 +48,7 @@ def Replace(
     else:
         replace_func = lambda _: "<scrubbed duration>"
 
-    content = re.sub(
+    return re.sub(
         r"""(?#
             Hours                           )(?P<hours>\d+)(?#
             sep                             )\:(?#
@@ -60,11 +60,6 @@ def Replace(
         content,
     )
 
-    # Remove any trailing whitespace
-    content = "\n".join(line.rstrip() for line in content.split("\n"))
-
-    return content
-
 
 # ----------------------------------------------------------------------
 def GenerateDoneManagerAndContent(
@@ -124,6 +119,9 @@ def GenerateDoneManagerAndContent(
         keep_seconds=keep_duration_seconds,
     )
 
+    # Remove any trailing whitespace
+    content = "\n".join(line.rstrip() for line in content.split("\n"))
+
     assert expected_result is None or final_result == expected_result, (
         expected_result,
         final_result,
@@ -144,6 +142,8 @@ def InitializeStreamCapabilities(
     ):
         Capabilities(
             columns=120,
+            is_headless=True,
+            is_interactive=False,
             supports_colors=True,
             stream=stream,
             no_column_warning=True,