temp

Author: kamath

.gitignore

@@ -15,3 +15,4 @@ dist
 codegen.log
 Brewfile.lock.json
 screenshot.png
+openapi.v1.yaml

e2e/test_session.py

@@ -0,0 +1,24 @@
+import os
+
+from dotenv import load_dotenv
+
+from browserbase import Browserbase
+
+# Load our environment variables
+load_dotenv(override=True)
+
+# Make sure we have the required environment variables
+BROWSERBASE_CONNECT_URL = os.environ.get(
+    "BROWSERBASE_CONNECT_URL", "wss://connect.browserbase.com"
+)
+_BROWSERBASE_API_KEY = os.environ.get("BROWSERBASE_API_KEY")
+if not _BROWSERBASE_API_KEY:
+    raise ValueError("BROWSERBASE_API_KEY is not set in environment")
+BROWSERBASE_API_KEY: str = _BROWSERBASE_API_KEY
+_BROWSERBASE_PROJECT_ID = os.environ.get("BROWSERBASE_PROJECT_ID")
+if not _BROWSERBASE_PROJECT_ID:
+    raise ValueError("BROWSERBASE_PROJECT_ID is not set in environment")
+BROWSERBASE_PROJECT_ID = _BROWSERBASE_PROJECT_ID or ""
+
+# Instantiate our Browserbase client
+bb = Browserbase(api_key=BROWSERBASE_API_KEY)

examples/__init__.py

@@ -0,0 +1,26 @@
+# End-to-end tests
+
+This directory contains end-to-end tests that run against a real Browserbase instance.
+
+## Running the tests
+
+To run the tests, you will need to set the following environment variables:
+
+- `BROWSERBASE_API_KEY`: Your Browserbase API key
+- `BROWSERBASE_PROJECT_ID`: The ID of the project you want to use for the tests
+
+You can set these variables in a `.env` file in the root of this directory.
+
+Then, run the tests with:
+
+```sh
+$ rye run test:e2e
+```
+
+## Writing tests
+
+The tests are written using pytest and the [pytest-playwright](https://playwright.dev/python/docs/pytest) plugin.
+
+You can find more information about writing tests in the [pytest documentation](https://docs.pytest.org/en/7.1.x/).
+
+To submit a test, create a new file in the `e2e` directory with a name that describes the test and starts with `test_`.

examples/e2e/README.md

@@ -0,0 +1 @@
+

examples/e2e/__init__.py

@@ -0,0 +1,22 @@
+import pytest
+from playwright.sync_api import Playwright, sync_playwright
+
+from browserbase import Browserbase
+
+from .. import (
+    BROWSERBASE_API_KEY,
+    playwright_basic,
+)
+
+bb = Browserbase(api_key=BROWSERBASE_API_KEY)
+
+
+@pytest.fixture(scope="session")
+def playwright():
+    with sync_playwright() as p:
+        yield p
+
+
+def test_create_session(playwright: Playwright):
+    page = playwright_basic.run(playwright)
+    assert page["title"] == "Hacker News"

examples/e2e/test_playwright_basic.py

@@ -0,0 +1,34 @@
+from playwright.sync_api import Playwright, sync_playwright
+
+from examples import (BROWSERBASE_API_KEY, BROWSERBASE_CONNECT_URL,
+                      BROWSERBASE_PROJECT_ID, bb)
+
+
+def run(playwright: Playwright):
+    # Create a session on Browserbase
+    session = bb.sessions.create(project_id=BROWSERBASE_PROJECT_ID)
+
+    # 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)
+    context = browser.contexts[0]
+    page = context.pages[0]
+
+    # Execute Playwright actions on the remote browser tab
+    page.goto("https://news.ycombinator.com/")
+    assert page.title == "Hacker News"
+    page.screenshot(path="screenshot.png")
+
+    updated_session = bb.sessions.retrieve(session.id)
+    print(updated_session.)
+
+    page.close()
+    browser.close()
+
+
+if __name__ == "__main__":
+    with sync_playwright() as playwright:
+        run(playwright)

examples/playwright_basic.py

@@ -82,9 +82,9 @@ format = { chain = [
 
 "check:importable" = "python -c 'import browserbase'"
 
-"example" = "python -c 'import sys; from pathlib import Path; example = Path(\"examples\") / sys.argv[1]; exec(open(example).read())'"
+"example" = "python -c 'import sys; from pathlib import Path; example = Path(\"examples\") / (sys.argv[1] + \".py\"); exec(open(example).read())'"
 
-"test:e2e" = "python -m pytest e2e"
+"test:e2e" = "python -m pytest examples/e2e"
 
 typecheck = { chain = [
   "typecheck:pyright",