dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-22 05:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Mocked the aider.scraper.Scraper.scrape method in the test_cmd_web_imports_playwright test.

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-07-16 11:14:29 +01:00
parent d41f07b024
commit d5a258356b
1 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/scrape/test_scrape.py
View file

@ -1,7 +1,5 @@
import unittest import unittest
from unittest.mock import patch from unittest.mock import patch, MagicMock
import sys
import io
from aider.commands import Commands from aider.commands import Commands
from aider.io import InputOutput from aider.io import InputOutput
@ -12,9 +10,19 @@ class TestScrape(unittest.TestCase):
self.io = InputOutput() self.io = InputOutput()
self.commands = Commands(self.io, None) self.commands = Commands(self.io, None)
def test_cmd_web_imports_playwright(self): @patch('aider.scraper.Scraper.scrape')
def test_cmd_web_imports_playwright(self, mock_scrape):
# Mock the scrape method
mock_scrape.return_value = "Mocked webpage content"
# Run the cmd_web command # Run the cmd_web command
self.commands.cmd_web("https://example.com") result = self.commands.cmd_web("https://example.com")
# Assert that the scrape method was called with the correct URL
mock_scrape.assert_called_once_with("https://example.com")
# Assert that the result contains the mocked content
self.assertIn("Mocked webpage content", result)
# Try to import playwright # Try to import playwright
try: try: