def test_functional_empty(self, mock_open):
with io.StringIO() as buffer, redirect_stdout(buffer):
mock_open.return_value = self.empty_html
ipfromwebpage.main(self.test_url)
out = buffer.getvalue()
self.assertEqual(out, "No ips found when scraping {}\n".format(self.test_url))
def test_functional_list(self, mock_open):
with io.StringIO() as buffer, redirect_stdout(buffer):
mock_open.return_value = self.test_html_doc
ipfromwebpage.main(self.test_url)
out = buffer.getvalue()
self.assertEqual(out, self.expected_print)
def test_functional_list(self, mock_open):
with io.StringIO() as buffer, redirect_stdout(buffer):
mock_open.return_value = self.test_html_doc
ipfromwebpage.main(self.test_url)
out = buffer.getvalue()
self.assertEqual(out, self.expected_print)
def test_functional_empty(self, mock_open):
with io.StringIO() as buffer, redirect_stdout(buffer):
mock_open.return_value = self.empty_html
test_url = 'http://test_html_empty.html'
ipfromwebpage.main(test_url)
out = buffer.getvalue()
expected = "================\nIPv4 addresses:\nNo addresses found when scraping {0}\n================\nIPv6 addresses:\nNo addresses found when scraping {0}\n".format(
test_url)
self.assertEqual(expected, out)
