import asyncio
import os
import sys
from browser_use.llm.openai.chat import ChatOpenAI
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
if project_root not in sys.path:
sys.path.insert(0, project_root)
import pytest
from dotenv import load_dotenv
# Third-party imports
from browser_use import Agent, Controller
# Local imports
from browser_use.browser import BrowserProfile, BrowserSession
# Load environment variables.
load_dotenv()
# Initialize language model and controller.
llm = ChatOpenAI(model='gpt-4.1')
controller = Controller()
@pytest.mark.skip(reason='this is for local testing only')
async def test_wait_for_element():
"""Test 'Wait for element' action."""
initial_actions = [
{'go_to_url': {'url': 'https://pypi.org/', 'new_tab': True}},
# Uncomment the line below to include the wait action in initial actions.
# {'wait_for_element': {'selector': '#search', 'timeout': 30}},
]
# Set up the browser session.
browser_session = BrowserSession(browser_profile=BrowserProfile(headless=True, disable_security=True))
await browser_session.start()
try:
# Create the agent with the task.
agent = Agent(
task="Wait for element '#search' to be visible with a timeout of 30 seconds.",
llm=llm,
browser_session=browser_session,
initial_actions=initial_actions,
controller=controller,
)
# Run the agent for a few steps to trigger navigation and then the wait action.
history = await agent.run(max_steps=3)
action_names = history.action_names()
# Ensure that the wait_for_element action was executed.
assert 'wait_for_element' in action_names, 'Expected wait_for_element action to be executed.'
# Verify that the #search element is visible by querying the page.
page = await browser_session.get_current_page()
header_handle = await page.query_selector('#search')
assert header_handle is not None, 'Expected to find a #search element on the page.'
is_visible = await header_handle.is_visible()
assert is_visible, 'Expected the #search element to be visible.'
finally:
await browser_session.stop()
if __name__ == '__main__':
asyncio.run(test_wait_for_element())
