Add additional failure condition when preview is reloading

Siecje · Feb 3, 2024 · de0e6a9 · de0e6a9
1 parent 232e818
commit de0e6a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_preview.py b/tests/test_preview.py
@@ -148,7 +148,7 @@ 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 +182,7 @@ 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: