def test__get_page_soup_returns_bs4(self, mock_requests): mock_requests.get.return_value.content = b'<h1>Test</h1>' scrape = PageScraper('www.google.com') scrape._get_page_contents() soup = scrape._get_page_soup() self.assertEqual(soup.get_text(), 'Test')
def test__get_page_contents_returns_content(self, mock_requests): mock_requests.get.return_value.content = b'page content' scraper = PageScraper('www.google.com') self.assertEqual(scraper._get_page_contents(), 'page content')