Method test_freshen_update

testUpdateHostsFile.py:313–340

(self, _)

    @mock.patch("updateHostsFile.query_yes_no", return_value=True)
    def test_freshen_update(self, _):
        hostsfile = os.path.join(self.test_dir, "hosts")
        hosts_data = "This data should not be overwritten"

        with self.mock_property("updateHostsFile.BASEDIR_PATH"):
            updateHostsFile.BASEDIR_PATH = self.test_dir

            with open(hostsfile, "w") as f:
                f.write(hosts_data)

            dir_count = self.dir_count

            for updateauto in (False, True):
                updatesources = prompt_for_update(
                    freshen=True, updateauto=updateauto
                )
                self.assertTrue(updatesources)

                output = sys.stdout.getvalue()
                self.assertEqual(output, "")

                sys.stdout = StringIO()

                self.assertEqual(self.dir_count, dir_count)

                with open(hostsfile, "r") as f:
                    contents = f.read()
                    self.assertEqual(contents, hosts_data)

    def tearDown(self):
        BaseStdout.tearDown(self)