Add additional failure condition when preview is reloading

Siecje · Feb 3, 2024 · 40e46d0 · 40e46d0
1 parent 232e818
commit 40e46d0
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/test_preview.py b/tests/test_preview.py
@@ -148,7 +148,10 @@ def test_preview_reload_css(run_start: CliRunner) -> None:  # noqa: ARG001
         while after == before:
             try:
                 response = requests.get(url, timeout=0.1)
-            except requests.exceptions.ReadTimeout:
+            except (
+                requests.exceptions.ReadTimeout,
+                requests.exceptions.ConnectionError,
+            ):
                 # happens during restart
                 read_timeout = True
             else:
@@ -182,7 +185,10 @@ def test_preview_reload_js(run_start: CliRunner) -> None:  # noqa: ARG001
         while after == before:
             try:
                 response = requests.get(url, timeout=0.1)
-            except requests.exceptions.ReadTimeout:
+            except (
+                requests.exceptions.ReadTimeout,
+                requests.exceptions.ConnectionError,
+            ):
                 # happens during restart
                 read_timeout = True
             else: