From 92442955bc692970f33b8d17504dd06f7cf49539 Mon Sep 17 00:00:00 2001 From: gfyoung Date: Thu, 6 Sep 2018 21:12:42 -0700 Subject: [PATCH] Minor house-cleaning * No parentheses single-line string * Remove builtins function, which was really for Python 2.x compatibility --- testUpdateHostsFile.py | 20 ++++++++------------ updateHostsFile.py | 4 ++-- 2 files changed, 10 insertions(+), 14 deletions(-) diff --git a/testUpdateHostsFile.py b/testUpdateHostsFile.py index e8b678d5b..171cb21ba 100644 --- a/testUpdateHostsFile.py +++ b/testUpdateHostsFile.py @@ -70,10 +70,6 @@ class BaseMockDir(Base): def tearDown(self): shutil.rmtree(self.test_dir) - - -def builtins(): - return "builtins" # End Test Helper Objects @@ -168,7 +164,7 @@ class TestPromptForUpdate(BaseStdout, BaseMockDir): contents = f.read() self.assertEqual(contents, "") - @mock.patch(builtins() + ".open") + @mock.patch("builtins.open") def test_no_freshen_fail_new_file(self, mock_open): for exc in (IOError, OSError): mock_open.side_effect = exc("failed open") @@ -554,7 +550,7 @@ class TestUpdateSourcesData(Base): @mock.patch("updateHostsFile.recursive_glob", return_value=[]) @mock.patch("updateHostsFile.path_join_robust", return_value="dirpath") - @mock.patch(builtins() + ".open", return_value=mock.Mock()) + @mock.patch("builtins.open", return_value=mock.Mock()) def test_no_update(self, mock_open, mock_join_robust, _): extensions = [] sources_data = [{"source": "source1.txt"}, {"source": "source2.txt"}] @@ -576,7 +572,7 @@ class TestUpdateSourcesData(Base): @mock.patch("updateHostsFile.recursive_glob", side_effect=[[], ["update1.txt", "update2.txt"]]) @mock.patch("json.load", return_value={"mock_source": "mock_source.ext"}) - @mock.patch(builtins() + ".open", return_value=mock.Mock()) + @mock.patch("builtins.open", return_value=mock.Mock()) @mock.patch("updateHostsFile.path_join_robust", return_value="dirpath") def test_update_only_extensions(self, mock_join_robust, *_): extensions = [".json"] @@ -594,7 +590,7 @@ class TestUpdateSourcesData(Base): {"mock_source": "mock_source2.txt"}, {"mock_source": "mock_source3.txt"}, {"mock_source": "mock_source4.txt"}]) - @mock.patch(builtins() + ".open", return_value=mock.Mock()) + @mock.patch("builtins.open", return_value=mock.Mock()) @mock.patch("updateHostsFile.path_join_robust", return_value="dirpath") def test_update_both_pathways(self, mock_join_robust, *_): extensions = [".json"] @@ -617,13 +613,13 @@ class TestUpdateAllSources(BaseStdout): self.source_data_filename = "data.json" self.host_filename = "hosts.txt" - @mock.patch(builtins() + ".open") + @mock.patch("builtins.open") @mock.patch("updateHostsFile.recursive_glob", return_value=[]) def test_no_sources(self, _, mock_open): update_all_sources(self.source_data_filename, self.host_filename) mock_open.assert_not_called() - @mock.patch(builtins() + ".open", return_value=mock.Mock()) + @mock.patch("builtins.open", return_value=mock.Mock()) @mock.patch("json.load", return_value={"url": "example.com"}) @mock.patch("updateHostsFile.recursive_glob", return_value=["foo"]) @mock.patch("updateHostsFile.write_data", return_value=0) @@ -638,7 +634,7 @@ class TestUpdateAllSources(BaseStdout): self.assertIn(expected, output) - @mock.patch(builtins() + ".open", return_value=mock.Mock()) + @mock.patch("builtins.open", return_value=mock.Mock()) @mock.patch("json.load", return_value={"url": "example.com"}) @mock.patch("updateHostsFile.recursive_glob", return_value=["foo"]) @mock.patch("updateHostsFile.write_data", return_value=0) @@ -655,7 +651,7 @@ class TestUpdateAllSources(BaseStdout): for expected in expecteds: self.assertIn(expected, output) - @mock.patch(builtins() + ".open", return_value=mock.Mock()) + @mock.patch("builtins.open", return_value=mock.Mock()) @mock.patch("json.load", side_effect=[{"url": "example.com"}, {"url": "example2.com"}]) @mock.patch("updateHostsFile.recursive_glob", return_value=["foo", "bar"]) diff --git a/updateHostsFile.py b/updateHostsFile.py index 307ad2158..850582f84 100644 --- a/updateHostsFile.py +++ b/updateHostsFile.py @@ -354,7 +354,7 @@ def prompt_for_move(final_file, **move_params): elif move_params["auto"] or skip_static_hosts: move_file = False else: - prompt = ("Do you want to replace your existing hosts file with the newly generated file?") + prompt = "Do you want to replace your existing hosts file with the newly generated file?" move_file = query_yes_no(prompt) if move_file: @@ -428,7 +428,7 @@ def gather_custom_exclusions(exclusion_pattern, exclusion_regexes): # We continue running this while-loop until the user # says that they have no more domains to exclude. while True: - domain_prompt = ("Enter the domain you want to exclude (e.g. facebook.com): ") + domain_prompt = "Enter the domain you want to exclude (e.g. facebook.com): " user_domain = input(domain_prompt) if is_valid_domain_format(user_domain):