From a03506046c53d230078d63ef9e8fddaa8ed505c6 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Sun, 28 Jul 2024 16:37:47 -0300 Subject: [PATCH] Add test to fetch self-signed SSL site with and without verification --- tests/scrape/test_scrape.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tests/scrape/test_scrape.py b/tests/scrape/test_scrape.py index 992b830d3..d3bde4805 100644 --- a/tests/scrape/test_scrape.py +++ b/tests/scrape/test_scrape.py @@ -8,6 +8,19 @@ from aider.scrape import Scraper class TestScrape(unittest.TestCase): + def test_scrape_self_signed_ssl(self): + # Test with SSL verification + scraper_verify = Scraper(print_error=MagicMock(), playwright_available=True, verify_ssl=True) + result_verify = scraper_verify.scrape("https://self-signed.badssl.com") + self.assertIsNone(result_verify) + scraper_verify.print_error.assert_called() + + # Test without SSL verification + scraper_no_verify = Scraper(print_error=MagicMock(), playwright_available=True, verify_ssl=False) + result_no_verify = scraper_no_verify.scrape("https://self-signed.badssl.com") + self.assertIsNotNone(result_no_verify) + self.assertIn("self-signed.badssl.com", result_no_verify) + scraper_no_verify.print_error.assert_not_called() def setUp(self): self.io = InputOutput(yes=True) self.commands = Commands(self.io, None)