Remove cloud_etc_hosts module when DHCP is disabled

Author: Cristi1324

coriolis/osmorphing/base.py

@@ -433,6 +433,16 @@ def _ensure_cloud_init_not_disabled(self):
     def _reset_cloud_init_run(self):
         self._exec_cmd_chroot("cloud-init clean --logs")
 
+    def _get_cloud_init_modules(self):
+        cloud_cfg_path = 'etc/cloud/cloud.cfg'
+        if not self._test_path(cloud_cfg_path):
+            return
+        cloud_cfg_content = self._read_file_sudo(cloud_cfg_path)
+        cloud_cfg = yaml.load(cloud_cfg_content, Loader=yaml.SafeLoader)
+        modules = cloud_cfg.get('cloud_init_modules', [])
+
+        return modules
+
     def _get_default_cloud_user(self):
         cloud_cfg_path = 'etc/cloud/cloud.cfg'
         if not self._test_path(cloud_cfg_path):
@@ -472,6 +482,10 @@ def _configure_cloud_init(self):
         if not self._osmorphing_parameters.get('set_dhcp', True):
             disabled_network_config = {"network": {"config": "disabled"}}
             cloud_cfg_mods.update(disabled_network_config)
+            modules = self._get_cloud_init_modules()
+            if 'update_etc_hosts' in modules:
+                modules = [m for m in modules if m != 'update_etc_hosts']
+                cloud_cfg_mods.update({'cloud_init_modules': modules})
 
         self._write_cloud_init_mods_config(cloud_cfg_mods)
 

coriolis/tests/osmorphing/test_base.py

@@ -661,6 +661,26 @@ def test__reset_cloud_init_run(self, mock__exec_cmd_chroot):
         mock__exec_cmd_chroot.assert_called_once_with(
             "cloud-init clean --logs")
 
+    @ddt.data(
+        (False, None, None),
+        (True, "other_modules:\n  - not_update_etc_hosts\n"
+         "cloud_init_modules:\n  - set_hostname\n  - update_etc_hosts\n",
+         ["set_hostname", "update_etc_hosts"]),
+        (True, "other_modules:\n  - update_etc_hosts\n", [])
+    )
+    @ddt.unpack
+    @mock.patch.object(base.BaseLinuxOSMorphingTools, "_read_file_sudo")
+    @mock.patch.object(base.BaseLinuxOSMorphingTools, "_test_path")
+    def test__get_cloud_init_modules(
+            self, test_path_result, file_content,
+            expected_result, mock__test_path, mock__read_file_sudo):
+        mock__test_path.return_value = test_path_result
+        mock__read_file_sudo.return_value = file_content
+
+        result = self.os_morphing_tools._get_cloud_init_modules()
+
+        self.assertEqual(result, expected_result)
+
     @ddt.data(
         (False, None, base.DEFAULT_CLOUD_USER),
         (True, "system_info:\n  default_user:\n    name: mock_user\n",
@@ -713,13 +733,15 @@ def test__create_cloudinit_user_already_exists(
             ["vim"],
             {},
             False,
+            [],
             None,
             False
         ),
         (
             ["cloud-init"],
             {"retain_user_credentials": True, "set_dhcp": False},
             False,
+            ["set_hostname"],
             {
                 "disable_root": False,
                 "ssh_pwauth": True,
@@ -732,14 +754,28 @@ def test__create_cloudinit_user_already_exists(
             ["cloud-init", "vim"],
             {"retain_user_credentials": False, "set_dhcp": True},
             True,
+            [],
             {},
             False
         ),
+        (
+            ["cloud-init", "vim"],
+            {"retain_user_credentials": False, "set_dhcp": False},
+            True,
+            ["update_etc_hosts", "set_hostname", "write_files"],
+            {
+                "network": {"config": "disabled"},
+                "cloud_init_modules": ["set_hostname", "write_files"]
+            },
+            False
+        ),
     )
     @ddt.unpack
     @mock.patch.object(base.BaseLinuxOSMorphingTools,
                        '_enable_systemd_service')
     @mock.patch.object(base.BaseLinuxOSMorphingTools, '_has_systemd_chroot')
+    @mock.patch.object(base.BaseLinuxOSMorphingTools,
+                       '_get_cloud_init_modules')
     @mock.patch.object(base.BaseLinuxOSMorphingTools,
                        '_write_cloud_init_mods_config')
     @mock.patch.object(base.BaseLinuxOSMorphingTools, '_create_cloudinit_user')
@@ -751,15 +787,18 @@ def test__create_cloudinit_user_already_exists(
     @mock.patch.object(base.BaseLinuxOSMorphingTools, 'get_packages')
     def test__configure_cloud_init(
             self, returned_packages, osmorphing_params, creates_cloudinit_user,
+            cloud_init_modules,
             expected_result, has_systemd_chroot, mock_get_packages,
             mock__disable_installer_cloud_config,
             mock__ensure_cloud_init_not_disabled, mock__reset_cloud_init_run,
             mock__create_cloudinit_user, mock__write_cloud_init_mods_config,
+            mock__get_cloud_init_modules,
             mock__has_systemd_chroot, mock__enable_systemd_service
     ):
         mock_get_packages.return_value = returned_packages
         self.os_morphing_tools._osmorphing_parameters = osmorphing_params
         mock__has_systemd_chroot.return_value = has_systemd_chroot
+        mock__get_cloud_init_modules.return_value = cloud_init_modules
 
         self.os_morphing_tools._configure_cloud_init()