mirror of
https://github.com/browser-use/browser-use.git
synced 2026-09-14 19:59:47 +08:00
ShawnPana
d3e1818496
Unify all CLI-managed files under ~/.browser-use/ (config, sockets, PIDs, binaries, tunnels) instead of scattering across ~/.config/browser-use/ and ~/.browser-use/run/. Add profile-use Go binary as managed subcommand via browser-use profile, with auto-download fallback and install.sh integration. Wire cloudflared and profile-use availability checks into browser-use doctor.
98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
"""Tests for doctor command."""
|
|
|
|
from browser_use.skill_cli.commands import doctor
|
|
|
|
|
|
async def test_doctor_handle_returns_valid_structure():
|
|
"""Test that doctor.handle() returns a valid result structure."""
|
|
result = await doctor.handle()
|
|
|
|
# Verify structure
|
|
assert 'status' in result
|
|
assert result['status'] in ('healthy', 'issues_found')
|
|
assert 'checks' in result
|
|
assert 'summary' in result
|
|
|
|
# Verify all expected checks are present
|
|
expected_checks = ['package', 'browser', 'network', 'cloudflared', 'profile_use']
|
|
for check in expected_checks:
|
|
assert check in result['checks']
|
|
assert 'status' in result['checks'][check]
|
|
assert 'message' in result['checks'][check]
|
|
|
|
|
|
def test_check_package_installed():
|
|
"""Test _check_package returns ok when browser-use is installed."""
|
|
# browser-use is always installed in the test environment
|
|
result = doctor._check_package()
|
|
assert result['status'] == 'ok'
|
|
assert 'browser-use' in result['message']
|
|
|
|
|
|
def test_check_browser_returns_valid_structure():
|
|
"""Test _check_browser returns a valid result."""
|
|
result = doctor._check_browser()
|
|
assert 'status' in result
|
|
assert result['status'] in ('ok', 'warning')
|
|
assert 'message' in result
|
|
|
|
|
|
async def test_check_network_returns_valid_structure():
|
|
"""Test _check_network returns a valid result structure."""
|
|
result = await doctor._check_network()
|
|
|
|
assert 'status' in result
|
|
assert result['status'] in ('ok', 'warning')
|
|
assert 'message' in result
|
|
|
|
|
|
def test_check_cloudflared_returns_valid_structure():
|
|
"""Test _check_cloudflared returns a valid result."""
|
|
result = doctor._check_cloudflared()
|
|
assert 'status' in result
|
|
assert result['status'] in ('ok', 'missing')
|
|
assert 'message' in result
|
|
|
|
|
|
def test_check_profile_use_returns_valid_structure():
|
|
"""Test _check_profile_use returns a valid result."""
|
|
result = doctor._check_profile_use()
|
|
assert 'status' in result
|
|
assert result['status'] in ('ok', 'missing')
|
|
assert 'message' in result
|
|
|
|
|
|
def test_summarize_checks_all_ok():
|
|
"""Test _summarize_checks when all checks pass."""
|
|
checks = {
|
|
'check1': {'status': 'ok'},
|
|
'check2': {'status': 'ok'},
|
|
'check3': {'status': 'ok'},
|
|
}
|
|
summary = doctor._summarize_checks(checks)
|
|
assert '3/3' in summary
|
|
|
|
|
|
def test_summarize_checks_mixed():
|
|
"""Test _summarize_checks with mixed results."""
|
|
checks = {
|
|
'check1': {'status': 'ok'},
|
|
'check2': {'status': 'warning'},
|
|
'check3': {'status': 'missing'},
|
|
}
|
|
summary = doctor._summarize_checks(checks)
|
|
assert '1/3' in summary
|
|
assert '1 warning' in summary
|
|
assert '1 missing' in summary
|
|
|
|
|
|
def test_summarize_checks_with_errors():
|
|
"""Test _summarize_checks with errors."""
|
|
checks = {
|
|
'check1': {'status': 'ok'},
|
|
'check2': {'status': 'error'},
|
|
}
|
|
summary = doctor._summarize_checks(checks)
|
|
assert '1/2' in summary
|
|
assert '1 error' in summary
|