andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 16:18:30 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Potential fix for code scanning alert no. 23: Incomplete URL substring sanitization

Browse Source 
Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>
This commit is contained in:
Rip&Tear
2025-11-08 21:34:00 +08:00
committed by GitHub
parent 0f1c173d02
commit c2d4073415
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lib/crewai-tools/tests/tools/stagehand_tool_test.py
View File

@@ -2,7 +2,7 @@ import sys
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
import pytest import pytest
from urllib.parse import urlparse
# Create mock classes that will be used by our fixture # Create mock classes that will be used by our fixture
class MockStagehandModule: class MockStagehandModule:
@@ -171,8 +171,13 @@ def test_navigate_command(mock_run, stagehand_tool):
) )
# Assertions # Assertions
assert "https://example.com" in result # Extract URL from result string and check its host
# Example result: "Successfully navigated to https://example.com"
import re
url_match = re.search(r'https?://[^\s]+', result)
assert url_match is not None
parsed = urlparse(url_match.group(0))
assert parsed.hostname == "example.com"
@patch( @patch(
"crewai_tools.tools.stagehand_tool.stagehand_tool.StagehandTool._run", autospec=True "crewai_tools.tools.stagehand_tool.stagehand_tool.StagehandTool._run", autospec=True