e2e tests passing playwright

Author: kamath

examples/e2e/test_playwright.py

@@ -14,6 +14,7 @@
     playwright_captcha,
     playwright_contexts,
     playwright_downloads,
+    playwright_upload,
 )
 
 bb = Browserbase(api_key=BROWSERBASE_API_KEY)
@@ -23,7 +24,6 @@
 
 
 @pytest.fixture(scope="session")
-@pytest.mark.skipif(True, reason="Flaky and fails on CI")
 def playwright() -> Generator[Playwright, None, None]:
     with sync_playwright() as p:
         yield p
@@ -33,17 +33,15 @@ def test_playwright_basic(playwright: Playwright) -> None:
     playwright_basic.run(playwright)
 
 
-@pytest.mark.skipif(True, reason="Flaky and fails on CI")
+@pytest.mark.skipif(True, reason="Flaky and fails often")
 def test_playwright_captcha(playwright: Playwright) -> None:
     playwright_captcha.run(playwright)
 
 
-@pytest.mark.skipif(True, reason="Flaky and fails on CI")
 def test_playwright_contexts(playwright: Playwright) -> None:
     playwright_contexts.run(playwright)
 
 
-@pytest.mark.skipif(True, reason="Flaky and fails on CI")
 def test_playwright_downloads(playwright: Playwright) -> None:
     playwright_downloads.run(playwright)
 
@@ -74,3 +72,7 @@ def test_playwright_proxy_geolocation_american_city(playwright: Playwright) -> N
 @pytest.mark.skipif(CI, reason="Flaky and fails on CI")
 def test_playwright_proxy_geolocation_non_american_city(playwright: Playwright) -> None:
     playwright_proxy.run_geolocation_non_american_city(playwright)
+
+
+def test_playwright_upload(playwright: Playwright) -> None:
+    playwright_upload.run(playwright)

examples/packages/logo.png

@@ -1,9 +1,7 @@
 from playwright.sync_api import Playwright, sync_playwright
 
 from examples import (
-    BROWSERBASE_API_KEY,
     BROWSERBASE_PROJECT_ID,
-    BROWSERBASE_CONNECT_URL,
     bb,
 )
 
@@ -15,11 +13,8 @@ def run(playwright: Playwright) -> None:
     assert session.status == "RUNNING", f"Session status is {session.status}"
 
     # Connect to the remote session
-    connect_url = (
-        f"{BROWSERBASE_CONNECT_URL}?sessionId={session.id}&apiKey={BROWSERBASE_API_KEY}"
-    )
     chromium = playwright.chromium
-    browser = chromium.connect_over_cdp(connect_url)
+    browser = chromium.connect_over_cdp(session.connect_url)
     context = browser.contexts[0]
     page = context.pages[0]
 

examples/playwright_basic.py

@@ -0,0 +1,62 @@
+from pathlib import Path
+from playwright.sync_api import Playwright, sync_playwright
+
+from examples import (
+    BROWSERBASE_API_KEY,
+    BROWSERBASE_PROJECT_ID,
+    BROWSERBASE_CONNECT_URL,
+    bb,
+)
+
+PATH_TO_UPLOAD = Path.cwd() / "examples" / "packages" / "logo.png"
+
+
+def run(playwright: Playwright) -> None:
+    # Create a session
+    session = bb.sessions.create(project_id=BROWSERBASE_PROJECT_ID)
+
+    # Construct the URL
+    url = (
+        f"{BROWSERBASE_CONNECT_URL}?apiKey={BROWSERBASE_API_KEY}&sessionId={session.id}"
+    )
+
+    # Connect to the browser
+    browser = playwright.chromium.connect_over_cdp(url)
+    context = browser.contexts[0]
+    page = context.pages[0]
+
+    try:
+        # Navigate to the upload test page
+        page.goto("https://browser-tests-alpha.vercel.app/api/upload-test")
+
+        # Locate the file input element
+        file_input = page.locator("#fileUpload")
+        file_input.set_input_files(str(PATH_TO_UPLOAD))
+
+        # Get the uploaded file name
+        file_name_span = page.locator("#fileName")
+        file_name = file_name_span.inner_text()
+
+        # Get the uploaded file size
+        file_size_span = page.locator("#fileSize")
+        file_size = int(file_size_span.inner_text())
+
+        # Assert the file name and size
+        assert (
+            file_name == "logo.png"
+        ), f"Expected file name to be 'logo.png', but got '{file_name}'"
+        assert (
+            file_size > 0
+        ), f"Expected file size to be greater than 0, but got {file_size}"
+
+        print("File upload test passed successfully!")
+
+    finally:
+        # Clean up
+        page.close()
+        browser.close()
+
+
+if __name__ == "__main__":
+    with sync_playwright() as playwright:
+        run(playwright)