ayrisdev/crawl4ai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1528 from unclecode/fix/managed-browser-cdp-timing

Browse Source 
Add CDP endpoint verification with exponential backoff for managed browsers
This commit is contained in:
Nasrin
2025-11-12 23:53:57 +08:00
committed by GitHub
parent be00fc3a42 34c0996ee4
commit b207ae2848
2 changed files with 301 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

314
tests/browser/test_cdp_strategy.py
View File

@@ -7,12 +7,13 @@ and serve as functional tests.
import asyncio
import os
import sys
import time
# Add the project root to Python path if running directly
if __name__ == "__main__":
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../..')))
from crawl4ai.browser import BrowserManager
from crawl4ai.browser_manager import BrowserManager
from crawl4ai.async_configs import BrowserConfig, CrawlerRunConfig
from crawl4ai.async_logger import AsyncLogger
@@ -24,8 +25,8 @@ async def test_cdp_launch_connect():
logger.info("Testing launch and connect via CDP", tag="TEST")
browser_config = BrowserConfig(
use_managed_browser=True,
browser_mode="cdp",
use_managed_browser=True,
headless=True
)
@@ -62,17 +63,18 @@ async def test_cdp_launch_connect():
return False
async def test_cdp_with_user_data_dir():
"""Test CDP browser with a user data directory."""
"""Test CDP browser with a user data directory and storage state."""
logger.info("Testing CDP browser with user data directory", tag="TEST")
# Create a temporary user data directory
import tempfile
user_data_dir = tempfile.mkdtemp(prefix="crawl4ai-test-")
storage_state_file = os.path.join(user_data_dir, "storage_state.json")
logger.info(f"Created temporary user data directory: {user_data_dir}", tag="TEST")
browser_config = BrowserConfig(
headless=True,
browser_mode="cdp",
use_managed_browser=True,
user_data_dir=user_data_dir
)
@@ -86,38 +88,59 @@ async def test_cdp_with_user_data_dir():
crawler_config = CrawlerRunConfig()
page, context = await manager.get_page(crawler_config)
# Set a cookie
# Visit the site first
await page.goto("https://example.com", wait_until="domcontentloaded")
# Set a cookie via JavaScript (more reliable for persistence)
await page.evaluate("""
document.cookie = 'test_cookie=test_value; path=/; max-age=86400';
""")
# Also set via context API for double coverage
await context.add_cookies([{
"name": "test_cookie",
"value": "test_value",
"url": "https://example.com"
"name": "test_cookie_api",
"value": "test_value_api",
"domain": "example.com",
"path": "/"
}])
# Visit the site
await page.goto("https://example.com")
# Verify cookie was set
# Verify cookies were set
cookies = await context.cookies(["https://example.com"])
has_test_cookie = any(cookie["name"] == "test_cookie" for cookie in cookies)
has_test_cookie = any(cookie["name"] in ["test_cookie", "test_cookie_api"] for cookie in cookies)
logger.info(f"Cookie set successfully: {has_test_cookie}", tag="TEST")
# Save storage state before closing
await context.storage_state(path=storage_state_file)
logger.info(f"Storage state saved to: {storage_state_file}", tag="TEST")
# Close the browser
await manager.close()
logger.info("First browser session closed", tag="TEST")
# Start a new browser with the same user data directory
# Wait a moment for clean shutdown
await asyncio.sleep(1.0)
# Start a new browser with the same user data directory and storage state
logger.info("Starting second browser session with same user data directory", tag="TEST")
manager2 = BrowserManager(browser_config=browser_config, logger=logger)
browser_config2 = BrowserConfig(
headless=True,
use_managed_browser=True,
user_data_dir=user_data_dir,
storage_state=storage_state_file
)
manager2 = BrowserManager(browser_config=browser_config2, logger=logger)
await manager2.start()
# Get a new page and check if the cookie persists
page2, context2 = await manager2.get_page(crawler_config)
await page2.goto("https://example.com")
await page2.goto("https://example.com", wait_until="domcontentloaded")
# Verify cookie persisted
cookies2 = await context2.cookies(["https://example.com"])
has_test_cookie2 = any(cookie["name"] == "test_cookie" for cookie in cookies2)
has_test_cookie2 = any(cookie["name"] in ["test_cookie", "test_cookie_api"] for cookie in cookies2)
logger.info(f"Cookie persisted across sessions: {has_test_cookie2}", tag="TEST")
logger.info(f"Cookies found: {[c['name'] for c in cookies2]}", tag="TEST")
# Clean up
await manager2.close()
@@ -134,6 +157,10 @@ async def test_cdp_with_user_data_dir():
await manager.close()
except:
pass
try:
await manager2.close()
except:
pass
# Clean up temporary directory
try:
@@ -145,7 +172,7 @@ async def test_cdp_with_user_data_dir():
return False
async def test_cdp_session_management():
"""Test session management with CDP browser."""
"""Test session management with CDP browser - focused on session tracking."""
logger.info("Testing session management with CDP browser", tag="TEST")
browser_config = BrowserConfig(
@@ -159,45 +186,104 @@ async def test_cdp_session_management():
await manager.start()
logger.info("Browser launched successfully", tag="TEST")
# Create two sessions
# Test session tracking and lifecycle management
session1_id = "test_session_1"
session2_id = "test_session_2"
# Set up first session
crawler_config1 = CrawlerRunConfig(session_id=session1_id)
page1, context1 = await manager.get_page(crawler_config1)
await page1.goto("https://example.com")
await page1.evaluate("localStorage.setItem('session1_data', 'test_value')")
logger.info(f"Set up session 1 with ID: {session1_id}", tag="TEST")
await page1.goto("https://example.com", wait_until="domcontentloaded")
# Set up second session
# Get page URL and title for verification
page1_url = page1.url
page1_title = await page1.title()
logger.info(f"Session 1 setup - URL: {page1_url}, Title: {page1_title}", tag="TEST")
# Set up second session
crawler_config2 = CrawlerRunConfig(session_id=session2_id)
page2, context2 = await manager.get_page(crawler_config2)
await page2.goto("https://example.org")
await page2.evaluate("localStorage.setItem('session2_data', 'test_value2')")
logger.info(f"Set up session 2 with ID: {session2_id}", tag="TEST")
await page2.goto("https://httpbin.org/html", wait_until="domcontentloaded")
# Get first session again
page1_again, _ = await manager.get_page(crawler_config1)
page2_url = page2.url
page2_title = await page2.title()
logger.info(f"Session 2 setup - URL: {page2_url}, Title: {page2_title}", tag="TEST")
# Verify it's the same page and data persists
# Verify sessions exist in manager
session1_exists = session1_id in manager.sessions
session2_exists = session2_id in manager.sessions
logger.info(f"Sessions in manager - S1: {session1_exists}, S2: {session2_exists}", tag="TEST")
# Test session reuse
page1_again, context1_again = await manager.get_page(crawler_config1)
is_same_page = page1 == page1_again
data1 = await page1_again.evaluate("localStorage.getItem('session1_data')")
logger.info(f"Session 1 reuse successful: {is_same_page}, data: {data1}", tag="TEST")
is_same_context = context1 == context1_again
# Kill first session
logger.info(f"Session 1 reuse - Same page: {is_same_page}, Same context: {is_same_context}", tag="TEST")
# Test that sessions are properly tracked with timestamps
session1_info = manager.sessions.get(session1_id)
session2_info = manager.sessions.get(session2_id)
session1_has_timestamp = session1_info and len(session1_info) == 3
session2_has_timestamp = session2_info and len(session2_info) == 3
logger.info(f"Session tracking - S1 complete: {session1_has_timestamp}, S2 complete: {session2_has_timestamp}", tag="TEST")
# In managed browser mode, pages might be shared. Let's test what actually happens
pages_same_or_different = page1 == page2
logger.info(f"Pages same object: {pages_same_or_different}", tag="TEST")
# Test that we can distinguish sessions by their stored info
session1_context, session1_page, session1_time = session1_info
session2_context, session2_page, session2_time = session2_info
sessions_have_different_timestamps = session1_time != session2_time
logger.info(f"Sessions have different timestamps: {sessions_have_different_timestamps}", tag="TEST")
# Test session killing
await manager.kill_session(session1_id)
logger.info(f"Killed session 1", tag="TEST")
# Verify second session still works
data2 = await page2.evaluate("localStorage.getItem('session2_data')")
logger.info(f"Session 2 still functional after killing session 1, data: {data2}", tag="TEST")
# Verify session was removed
session1_removed = session1_id not in manager.sessions
session2_still_exists = session2_id in manager.sessions
logger.info(f"After kill - S1 removed: {session1_removed}, S2 exists: {session2_still_exists}", tag="TEST")
# Test page state after killing session
page1_closed = page1.is_closed()
logger.info(f"Page1 closed after kill: {page1_closed}", tag="TEST")
# Clean up remaining session
try:
await manager.kill_session(session2_id)
logger.info("Killed session 2", tag="TEST")
session2_removed = session2_id not in manager.sessions
except Exception as e:
logger.info(f"Session 2 cleanup: {e}", tag="TEST")
session2_removed = False
# Clean up
await manager.close()
logger.info("Browser closed successfully", tag="TEST")
return is_same_page and data1 == "test_value" and data2 == "test_value2"
# Success criteria for managed browser sessions:
# 1. Sessions can be created and tracked with proper info
# 2. Same page/context returned for same session ID
# 3. Sessions have proper timestamp tracking
# 4. Sessions can be killed and removed from tracking
# 5. Session cleanup works properly
success = (session1_exists and
session2_exists and
is_same_page and
session1_has_timestamp and
session2_has_timestamp and
sessions_have_different_timestamps and
session1_removed and
session2_removed)
logger.info(f"Test success: {success}", tag="TEST")
return success
except Exception as e:
logger.error(f"Test failed: {str(e)}", tag="TEST")
try:
@@ -206,14 +292,170 @@ async def test_cdp_session_management():
pass
return False
async def test_cdp_timing_fix_fast_startup():
"""
Test that the CDP timing fix handles fast browser startup correctly.
This should work without any delays or retries.
"""
logger.info("Testing CDP timing fix with fast startup", tag="TEST")
browser_config = BrowserConfig(
use_managed_browser=True,
browser_mode="cdp",
headless=True,
debugging_port=9223, # Use different port to avoid conflicts
verbose=True
)
manager = BrowserManager(browser_config=browser_config, logger=logger)
try:
start_time = time.time()
await manager.start()
startup_time = time.time() - start_time
logger.info(f"Browser started successfully in {startup_time:.2f}s", tag="TEST")
# Test basic functionality
crawler_config = CrawlerRunConfig(url="https://example.com")
page, context = await manager.get_page(crawler_config)
await page.goto("https://example.com", wait_until="domcontentloaded")
title = await page.title()
logger.info(f"Successfully navigated to page: {title}", tag="TEST")
await manager.close()
logger.success("test_cdp_timing_fix_fast_startup completed successfully", tag="TEST")
return True
except Exception as e:
logger.error(f"test_cdp_timing_fix_fast_startup failed: {str(e)}", tag="TEST")
try:
await manager.close()
except:
pass
return False
async def test_cdp_timing_fix_delayed_browser_start():
"""
Test CDP timing fix by actually delaying the browser startup process.
This simulates a real scenario where the browser takes time to expose CDP.
"""
logger.info("Testing CDP timing fix with delayed browser startup", tag="TEST")
browser_config = BrowserConfig(
use_managed_browser=True,
browser_mode="cdp",
headless=True,
debugging_port=9224,
verbose=True
)
# Start the managed browser separately to control timing
from crawl4ai.browser_manager import ManagedBrowser
managed_browser = ManagedBrowser(browser_config=browser_config, logger=logger)
try:
# Start browser process but it will take time for CDP to be ready
cdp_url = await managed_browser.start()
logger.info(f"Managed browser started at {cdp_url}", tag="TEST")
# Small delay to simulate the browser needing time to fully initialize CDP
await asyncio.sleep(1.0)
# Now create BrowserManager and connect - this should use the CDP verification fix
manager = BrowserManager(browser_config=browser_config, logger=logger)
manager.config.cdp_url = cdp_url # Use the CDP URL from managed browser
start_time = time.time()
await manager.start()
startup_time = time.time() - start_time
logger.info(f"BrowserManager connected successfully in {startup_time:.2f}s", tag="TEST")
# Test basic functionality
crawler_config = CrawlerRunConfig(url="https://example.com")
page, context = await manager.get_page(crawler_config)
await page.goto("https://example.com", wait_until="domcontentloaded")
title = await page.title()
logger.info(f"Successfully navigated to page: {title}", tag="TEST")
# Clean up
await manager.close()
await managed_browser.cleanup()
logger.success("test_cdp_timing_fix_delayed_browser_start completed successfully", tag="TEST")
return True
except Exception as e:
logger.error(f"test_cdp_timing_fix_delayed_browser_start failed: {str(e)}", tag="TEST")
try:
await manager.close()
await managed_browser.cleanup()
except:
pass
return False
async def test_cdp_verification_backoff_behavior():
"""
Test the exponential backoff behavior of CDP verification in isolation.
"""
logger.info("Testing CDP verification exponential backoff behavior", tag="TEST")
browser_config = BrowserConfig(
use_managed_browser=True,
debugging_port=9225, # Use different port
verbose=True
)
manager = BrowserManager(browser_config=browser_config, logger=logger)
try:
# Test with a non-existent CDP URL to trigger retries
fake_cdp_url = "http://localhost:19999" # This should not exist
start_time = time.time()
result = await manager._verify_cdp_ready(fake_cdp_url)
elapsed_time = time.time() - start_time
# Should return False after all retries
assert result is False, "Expected CDP verification to fail with non-existent endpoint"
# Should take some time due to retries and backoff
assert elapsed_time > 2.0, f"Expected backoff delays, but took only {elapsed_time:.2f}s"
logger.info(f"CDP verification correctly failed after {elapsed_time:.2f}s with exponential backoff", tag="TEST")
logger.success("test_cdp_verification_backoff_behavior completed successfully", tag="TEST")
return True
except Exception as e:
logger.error(f"test_cdp_verification_backoff_behavior failed: {str(e)}", tag="TEST")
return False
async def run_tests():
"""Run all tests sequentially."""
import time
results = []
# Original CDP strategy tests
logger.info("Running original CDP strategy tests", tag="SUITE")
# results.append(await test_cdp_launch_connect())
results.append(await test_cdp_with_user_data_dir())
results.append(await test_cdp_session_management())
# CDP timing fix tests
logger.info("Running CDP timing fix tests", tag="SUITE")
results.append(await test_cdp_timing_fix_fast_startup())
results.append(await test_cdp_timing_fix_delayed_browser_start())
results.append(await test_cdp_verification_backoff_behavior())
# Print summary
total = len(results)
passed = sum(results)