mirror of
https://github.com/browser-use/browser-use.git
synced 2025-02-18 01:18:20 +03:00
fde8544fba
- Refactor dropdown tests to be more robust - Remove hardcoded dependencies - Add minimal test scenarios - Improve test flexibility and reliability - Ensure compatibility with project guidelines Resolves: #543 (Text Input Handling)
68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
"""
|
|
Test configuration for browser-use.
|
|
"""
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
# Ensure the project root is in the Python path
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
|
|
# Load environment variables
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
# Configure logging
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
from langchain_core.language_models.fake import FakeListChatModel
|
|
from langchain_openai import ChatOpenAI
|
|
|
|
from browser_use.browser.browser import Browser, BrowserConfig
|
|
from browser_use.browser.context import BrowserContext
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def llm():
|
|
"""
|
|
Fixture to provide a ChatOpenAI instance or a mock for testing.
|
|
Uses a mock if OPENAI_API_KEY is not set.
|
|
"""
|
|
api_key = os.getenv("OPENAI_API_KEY")
|
|
logger.debug(f"API Key present: {bool(api_key)}")
|
|
if not api_key:
|
|
# Create a mock LLM that always returns a predefined response
|
|
logger.debug("Using FakeListChatModel for testing")
|
|
return FakeListChatModel(
|
|
responses=[
|
|
"Go to a website",
|
|
"Select an option",
|
|
"Perform an action",
|
|
"Test completed"
|
|
]
|
|
)
|
|
logger.debug("Using actual ChatOpenAI model")
|
|
return ChatOpenAI(model='gpt-4o', api_key=api_key)
|
|
|
|
@pytest.fixture(scope="session")
|
|
def browser():
|
|
"""
|
|
Fixture to provide a Browser instance for testing.
|
|
"""
|
|
logger.debug("Creating Browser instance for testing")
|
|
return Browser(config=BrowserConfig(headless=True, disable_security=True))
|
|
|
|
@pytest.fixture(scope="function")
|
|
async def browser_context(browser):
|
|
"""
|
|
Fixture to provide a BrowserContext instance for testing.
|
|
"""
|
|
logger.debug("Creating BrowserContext instance for testing")
|
|
context = BrowserContext(browser=browser)
|
|
yield context
|
|
await context.close()
|