basic selenium

Author: kamath

examples/e2e/test_playwright.py

@@ -20,9 +20,10 @@
 load_dotenv()
 
 CI = os.getenv("CI", "false").lower() == "true"
+MAX_RETRIES = 3
 
 
-@pytest.fixture(scope="session")
+@pytest.fixture(scope="function")  # Changed from "session" to "function"
 def playwright() -> Generator[Playwright, None, None]:
     with sync_playwright() as p:
         yield p

examples/e2e/test_selenium.py

@@ -0,0 +1,5 @@
+from .. import selenium_basic
+
+
+def test_selenium_basic() -> None:
+    selenium_basic.run()

examples/playwright_contexts.py

@@ -1,5 +1,6 @@
 import time
 from typing import Optional
+from pydantic import TypeAdapter
 
 from playwright.sync_api import Cookie, Browser, Playwright, sync_playwright
 
@@ -47,10 +48,11 @@ def run(playwright: Playwright) -> None:
     # Step 2: Creates a session with the context
     session = bb.sessions.create(
         project_id=BROWSERBASE_PROJECT_ID,
-        browser_settings=BrowserSettings(
-            context=BrowserSettingsContext(id=context_id, persist=True),
+        browser_settings=TypeAdapter(BrowserSettings).validate_python(
+            {"context": {"id": context_id, "persist": True}}
         ),
     )
+    print(session)
 
     assert (
         session.context_id == context_id

examples/selenium_basic.py

@@ -1,24 +1,24 @@
 from examples import bb, BROWSERBASE_PROJECT_ID, BROWSERBASE_API_KEY
-from browserbase.types import SessionCreateResponse
 from selenium import webdriver
-from selenium.webdriver.remote.webdriver import WebDriver
 from selenium.webdriver.remote.remote_connection import RemoteConnection
+from typing import Dict
 
 
 class BrowserbaseConnection(RemoteConnection):
     """
     Manage a single session with Browserbase.
     """
 
-    browserbase_session: SessionCreateResponse
     session_id: str
 
-    def __init__(self, session_id: str, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(self, session_id: str, *args, **kwargs):  # type: ignore
+        super().__init__(*args, **kwargs)  # type: ignore
         self.session_id = session_id
 
-    def get_remote_connection_headers(self, parsed_url, keep_alive=False):
-        headers = super().get_remote_connection_headers(parsed_url, keep_alive)
+    def get_remote_connection_headers(  # type: ignore
+        self, parsed_url: str, keep_alive: bool = False
+    ) -> Dict[str, str]:
+        headers = super().get_remote_connection_headers(parsed_url, keep_alive)  # type: ignore
 
         # Update headers to include the Browserbase required information
         headers["x-bb-api-key"] = BROWSERBASE_API_KEY
@@ -27,31 +27,32 @@ def get_remote_connection_headers(self, parsed_url, keep_alive=False):
         return headers
 
 
-def run(driver: WebDriver):
-    # Instruct the browser to go to the SF MOMA page
-    driver.get("https://www.sfmoma.org")
+def run():
+    # Use the custom class to create and connect to a new browser session
+    session = bb.sessions.create(project_id=BROWSERBASE_PROJECT_ID)
+    connection = BrowserbaseConnection(session.id, session.selenium_remote_url)
+    driver = webdriver.Remote(
+        command_executor=connection, options=webdriver.ChromeOptions()  # type: ignore
+    )
 
-    # Print out a bit of info about the page it landed on
-    print(f"{driver.current_url=} | {driver.title=}")
+    # Print a bit of info about the browser we've connected to
+    print(
+        "Connected to Browserbase",
+        f"{driver.name} version {driver.caps['browserVersion']}",  # type: ignore
+    )
 
-    ...
+    try:
+        # Perform our browser commands
+        driver.get("https://www.sfmoma.org")
+        print(f"At URL: {driver.current_url} | Title: {driver.title}")
+        assert driver.current_url == "https://www.sfmoma.org/"
+        assert driver.title == "SFMOMA"
 
+    finally:
+        # Make sure to quit the driver so your session is ended!
+        print("Quitting driver")
+        # driver.quit()
 
-# Use the custom class to create and connect to a new browser session
-session = bb.sessions.create(project_id=BROWSERBASE_PROJECT_ID)
-connection = BrowserbaseConnection(session.id, session.selenium_remote_url)
-driver = webdriver.Remote(connection, options=webdriver.ChromeOptions())
 
-# Print a bit of info about the browser we've connected to
-print(
-    "Connected to Browserbase",
-    f"{driver.name} version {driver.caps['browserVersion']}",
-)
-
-try:
-    # Perform our browser commands
-    run(driver)
-
-finally:
-    # Make sure to quit the driver so your session is ended!
-    driver.quit()
+if __name__ == "__main__":
+    run()