Files
wehub-resource-sync 4cd2d4af2b
Test Browser Use CLI Install / uv pip install (macos-latest) (push) Waiting to run
Test Browser Use CLI Install / uv pip install (windows-latest) (push) Waiting to run
Test Browser Use CLI Install / uv pip install (ubuntu-latest) (push) Failing after 1s
Test Browser Use CLI Install / uvx browser-use from local wheel (push) Failing after 1s
Test Browser Use CLI Install / uvx browser-use[cli] from PyPI (push) Failing after 1s
package / pip-install-on-macos-latest-py-3.11 (push) Has been skipped
package / pip-install-on-macos-latest-py-3.13 (push) Has been skipped
package / pip-install-on-ubuntu-latest-py-3.11 (push) Has been skipped
package / pip-install-on-windows-latest-py-3.13 (push) Has been skipped
cloud_evals / trigger_cloud_eval_image_build (push) Failing after 1s
docker / build_publish_image (push) Failing after 1s
Test Browser Use CLI Install / browser-use skill sync (push) Failing after 1s
lint / code-style (push) Failing after 0s
lint / type-checker (push) Failing after 1s
package / pip-build (push) Failing after 1s
lint / syntax-errors (push) Failing after 3s
package / pip-install-on-ubuntu-latest-py-3.13 (push) Has been skipped
package / pip-install-on-windows-latest-py-3.11 (push) Has been skipped
test / ${{ matrix.test_filename }} (push) Has been skipped
test / evaluate-tasks (push) Has been skipped
test / setup-chromium (push) Failing after 2s
test / find_tests (push) Failing after 2s
chore: import upstream snapshot with attribution
2026-07-13 12:02:32 +08:00

72 lines
2.8 KiB
Python

"""Test Google model button click."""
from browser_use.llm.google.chat import ChatGoogle
from tests.ci.models.model_test_helper import run_model_button_click_test
async def test_google_gemini_3_flash_preview(httpserver):
"""Test Google gemini-3-flash-preview can click a button."""
await run_model_button_click_test(
model_class=ChatGoogle,
model_name='gemini-3-flash-preview',
api_key_env='GOOGLE_API_KEY',
extra_kwargs={},
httpserver=httpserver,
)
def test_x_goog_api_client_header_is_set():
"""Test that the x-goog-api-client header is correctly set in the HTTP options."""
chat = ChatGoogle(model='gemini-flash-latest', api_key='fake')
# Generate the params used for genai.Client
params = chat._get_client_params()
# Extract the header
http_options = params.get('http_options', {})
headers = http_options.get('headers', {})
assert 'x-goog-api-client' in headers, 'x-goog-api-client header missing'
assert 'browser-use/' in headers['x-goog-api-client'], 'browser-use not found in x-goog-api-client header'
def test_x_goog_api_client_header_with_none_http_options():
"""Test setting header when http_options is None."""
chat = ChatGoogle(model='gemini-flash-latest', api_key='fake', http_options=None)
params = chat._get_client_params()
http_opts = params.get('http_options', {})
assert http_opts.get('headers', {}).get('x-goog-api-client', '').startswith('browser-use/')
def test_x_goog_api_client_header_with_pydantic_http_options():
"""Test setting header when http_options is a types.HttpOptions Pydantic model."""
from google.genai import types
pydantic_opts = types.HttpOptions(timeout=30, headers={'custom-header': 'value'})
chat = ChatGoogle(model='gemini-flash-latest', api_key='fake', http_options=pydantic_opts)
params = chat._get_client_params()
http_opts = params.get('http_options', {})
# Verify it extracts and preserves timeout and custom-header
assert http_opts.get('timeout') == 30
assert http_opts.get('headers', {}).get('custom-header') == 'value'
assert http_opts.get('headers', {}).get('x-goog-api-client', '').startswith('browser-use/')
def test_x_goog_api_client_header_with_dict_http_options():
"""Test setting header when http_options is a dictionary (types.HttpOptionsDict)."""
from google.genai import types
dict_opts: types.HttpOptionsDict = {
'timeout': 45,
'headers': {'another-header': 'another-value'},
}
chat = ChatGoogle(model='gemini-flash-latest', api_key='fake', http_options=dict_opts)
params = chat._get_client_params()
http_opts = params.get('http_options', {})
# Verify it preserves dictionary values and appends the tracking header
assert http_opts.get('timeout') == 45
assert http_opts.get('headers', {}).get('another-header') == 'another-value'
assert http_opts.get('headers', {}).get('x-goog-api-client', '').startswith('browser-use/')