c4abf63f38
Some checks failed
Release / semantic-release (push) Successful in 59s
tests / Unit tests (Linux, Python 3.11) (push) Successful in 13m45s
Release / build-linux (push) Failing after 7m47s
Release / build-windows (push) Has been cancelled
Release / build-macos (arm64, macos-latest) (push) Has been cancelled
Release / build-macos (x64, macos-15-intel) (push) Has been cancelled
Release / release-main (push) Has been cancelled
Release / release-develop (push) Has been cancelled
Transform isair/jarvis into a Discord-controlled voice assistant running on the Ubuntu VNC desktop, keeping the mature ~39k-line Python brain intact. - bot/ (Node + bun, discord.js): /자비스 slash commands (ephemeral), voice channel join + voice receive/playback, pluggable VNC screen broadcast (selfbot live / noVNC / screenshot) - bridge/ (Python, Flask): wraps jarvis STT + run_reply_engine + Piper TTS behind a thin localhost HTTP API - .env.example, scripts/ (start_bridge/start_bot/dev), README rewrite, docs/language-comparison.md and docs/vnc-xfce-setup.md Language decision: hybrid (Python brain + Node/bun Discord layer) because Discord blocks bot video; native screen broadcast only works via a Node selfbot library.
157 lines
5.9 KiB
Python
157 lines
5.9 KiB
Python
"""Tests for fetch web page tool."""
|
|
|
|
import pytest
|
|
from unittest.mock import Mock, patch
|
|
import requests
|
|
|
|
from src.jarvis.tools.builtin.fetch_web_page import FetchWebPageTool
|
|
from src.jarvis.tools.base import ToolContext
|
|
from src.jarvis.tools.types import ToolExecutionResult
|
|
|
|
|
|
def _make_response_mock(**attrs) -> Mock:
|
|
"""Build a Mock that doubles as both the requests response and a context
|
|
manager (the production code uses ``with requests.get(...) as resp`` so
|
|
the connection is released deterministically).
|
|
"""
|
|
resp = Mock(**attrs)
|
|
resp.__enter__ = Mock(return_value=resp)
|
|
resp.__exit__ = Mock(return_value=False)
|
|
return resp
|
|
|
|
|
|
class TestFetchWebPageTool:
|
|
"""Test fetch web page tool functionality."""
|
|
|
|
def setup_method(self):
|
|
"""Set up test fixtures."""
|
|
self.tool = FetchWebPageTool()
|
|
self.context = Mock(spec=ToolContext)
|
|
self.context.user_print = Mock()
|
|
|
|
def test_tool_properties(self):
|
|
"""Test tool metadata properties."""
|
|
assert self.tool.name == "fetchWebPage"
|
|
assert "fetch" in self.tool.description.lower()
|
|
assert self.tool.inputSchema["type"] == "object"
|
|
assert "url" in self.tool.inputSchema["required"]
|
|
|
|
def test_run_no_args(self):
|
|
"""Test fetch web page with no arguments."""
|
|
result = self.tool.run(None, self.context)
|
|
|
|
assert isinstance(result, ToolExecutionResult)
|
|
assert result.success is False
|
|
assert "url" in result.reply_text.lower()
|
|
|
|
def test_run_empty_url(self):
|
|
"""Test fetch web page with empty URL."""
|
|
args = {"url": ""}
|
|
result = self.tool.run(args, self.context)
|
|
|
|
assert isinstance(result, ToolExecutionResult)
|
|
assert result.success is False
|
|
assert "url" in result.reply_text.lower()
|
|
|
|
@patch('requests.get')
|
|
def test_run_success(self, mock_get):
|
|
"""Test successful web page fetch."""
|
|
mock_response = _make_response_mock(
|
|
status_code=200,
|
|
text='<html><head><title>Test</title></head><body><p>Content</p></body></html>',
|
|
content=b'<html><head><title>Test</title></head><body><p>Content</p></body></html>',
|
|
headers={'content-type': 'text/html'},
|
|
raise_for_status=Mock(),
|
|
)
|
|
mock_get.return_value = mock_response
|
|
|
|
args = {"url": "https://example.com"}
|
|
result = self.tool.run(args, self.context)
|
|
|
|
assert isinstance(result, ToolExecutionResult)
|
|
assert result.success is True
|
|
assert "example.com" in result.reply_text
|
|
self.context.user_print.assert_called()
|
|
|
|
@patch('requests.get')
|
|
def test_run_success_without_beautifulsoup(self, mock_get):
|
|
"""Test successful web page fetch without BeautifulSoup."""
|
|
mock_response = _make_response_mock(
|
|
status_code=200,
|
|
text='<html><body>Raw content</body></html>',
|
|
content=b'<html><body>Raw content</body></html>',
|
|
headers={'content-type': 'text/html'},
|
|
raise_for_status=Mock(),
|
|
)
|
|
mock_get.return_value = mock_response
|
|
|
|
with patch('builtins.__import__', side_effect=ImportError):
|
|
args = {"url": "https://example.com"}
|
|
result = self.tool.run(args, self.context)
|
|
|
|
assert isinstance(result, ToolExecutionResult)
|
|
assert result.success is True
|
|
assert "Raw Content" in result.reply_text
|
|
|
|
@patch('requests.get')
|
|
def test_run_http_error(self, mock_get):
|
|
"""Test fetch web page with HTTP error."""
|
|
mock_response = _make_response_mock(status_code=404)
|
|
mock_response.raise_for_status.side_effect = requests.exceptions.HTTPError("404 Not Found")
|
|
mock_get.return_value = mock_response
|
|
|
|
args = {"url": "https://example.com/notfound"}
|
|
result = self.tool.run(args, self.context)
|
|
|
|
assert isinstance(result, ToolExecutionResult)
|
|
assert result.success is False
|
|
assert "Failed to fetch page" in result.reply_text
|
|
|
|
@patch('requests.get')
|
|
def test_run_request_error(self, mock_get):
|
|
"""Test fetch web page with network error."""
|
|
mock_get.side_effect = requests.exceptions.RequestException("Network error")
|
|
|
|
args = {"url": "https://example.com"}
|
|
result = self.tool.run(args, self.context)
|
|
|
|
assert isinstance(result, ToolExecutionResult)
|
|
assert result.success is False
|
|
assert "Failed to fetch page" in result.reply_text
|
|
|
|
def test_run_invalid_url(self):
|
|
"""Test fetch web page with invalid URL."""
|
|
args = {"url": "not-a-url"}
|
|
result = self.tool.run(args, self.context)
|
|
assert isinstance(result, ToolExecutionResult)
|
|
assert result.success is False
|
|
assert "failed" in result.reply_text.lower() or "error" in result.reply_text.lower()
|
|
|
|
@patch('requests.get')
|
|
def test_run_with_links_extraction(self, mock_get):
|
|
"""Test fetch web page including link extraction when include_links=True."""
|
|
html = (
|
|
'<html><head><title>Links Page</title></head>'
|
|
'<body><p>Intro</p>'
|
|
'<a href="/relative">Relative Link</a>'
|
|
'<a href="https://absolute.test/page">Absolute Link</a>'
|
|
'<a href="mailto:test@example.com">Mail</a>'
|
|
'</body></html>'
|
|
)
|
|
mock_response = _make_response_mock(
|
|
status_code=200,
|
|
text=html,
|
|
content=html.encode(),
|
|
raise_for_status=Mock(),
|
|
)
|
|
mock_get.return_value = mock_response
|
|
|
|
args = {"url": "https://example.com", "include_links": True}
|
|
result = self.tool.run(args, self.context)
|
|
assert result.success is True
|
|
assert isinstance(result, ToolExecutionResult)
|
|
assert "Links found on page" in result.reply_text
|
|
# relative link should be resolved to absolute
|
|
assert "https://example.com/relative" in result.reply_text
|
|
assert "absolute.test" in result.reply_text
|