From 513e8081eb78a63ec67c17a9c4532211321f75dc Mon Sep 17 00:00:00 2001 From: Yilei Pan Date: Fri, 15 Dec 2023 09:58:01 +0100 Subject: [PATCH] fix: repair mock errors in unit tests --- tests/tests_crawler/test_middlewares.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/tests_crawler/test_middlewares.py b/tests/tests_crawler/test_middlewares.py index 3b86527..84d3f5d 100644 --- a/tests/tests_crawler/test_middlewares.py +++ b/tests/tests_crawler/test_middlewares.py @@ -87,8 +87,10 @@ def setUp(self): ) def test_process_response(self, mock_mkdir, mock_write_text): mock_response = self.mock_response() + mock_spider = self.mock_spider() + mock_spider.allowed_url = "/test" result = self.middleware.process_response( - self.mock_request(), mock_response, self.mock_spider() + self.mock_request(), mock_response, mock_spider ) mock_mkdir.assert_called_once() mock_write_text.assert_called_once_with(self.mock_response().text) @@ -101,10 +103,12 @@ def test_process_response(self, mock_mkdir, mock_write_text): ) def test_robots_txt_skip(self, mock_mkdir, mock_write_text): mock_response = self.mock_response(url="http://example.com/robots.txt") + mock_spider = self.mock_spider() + mock_spider.allowed_url = "/robots.txt" result = self.middleware.process_response( self.mock_request(url="http://example.com/robots.txt"), mock_response, - self.mock_spider(), + mock_spider, ) mock_mkdir.assert_not_called() mock_write_text.assert_not_called()