From 298b386d04fd1c91b69f8788828ef5c36d62e12d Mon Sep 17 00:00:00 2001
From: Myles Scolnick <myles@marimo.io>
Date: Thu, 12 Dec 2024 19:02:18 -0500
Subject: [PATCH] tests: revert cli debug (#3153)

---
 tests/_cli/test_cli.py | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/tests/_cli/test_cli.py b/tests/_cli/test_cli.py
index b7304b60633..ed7a0880fb8 100644
--- a/tests/_cli/test_cli.py
+++ b/tests/_cli/test_cli.py
@@ -199,7 +199,7 @@ def test_cli_edit_none() -> None:
     # smoke test: makes sure CLI starts and has basic things we expect
     # helpful for catching issues related to
     port = _get_port()
-    with subprocess.Popen(
+    p = subprocess.Popen(
         [
             "marimo",
             "edit",
@@ -209,14 +209,13 @@ def test_cli_edit_none() -> None:
             "--no-token",
             "--skip-update-check",
         ]
-    ):
-        contents = _try_fetch(port)
-        assert contents is not None
-        assert b"marimo-mode data-mode='home'" in contents
-        assert (
-            f"marimo-version data-version='{__version__}'".encode() in contents
-        )
-        assert b"marimo-server-token" in contents
+    )
+    contents = _try_fetch(port)
+    _check_contents(p, b"marimo-mode data-mode='home'", contents)
+    _check_contents(
+        p, f"marimo-version data-version='{__version__}'".encode(), contents
+    )
+    _check_contents(p, b"marimo-server-token", contents)
 
 
 def test_cli_edit_token() -> None: