diff --git a/testUpdateHostsFile.py b/testUpdateHostsFile.py index 3d80691f3..3f09e2308 100644 --- a/testUpdateHostsFile.py +++ b/testUpdateHostsFile.py @@ -1642,6 +1642,20 @@ class GetFileByUrl(BaseStdout): self.assertEqual(expected, actual) + def test_connect_unknown_domain(self): + test_url = "http://doesnotexist.google.com" + return_value = get_file_by_url(test_url) + self.assertIsNone(return_value) + printed_output = sys.stdout.getvalue() + self.assertEqual(printed_output, "Error retrieving data from {}\n".format(test_url)) + + def test_invalid_url(self): + test_url = "http://fe80::5054:ff:fe5a:fc0" + return_value = get_file_by_url(test_url) + self.assertIsNone(return_value) + printed_output = sys.stdout.getvalue() + self.assertEqual(printed_output, "Error retrieving data from {}\n".format(test_url)) + class TestWriteData(Base): def test_write_basic(self):