From b350c685405353685f00b98f62792327ab03f2f2 Mon Sep 17 00:00:00 2001
From: Alexander Cecile <35971201+AlexanderCecile@users.noreply.github.com>
Date: Wed, 2 Sep 2020 20:25:01 -0400
Subject: [PATCH] Improved tests for get_file_by_url

---
 testUpdateHostsFile.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/testUpdateHostsFile.py b/testUpdateHostsFile.py
index 7ca920280..e0199503c 100644
--- a/testUpdateHostsFile.py
+++ b/testUpdateHostsFile.py
@@ -1572,15 +1572,17 @@ 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)
+        test_url = "http://doesnotexist.google.com"  # leads to exception: ConnectionError
+        with mock.patch("requests.get", side_effect=requests.exceptions.ConnectionError):
+            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)
+        test_url = "http://fe80::5054:ff:fe5a:fc0"  # leads to exception: InvalidURL
+        with mock.patch("requests.get", side_effect=requests.exceptions.ConnectionError):
+            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))