mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
Potential fix for code scanning alert no. 23: Incomplete URL substring sanitization
Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>
This commit is contained in:
@@ -2,7 +2,7 @@ import sys
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
from urllib.parse import urlparse
|
||||
|
||||
# Create mock classes that will be used by our fixture
|
||||
class MockStagehandModule:
|
||||
@@ -171,8 +171,13 @@ def test_navigate_command(mock_run, stagehand_tool):
|
||||
)
|
||||
|
||||
# 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(
|
||||
"crewai_tools.tools.stagehand_tool.stagehand_tool.StagehandTool._run", autospec=True
|
||||
|
||||
Reference in New Issue
Block a user