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

finished scrape tests

Browse source
This commit is contained in:
Paul Gauthier 2024-07-16 11:27:27 +01:00
parent d5a258356b
commit 96b1e1d04f
2 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/scrape/test_scrape.py
View file

@ -1,5 +1,5 @@
import unittest
from unittest.mock import patch, MagicMock
from unittest.mock import patch
from aider.commands import Commands
from aider.io import InputOutput
@ -7,10 +7,10 @@ from aider.io import InputOutput
class TestScrape(unittest.TestCase):
def setUp(self):
self.io = InputOutput()
self.io = InputOutput(yes=True)
self.commands = Commands(self.io, None)
@patch('aider.scraper.Scraper.scrape')
@patch("aider.scrape.Scraper.scrape")
def test_cmd_web_imports_playwright(self, mock_scrape):
# Mock the scrape method
mock_scrape.return_value = "Mocked webpage content"
@ -26,15 +26,15 @@ class TestScrape(unittest.TestCase):
# Try to import playwright
try:
import playwright
import playwright # noqa: F401
playwright_imported = True
except ImportError:
playwright_imported = False
# Assert that playwright was successfully imported
self.assertTrue(
playwright_imported,
"Playwright should be importable after running cmd_web"
playwright_imported, "Playwright should be importable after running cmd_web"
)