From d600f47e91b904243263358324c413c4f7e5cf50 Mon Sep 17 00:00:00 2001 From: Anh Vo Date: Wed, 22 Jul 2020 12:51:01 -0400 Subject: azure: disable bouncing hostname when setting hostname fails (#494) DataSourceAzure: Gracefully handle the case of set hostname failure during provisioning --- tests/unittests/test_datasource/test_azure.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'tests/unittests') diff --git a/tests/unittests/test_datasource/test_azure.py b/tests/unittests/test_datasource/test_azure.py index a99cbd41..96bcc7a2 100644 --- a/tests/unittests/test_datasource/test_azure.py +++ b/tests/unittests/test_datasource/test_azure.py @@ -1524,6 +1524,17 @@ class TestAzureBounce(CiTestCase): self.assertEqual(0, self.set_hostname.call_count) + @mock.patch(MOCKPATH + 'perform_hostname_bounce') + def test_set_hostname_failed_disable_bounce( + self, perform_hostname_bounce): + cfg = {'set_hostname': True, 'hostname_bounce': {'policy': 'force'}} + self.get_hostname.return_value = "old-hostname" + self.set_hostname.side_effect = Exception + data = self.get_ovf_env_with_dscfg('some-hostname', cfg) + self._get_ds(data).get_data() + + self.assertEqual(0, perform_hostname_bounce.call_count) + class TestLoadAzureDsDir(CiTestCase): """Tests for load_azure_ds_dir.""" -- cgit v1.2.3