fix three tests

Author: adhami3310

reflex/.templates/web/vite.config.js

@@ -14,7 +14,7 @@ export default defineConfig((config) => ({
     port: process.env.PORT,
   },
   resolve: {
-    mainFields: ['browser', 'module', 'jsnext'],
+    mainFields: ["browser", "module", "jsnext"],
     alias: [
       {
         find: "$",

tests/integration/test_login_flow.py

@@ -113,19 +113,21 @@ def test_login_flow(
     assert login_sample.frontend_url is not None
     local_storage.clear()
 
+    login_button = login_sample.poll_for_result(
+        lambda: driver.find_element(By.ID, "login")
+    )
     with pytest.raises(NoSuchElementException):
         driver.find_element(By.ID, "auth-token")
 
-    login_button = driver.find_element(By.ID, "login")
     login_sample.poll_for_content(login_button)
     with utils.poll_for_navigation(driver):
         login_button.click()
-    assert driver.current_url.endswith("/login/")
+    assert driver.current_url.endswith("/login")
 
     do_it_button = driver.find_element(By.ID, "doit")
     with utils.poll_for_navigation(driver):
         do_it_button.click()
-    assert driver.current_url == login_sample.frontend_url + "/"
+    assert driver.current_url == login_sample.frontend_url
 
     def check_auth_token_header():
         try:

tests/integration/test_memo.py

@@ -86,7 +86,9 @@ async def test_memo_app(memo_app: AppHarness):
     driver = memo_app.frontend()
 
     # check that the output matches
-    memo_custom_code_stack = driver.find_element(By.ID, "memo-custom-code")
+    memo_custom_code_stack = memo_app.poll_for_result(
+        lambda: driver.find_element(By.ID, "memo-custom-code")
+    )
     assert (
         memo_app.poll_for_content(memo_custom_code_stack, exp_not_equal="")
         == "foobarbarbar"

tests/integration/test_navigation.py

@@ -74,11 +74,13 @@ async def test_navigation_app(navigation_app: AppHarness):
 
     with poll_for_navigation(driver):
         internal_link.click()
-    assert urlsplit(driver.current_url).path == "/internal/"
+    assert urlsplit(driver.current_url).path == "/internal"
     with poll_for_navigation(driver):
         driver.back()
 
-    external_link = driver.find_element(By.ID, "external")
+    external_link = navigation_app.poll_for_result(
+        lambda: driver.find_element(By.ID, "external")
+    )
     external2_link = driver.find_element(By.ID, "external2")
 
     external_link.click()