From d96f0514861d5f496de6a78383e584ced20a961e Mon Sep 17 00:00:00 2001
From: benedikt-bartscher
 <31854409+benedikt-bartscher@users.noreply.github.com>
Date: Thu, 23 May 2024 18:23:07 +0200
Subject: [PATCH] set config.deploy_url during AppHarness tests (#3359)

---
 integration/test_deploy_url.py | 98 ++++++++++++++++++++++++++++++++++
 reflex/testing.py              |  2 +
 2 files changed, 100 insertions(+)
 create mode 100644 integration/test_deploy_url.py

diff --git a/integration/test_deploy_url.py b/integration/test_deploy_url.py
new file mode 100644
index 000000000..b0421cfb7
--- /dev/null
+++ b/integration/test_deploy_url.py
@@ -0,0 +1,98 @@
+"""Integration tests for deploy_url."""
+
+from __future__ import annotations
+
+from typing import Generator
+
+import pytest
+from selenium.webdriver.common.by import By
+from selenium.webdriver.remote.webdriver import WebDriver
+from selenium.webdriver.support.ui import WebDriverWait
+
+from reflex.testing import AppHarness
+
+
+def DeployUrlSample() -> None:
+    """Sample app for testing config deploy_url is correct (in tests)."""
+    import reflex as rx
+
+    class State(rx.State):
+        def goto_self(self):
+            return rx.redirect(rx.config.get_config().deploy_url)  # type: ignore
+
+    def index():
+        return rx.fragment(
+            rx.button("GOTO SELF", on_click=State.goto_self, id="goto_self")
+        )
+
+    app = rx.App(state=rx.State)
+    app.add_page(index)
+
+
+@pytest.fixture(scope="module")
+def deploy_url_sample(
+    tmp_path_factory: pytest.TempPathFactory,
+) -> Generator[AppHarness, None, None]:
+    """AppHarness fixture for testing deploy_url.
+
+    Args:
+        tmp_path_factory: pytest fixture for creating temporary directories.
+
+    Yields:
+        AppHarness: An AppHarness instance.
+    """
+    with AppHarness.create(
+        root=tmp_path_factory.mktemp("deploy_url_sample"),
+        app_source=DeployUrlSample,  # type: ignore
+    ) as harness:
+        yield harness
+
+
+@pytest.fixture()
+def driver(deploy_url_sample: AppHarness) -> Generator[WebDriver, None, None]:
+    """WebDriver fixture for testing deploy_url.
+
+    Args:
+        deploy_url_sample: AppHarness fixture for testing deploy_url.
+
+    Yields:
+        WebDriver: A WebDriver instance.
+    """
+    assert deploy_url_sample.app_instance is not None, "app is not running"
+    driver = deploy_url_sample.frontend()
+    try:
+        yield driver
+    finally:
+        driver.quit()
+
+
+def test_deploy_url(deploy_url_sample: AppHarness, driver: WebDriver) -> None:
+    """Test deploy_url is correct.
+
+    Args:
+        deploy_url_sample: AppHarness fixture for testing deploy_url.
+        driver: WebDriver fixture for testing deploy_url.
+    """
+    import reflex as rx
+
+    deploy_url = rx.config.get_config().deploy_url
+    assert deploy_url is not None
+    assert deploy_url != "http://localhost:3000"
+    assert deploy_url == deploy_url_sample.frontend_url
+    driver.get(deploy_url)
+    assert driver.current_url == deploy_url + "/"
+
+
+def test_deploy_url_in_app(deploy_url_sample: AppHarness, driver: WebDriver) -> None:
+    """Test deploy_url is correct in app.
+
+    Args:
+        deploy_url_sample: AppHarness fixture for testing deploy_url.
+        driver: WebDriver fixture for testing deploy_url.
+    """
+    driver.implicitly_wait(10)
+    driver.find_element(By.ID, "goto_self").click()
+
+    WebDriverWait(driver, 10).until(
+        lambda driver: driver.current_url == f"{deploy_url_sample.frontend_url}/"
+    )
diff --git a/reflex/testing.py b/reflex/testing.py
index aa8cbb893..ad0839632 100644
--- a/reflex/testing.py
+++ b/reflex/testing.py
@@ -325,6 +325,8 @@ class AppHarness:
             m = re.search(reflex.constants.Next.FRONTEND_LISTENING_REGEX, line)
             if m is not None:
                 self.frontend_url = m.group(1)
+                config = reflex.config.get_config()
+                config.deploy_url = self.frontend_url
                 break
         if self.frontend_url is None:
             raise RuntimeError("Frontend did not start")