1
diff --git a/.gitignore b/.gitignore
2
index b0500a6..75565ed 100644
3
--- a/.gitignore
4
+++ b/.gitignore
5
@@ -10,3 +10,4 @@ parts
6
10
prime
10
prime
7
11
stage
11
stage
8
12
*.snap
12
*.snap
9
13
*.cover
10
diff --git a/cloudinit/sources/DataSourceAzure.py b/cloudinit/sources/DataSourceAzure.py
11
index d1d0975..0bcbc99 100644
12
--- a/cloudinit/sources/DataSourceAzure.py
13
+++ b/cloudinit/sources/DataSourceAzure.py
14
@@ -11,6 +11,8 @@ from functools import partial
15
11
import os
11
import os
16
12
import os.path
12
import os.path
17
13
import re
13
import re
18
14
from time import time
19
15
from uuid import uuid4
20
14
from xml.dom import minidom
16
from xml.dom import minidom
21
15
import xml.etree.ElementTree as ET
17
import xml.etree.ElementTree as ET
22
16
18
23
@@ -18,6 +20,7 @@ from cloudinit import log as logging
24
18
from cloudinit import net
20
from cloudinit import net
25
19
from cloudinit import sources
21
from cloudinit import sources
26
20
from cloudinit.sources.helpers.azure import get_metadata_from_fabric
22
from cloudinit.sources.helpers.azure import get_metadata_from_fabric
27
23
from cloudinit.url_helper import readurl, wait_for_url, UrlError
28
21
from cloudinit import util
24
from cloudinit import util
29
22
25
30
23
LOG = logging.getLogger(__name__)
26
LOG = logging.getLogger(__name__)
31
@@ -44,6 +47,8 @@ LEASE_FILE = '/var/lib/dhcp/dhclient.eth0.leases'
32
44
DEFAULT_FS = 'ext4'
47
DEFAULT_FS = 'ext4'
33
45
# DMI chassis-asset-tag is set static for all azure instances
48
# DMI chassis-asset-tag is set static for all azure instances
34
46
AZURE_CHASSIS_ASSET_TAG = '7783-7084-3265-9085-8269-3286-77'
49
AZURE_CHASSIS_ASSET_TAG = '7783-7084-3265-9085-8269-3286-77'
35
50
REPROVISION_MARKER_FILE = "/var/lib/cloud/data/poll_imds"
36
51
IMDS_URL = "http://169.254.169.254/metadata/reprovisiondata"
37
47
52
38
48
53
39
49
def find_storvscid_from_sysctl_pnpinfo(sysctl_out, deviceid):
54
def find_storvscid_from_sysctl_pnpinfo(sysctl_out, deviceid):
40
@@ -276,19 +281,20 @@ class DataSourceAzure(sources.DataSource):
41
276
281
42
277
        with temporary_hostname(azure_hostname, self.ds_cfg,
282
        with temporary_hostname(azure_hostname, self.ds_cfg,
43
278
                                hostname_command=hostname_command) \
283
                                hostname_command=hostname_command) \
46
279
                as previous_hostname:
284
                as previous_hn:
47
280
            if (previous_hostname is not None and
285
            if (previous_hn is not None and
48
281
                    util.is_true(self.ds_cfg.get('set_hostname'))):
286
                    util.is_true(self.ds_cfg.get('set_hostname'))):
49
282
                cfg = self.ds_cfg['hostname_bounce']
287
                cfg = self.ds_cfg['hostname_bounce']
50
283
288
51
284
                # "Bouncing" the network
289
                # "Bouncing" the network
52
285
                try:
290
                try:
56
286
                    perform_hostname_bounce(hostname=azure_hostname,
291
                    return perform_hostname_bounce(hostname=azure_hostname,
57
287
                                            cfg=cfg,
292
                                                   cfg=cfg,
58
288
                                            prev_hostname=previous_hostname)
293
                                                   prev_hostname=previous_hn)
59
289
                except Exception as e:
294
                except Exception as e:
60
290
                    LOG.warning("Failed publishing hostname: %s", e)
295
                    LOG.warning("Failed publishing hostname: %s", e)
61
291
                    util.logexc(LOG, "handling set_hostname failed")
296
                    util.logexc(LOG, "handling set_hostname failed")
62
297
        return False
63
292
298
64
293
    def get_metadata_from_agent(self):
299
    def get_metadata_from_agent(self):
65
294
        temp_hostname = self.metadata.get('local-hostname')
300
        temp_hostname = self.metadata.get('local-hostname')
66
@@ -345,15 +351,20 @@ class DataSourceAzure(sources.DataSource):
67
345
        ddir = self.ds_cfg['data_dir']
351
        ddir = self.ds_cfg['data_dir']
68
346
352
69
347
        candidates = [self.seed_dir]
353
        candidates = [self.seed_dir]
70
354
        if os.path.isfile(REPROVISION_MARKER_FILE):
71
355
            candidates.insert(0, "IMDS")
72
348
        candidates.extend(list_possible_azure_ds_devs())
356
        candidates.extend(list_possible_azure_ds_devs())
73
349
        if ddir:
357
        if ddir:
74
350
            candidates.append(ddir)
358
            candidates.append(ddir)
75
351
359
76
352
        found = None
360
        found = None
78
353
361
        reprovision = False
79
354
        for cdev in candidates:
362
        for cdev in candidates:
80
355
            try:
363
            try:
82
356
                if cdev.startswith("/dev/"):
364
                if cdev == "IMDS":
83
365
                    ret = None
84
366
                    reprovision = True
85
367
                elif cdev.startswith("/dev/"):
86
357
                    if util.is_FreeBSD():
368
                    if util.is_FreeBSD():
87
358
                        ret = util.mount_cb(cdev, load_azure_ds_dir,
369
                        ret = util.mount_cb(cdev, load_azure_ds_dir,
88
359
                                            mtype="udf", sync=False)
370
                                            mtype="udf", sync=False)
89
@@ -370,6 +381,8 @@ class DataSourceAzure(sources.DataSource):
90
370
                LOG.warning("%s was not mountable", cdev)
381
                LOG.warning("%s was not mountable", cdev)
91
371
                continue
382
                continue
92
372
383
93
384
            if reprovision or self._should_reprovision(ret):
94
385
                ret = self._reprovision()
95
373
            (md, self.userdata_raw, cfg, files) = ret
386
            (md, self.userdata_raw, cfg, files) = ret
96
374
            self.seed = cdev
387
            self.seed = cdev
97
375
            self.metadata = util.mergemanydict([md, DEFAULT_METADATA])
388
            self.metadata = util.mergemanydict([md, DEFAULT_METADATA])
98
@@ -428,6 +441,83 @@ class DataSourceAzure(sources.DataSource):
99
428
            LOG.debug("negotiating already done for %s",
441
            LOG.debug("negotiating already done for %s",
100
429
                      self.get_instance_id())
442
                      self.get_instance_id())
101
430
443
102
444
    def _poll_imds(self):
103
445
        """Poll IMDS for the new provisioning data until we get a valid
104
446
        response. Then returned the returned JSON object."""
105
447
        url = IMDS_URL + "?api-version=2017-04-02"
106
448
        headers = {"Metadata": "true"}
107
449
        LOG.debug("Start polling IMDS")
108
450
109
451
        def exception_cb(msg, exception):
110
452
            if isinstance(exception, UrlError) and exception.code == 404:
111
453
                return
112
454
            LOG.warning("Exception during polling. Will try DHCP.",
113
455
                        exc_info=True)
114
456
            # On certain versions of the networking stack, the link
115
457
            # state disconnect is not long enough to retrigger DHCP.
116
458
            # If we get an exception while trying to call IMDS, we
117
459
            # bounce the network to force DHCP to acquire the new IP.
118
460
            if not self.bounce_network_with_azure_hostname():
119
461
                # On ubuntu releases > xenial, we rely on systemd-networkd to
120
462
                # retrigger DHCP when the hostname changes as the ifup/down
121
463
                # commands are missing. In that case, we will simply change
122
464
                # the hostname. We can do this as the real hostname will still
123
465
                # be applied after receiving the ovf_env.xml.
124
466
                hn_cmd = self.ds_cfg['hostname_bounce']['hostname_command']
125
467
                set_hostname(str(uuid4()), hn_cmd)
126
468
127
469
        wait_for_url([url], max_wait=None, timeout=60, status_cb=LOG.info,
128
470
                     headers_cb=lambda url: headers, sleep_time=1,
129
471
                     exception_cb=exception_cb)
130
472
        return str(readurl(url, headers=headers))
131
473
132
474
    def _report_ready(self):
133
475
        """Tells the fabric provisioning has completed
134
476
           before we go into our polling loop."""
135
477
        try:
136
478
            get_metadata_from_fabric(self.dhclient_lease_file)
137
479
        except Exception as exc:
138
480
            LOG.warning(
139
481
                "Error communicating with Azure fabric; You may experience."
140
482
                "connectivity issues.", exc_info=True)
141
483
142
484
    def _should_reprovision(self, ret):
143
485
        """Whether or not we should poll IMDS for reprovisioning data.
144
486
        Also sets a marker file to poll IMDS.
145
487
146
488
        The marker file is used for the following scenario: the VM boots into
147
489
        this polling loop, which we expect to be proceeding infinitely until
148
490
        the VM is picked. If for whatever reason the platform moves us to a
149
491
        new host (for instance a hardware issue), we need to keep polling.
150
492
        However, since the VM reports ready to the Fabric, we will not attach
151
493
        the ISO, thus cloud-init needs to have a way of knowing that it should
152
494
        jump back into the polling loop in order to retrieve the ovf_env."""
153
495
        if not ret:
154
496
            return False
155
497
        (md, self.userdata_raw, cfg, files) = ret
156
498
        path = REPROVISION_MARKER_FILE
157
499
        if (cfg.get('PreprovisionedVm') is True or
158
500
                os.path.isfile(path)):
159
501
            if not os.path.isfile(path):
160
502
                LOG.info("Creating a marker file to poll imds")
161
503
                util.write_file(path, "%s: %s\n" % (os.getpid(), time()))
162
504
            return True
163
505
        return False
164
506
165
507
    def _reprovision(self):
166
508
        """Initiate the reprovisioning workflow."""
167
509
        LOG.info("bouncing network to enable IMDS polling")
168
510
        if self.metadata is None:
169
511
            self.metadata = {}
170
512
        self.metadata['local-hostname'] = 'azurevm'
171
513
        # This is needed in order to report our temp hostname to the platform
172
514
        # and trigger DHCP in order to get an IP address.
173
515
        self.bounce_network_with_azure_hostname()
174
516
        self._report_ready()
175
517
        contents = self._poll_imds()
176
518
        md, ud, cfg = read_azure_ovf(contents)
177
519
        return (md, ud, cfg, {'ovf-env.xml': contents})
178
520
179
431
    def _negotiate(self):
521
    def _negotiate(self):
180
432
        """Negotiate with fabric and return data from it.
522
        """Negotiate with fabric and return data from it.
181
433
523
182
@@ -453,7 +543,7 @@ class DataSourceAzure(sources.DataSource):
183
453
                "Error communicating with Azure fabric; You may experience."
543
                "Error communicating with Azure fabric; You may experience."
184
454
                "connectivity issues.", exc_info=True)
544
                "connectivity issues.", exc_info=True)
185
455
            return False
545
            return False
187
456
546
        util.del_file(REPROVISION_MARKER_FILE)
188
457
        return fabric_data
547
        return fabric_data
189
458
548
190
459
    def activate(self, cfg, is_new_instance):
549
    def activate(self, cfg, is_new_instance):
191
@@ -595,6 +685,7 @@ def address_ephemeral_resize(devpath=RESOURCE_DISK_PATH, maxwait=120,
192
595
def perform_hostname_bounce(hostname, cfg, prev_hostname):
685
def perform_hostname_bounce(hostname, cfg, prev_hostname):
193
596
    # set the hostname to 'hostname' if it is not already set to that.
686
    # set the hostname to 'hostname' if it is not already set to that.
194
597
    # then, if policy is not off, bounce the interface using command
687
    # then, if policy is not off, bounce the interface using command
195
688
    # Returns True if the network was bounced, False otherwise.
196
598
    command = cfg['command']
689
    command = cfg['command']
197
599
    interface = cfg['interface']
690
    interface = cfg['interface']
198
600
    policy = cfg['policy']
691
    policy = cfg['policy']
199
@@ -614,7 +705,8 @@ def perform_hostname_bounce(hostname, cfg, prev_hostname):
200
614
        else:
705
        else:
201
615
            LOG.debug(
706
            LOG.debug(
202
616
                "Skipping network bounce: ifupdown utils aren't present.")
707
                "Skipping network bounce: ifupdown utils aren't present.")
204
617
            return  # Don't bounce as networkd handles hostname DDNS updates
708
            # Don't bounce as networkd handles hostname DDNS updates
205
709
            return False
206
618
    LOG.debug("pubhname: publishing hostname [%s]", msg)
710
    LOG.debug("pubhname: publishing hostname [%s]", msg)
207
619
    shell = not isinstance(command, (list, tuple))
711
    shell = not isinstance(command, (list, tuple))
208
620
    # capture=False, see comments in bug 1202758 and bug 1206164.
712
    # capture=False, see comments in bug 1202758 and bug 1206164.
209
@@ -622,6 +714,7 @@ def perform_hostname_bounce(hostname, cfg, prev_hostname):
210
622
                  get_uptime=True, func=util.subp,
714
                  get_uptime=True, func=util.subp,
211
623
                  kwargs={'args': command, 'shell': shell, 'capture': False,
715
                  kwargs={'args': command, 'shell': shell, 'capture': False,
212
624
                          'env': env})
716
                          'env': env})
213
717
    return True
214
625
718
215
626
719
216
627
def crtfile_to_pubkey(fname, data=None):
720
def crtfile_to_pubkey(fname, data=None):
217
@@ -838,9 +931,35 @@ def read_azure_ovf(contents):
218
838
    if 'ssh_pwauth' not in cfg and password:
931
    if 'ssh_pwauth' not in cfg and password:
219
839
        cfg['ssh_pwauth'] = True
932
        cfg['ssh_pwauth'] = True
220
840
933
221
934
    cfg['PreprovisionedVm'] = _extract_preprovisioned_vm_setting(dom)
222
935
223
841
    return (md, ud, cfg)
936
    return (md, ud, cfg)
224
842
937
225
843
938
226
939
def _extract_preprovisioned_vm_setting(dom):
227
940
    """Read the preprovision flag from the ovf. It should not
228
941
       exist unless true."""
229
942
    platform_settings_section = find_child(
230
943
        dom.documentElement,
231
944
        lambda n: n.localName == "PlatformSettingsSection")
232
945
    if not platform_settings_section or len(platform_settings_section) == 0:
233
946
        LOG.debug("PlatformSettingsSection not found")
234
947
        return False
235
948
    platform_settings = find_child(
236
949
        platform_settings_section[0],
237
950
        lambda n: n.localName == "PlatformSettings")
238
951
    if not platform_settings or len(platform_settings) == 0:
239
952
        LOG.debug("PlatformSettings not found")
240
953
        return False
241
954
    preprovisionedVm = find_child(
242
955
        platform_settings[0],
243
956
        lambda n: n.localName == "PreprovisionedVm")
244
957
    if not preprovisionedVm or len(preprovisionedVm) == 0:
245
958
        LOG.debug("PreprovisionedVm not found")
246
959
        return False
247
960
    return bool(preprovisionedVm[0])
248
961
249
962
250
844
def encrypt_pass(password, salt_id="$6$"):
963
def encrypt_pass(password, salt_id="$6$"):
251
845
    return crypt.crypt(password, salt_id + util.rand_str(strlen=16))
964
    return crypt.crypt(password, salt_id + util.rand_str(strlen=16))
252
846
965
253
diff --git a/cloudinit/url_helper.py b/cloudinit/url_helper.py
254
index 0e0f5b4..281c87c 100644
255
--- a/cloudinit/url_helper.py
256
+++ b/cloudinit/url_helper.py
257
@@ -301,6 +301,8 @@ def wait_for_url(urls, max_wait=None, timeout=None,
258
301
    service but is not going to find one.  It is possible that the instance
301
    service but is not going to find one.  It is possible that the instance
259
302
    data host (169.254.169.254) may be firewalled off Entirely for a sytem,
302
    data host (169.254.169.254) may be firewalled off Entirely for a sytem,
260
303
    meaning that the connection will block forever unless a timeout is set.
303
    meaning that the connection will block forever unless a timeout is set.
261
304
262
305
    A value of None for max_wait will retry indefinitely.
263
304
    """
306
    """
264
305
    start_time = time.time()
307
    start_time = time.time()
265
306
308
266
@@ -311,8 +313,9 @@ def wait_for_url(urls, max_wait=None, timeout=None,
267
311
        status_cb = log_status_cb
313
        status_cb = log_status_cb
268
312
314
269
313
    def timeup(max_wait, start_time):
315
    def timeup(max_wait, start_time):
272
314
        return ((max_wait <= 0 or max_wait is None) or
316
        if (max_wait is None):
273
315
                (time.time() - start_time > max_wait))
317
            return False
274
318
        return ((max_wait <= 0) or (time.time() - start_time > max_wait))
275
316
319
276
317
    loop_n = 0
320
    loop_n = 0
277
318
    while True:
321
    while True:
278
@@ -322,7 +325,8 @@ def wait_for_url(urls, max_wait=None, timeout=None,
279
322
            if loop_n != 0:
325
            if loop_n != 0:
280
323
                if timeup(max_wait, start_time):
326
                if timeup(max_wait, start_time):
281
324
                    break
327
                    break
283
325
                if timeout and (now + timeout > (start_time + max_wait)):
328
                if (max_wait is not None and
284
329
                        timeout and (now + timeout > (start_time + max_wait))):
285
326
                    # shorten timeout to not run way over max_time
330
                    # shorten timeout to not run way over max_time
286
327
                    timeout = int((start_time + max_wait) - now)
331
                    timeout = int((start_time + max_wait) - now)
287
328
332
288
@@ -354,10 +358,11 @@ def wait_for_url(urls, max_wait=None, timeout=None,
289
354
                url_exc = e
358
                url_exc = e
290
355
359
291
356
            time_taken = int(time.time() - start_time)
360
            time_taken = int(time.time() - start_time)
296
357
            status_msg = "Calling '%s' failed [%s/%ss]: %s" % (url,
361
            max_wait_str = "%ss" % max_wait if max_wait else "unlimited"
297
358
                                                               time_taken,
362
            status_msg = "Calling '%s' failed [%s/%s]: %s" % (url,
298
359
                                                               max_wait,
363
                                                              time_taken,
299
360
                                                               reason)
364
                                                              max_wait_str,
300
365
                                                              reason)
301
361
            status_cb(status_msg)
366
            status_cb(status_msg)
302
362
            if exception_cb:
367
            if exception_cb:
303
363
                # This can be used to alter the headers that will be sent
368
                # This can be used to alter the headers that will be sent
304
diff --git a/tests/unittests/test_datasource/test_azure.py b/tests/unittests/test_datasource/test_azure.py
305
index 6341e1e..9680ac6 100644
306
--- a/tests/unittests/test_datasource/test_azure.py
307
+++ b/tests/unittests/test_datasource/test_azure.py
308
@@ -5,18 +5,20 @@ from cloudinit.util import b64e, decode_binary, load_file, write_file
309
5
from cloudinit.sources import DataSourceAzure as dsaz
5
from cloudinit.sources import DataSourceAzure as dsaz
310
6
from cloudinit.util import find_freebsd_part
6
from cloudinit.util import find_freebsd_part
311
7
from cloudinit.util import get_path_dev_freebsd
7
from cloudinit.util import get_path_dev_freebsd
313
8
8
from cloudinit.version import version_string as vs
314
9
from cloudinit.tests.helpers import (CiTestCase, TestCase, populate_dir, mock,
9
from cloudinit.tests.helpers import (CiTestCase, TestCase, populate_dir, mock,
315
10
                                     ExitStack, PY26, SkipTest)
10
                                     ExitStack, PY26, SkipTest)
316
11
11
317
12
import crypt
12
import crypt
318
13
import os
13
import os
319
14
import stat
14
import stat
320
15
import tempfile
321
15
import xml.etree.ElementTree as ET
16
import xml.etree.ElementTree as ET
322
16
import yaml
17
import yaml
323
17
18
324
18
19
326
19
def construct_valid_ovf_env(data=None, pubkeys=None, userdata=None):
20
def construct_valid_ovf_env(data=None, pubkeys=None,
327
21
                            userdata=None, platform_settings=None):
328
20
    if data is None:
22
    if data is None:
329
21
        data = {'HostName': 'FOOHOST'}
23
        data = {'HostName': 'FOOHOST'}
330
22
    if pubkeys is None:
24
    if pubkeys is None:
331
@@ -66,10 +68,12 @@ def construct_valid_ovf_env(data=None, pubkeys=None, userdata=None):
332
66
  xmlns:i="http://www.w3.org/2001/XMLSchema-instance">
68
  xmlns:i="http://www.w3.org/2001/XMLSchema-instance">
333
67
 <KmsServerHostname>kms.core.windows.net</KmsServerHostname>
69
 <KmsServerHostname>kms.core.windows.net</KmsServerHostname>
334
68
 <ProvisionGuestAgent>false</ProvisionGuestAgent>
70
 <ProvisionGuestAgent>false</ProvisionGuestAgent>
339
69
 <GuestAgentPackageName i:nil="true" />
71
 <GuestAgentPackageName i:nil="true" />"""
340
70
 </PlatformSettings></wa:PlatformSettingsSection>
72
    if platform_settings:
341
71
</Environment>
73
        for k, v in platform_settings.items():
342
72
    """
74
            content += "<%s>%s</%s>\n" % (k, v, k)
343
75
    content += """</PlatformSettings></wa:PlatformSettingsSection>
344
76
</Environment>"""
345
73
77
346
74
    return content
78
    return content
347
75
79
348
@@ -1107,4 +1111,111 @@ class TestAzureNetExists(CiTestCase):
349
1107
        self.assertTrue(hasattr(dsaz, "DataSourceAzureNet"))
1111
        self.assertTrue(hasattr(dsaz, "DataSourceAzureNet"))
350
1108
1112
351
1109
1113
352
1114
@mock.patch('cloudinit.sources.DataSourceAzure.util.subp')
353
1115
@mock.patch.object(dsaz, 'get_hostname')
354
1116
@mock.patch.object(dsaz, 'set_hostname')
355
1117
class TestAzureDataSourcePreprovisioning(CiTestCase):
356
1118
357
1119
    def setUp(self):
358
1120
        super(TestAzureDataSourcePreprovisioning, self).setUp()
359
1121
        tmp = self.tmp_dir()
360
1122
        self.waagent_d = self.tmp_path('/var/lib/waagent', tmp)
361
1123
        self.paths = helpers.Paths({'cloud_dir': tmp})
362
1124
        dsaz.BUILTIN_DS_CONFIG['data_dir'] = self.waagent_d
363
1125
364
1126
    def test_read_azure_ovf_with_flag(self, *args):
365
1127
        """The read_azure_ovf method should set the PreprovisionedVM
366
1128
           cfg flag if the propper setting is present."""
367
1129
        content = construct_valid_ovf_env(
368
1130
            platform_settings={"PreprovisionedVm": "True"})
369
1131
        ret = dsaz.read_azure_ovf(content)
370
1132
        cfg = ret[2]
371
1133
        self.assertTrue(cfg['PreprovisionedVm'])
372
1134
373
1135
    def test_read_azure_ovf_without_flag(self, *args):
374
1136
        """The read_azure_ovf method should not set the
375
1137
           PreprovisionedVM cfg flag."""
376
1138
        content = construct_valid_ovf_env()
377
1139
        ret = dsaz.read_azure_ovf(content)
378
1140
        cfg = ret[2]
379
1141
        self.assertFalse(cfg['PreprovisionedVm'])
380
1142
381
1143
    @mock.patch('requests.Session.request')
382
1144
    def test_poll_imds_returns_ovf_env(self, fake_resp, *args):
383
1145
        """The _poll_imds method should return the ovf_env.xml."""
384
1146
        url = 'http://{0}/metadata/reprovisiondata?api-version=2017-04-02'
385
1147
        host = "169.254.169.254"
386
1148
        full_url = url.format(host)
387
1149
        fake_resp.return_value = mock.MagicMock(status_code=200, text="ovf")
388
1150
        dsa = dsaz.DataSourceAzure({}, distro=None, paths=self.paths)
389
1151
        self.assertTrue(len(dsa._poll_imds()) > 0)
390
1152
        print(fake_resp.call_args_list)
391
1153
        self.assertEqual(fake_resp.call_args_list,
392
1154
                         [mock.call(allow_redirects=True,
393
1155
                                    headers={'Metadata': 'true',
394
1156
                                             'User-Agent':
395
1157
                                             'Cloud-Init/%s' % vs()
396
1158
                                             }, method='GET', timeout=60.0,
397
1159
                                    url=full_url),
398
1160
                          mock.call(allow_redirects=True,
399
1161
                                    headers={'Metadata': 'true',
400
1162
                                             'User-Agent':
401
1163
                                             'Cloud-Init/%s' % vs()
402
1164
                                             }, method='GET', url=full_url)])
403
1165
404
1166
    @mock.patch('requests.Session.request')
405
1167
    def test__reprovision_calls__poll_imds(self, fake_resp, *args):
406
1168
        """The _reprovision method should call poll IMDS."""
407
1169
        url = 'http://{0}/metadata/reprovisiondata?api-version=2017-04-02'
408
1170
        host = "169.254.169.254"
409
1171
        full_url = url.format(host)
410
1172
        hostname = "myhost"
411
1173
        username = "myuser"
412
1174
        odata = {'HostName': hostname, 'UserName': username}
413
1175
        content = construct_valid_ovf_env(data=odata)
414
1176
        fake_resp.return_value = mock.MagicMock(status_code=200, text=content)
415
1177
        dsa = dsaz.DataSourceAzure({}, distro=None, paths=self.paths)
416
1178
        md, ud, cfg, d = dsa._reprovision()
417
1179
        self.assertEqual(md['local-hostname'], hostname)
418
1180
        self.assertEqual(cfg['system_info']['default_user']['name'], username)
419
1181
        self.assertEqual(fake_resp.call_args_list,
420
1182
                         [mock.call(allow_redirects=True,
421
1183
                                    headers={'Metadata': 'true',
422
1184
                                             'User-Agent':
423
1185
                                             'Cloud-Init/%s' % vs()},
424
1186
                                    method='GET', timeout=60.0, url=full_url),
425
1187
                          mock.call(allow_redirects=True,
426
1188
                                    headers={'Metadata': 'true',
427
1189
                                             'User-Agent':
428
1190
                                             'Cloud-Init/%s' % vs()},
429
1191
                                    method='GET', url=full_url)])
430
1192
431
1193
    @mock.patch('os.path.isfile')
432
1194
    @mock.patch.object(dsaz, 'open', create=True)
433
1195
    def test__should_reprovision_with_true_cfg(self, isfile, myopen, *args):
434
1196
        """The _should_reprovision method should return true with config
435
1197
           flag present."""
436
1198
        isfile.return_value = False
437
1199
        myopen.return_value = tempfile.TemporaryFile()
438
1200
        dsa = dsaz.DataSourceAzure({}, distro=None, paths=self.paths)
439
1201
        self.assertTrue(dsa._should_reprovision(
440
1202
            (None, None, {'PreprovisionedVm': True}, None)))
441
1203
442
1204
    @mock.patch('os.path.isfile')
443
1205
    def test__should_reprovision_with_file_existing(self, isfile, *args):
444
1206
        """The _should_reprovision method should return True if the sentinal
445
1207
           exists."""
446
1208
        isfile.return_value = True
447
1209
        dsa = dsaz.DataSourceAzure({}, distro=None, paths=self.paths)
448
1210
        self.assertTrue(dsa._should_reprovision(
449
1211
            (None, None, {'preprovisionedvm': False}, None)))
450
1212
451
1213
    @mock.patch('os.path.isfile')
452
1214
    def test__should_reprovision_returns_false(self, isfile, *args):
453
1215
        """The _should_reprovision method should return False
454
1216
           if config and sentinal are not present."""
455
1217
        isfile.return_value = False
456
1218
        dsa = dsaz.DataSourceAzure({}, distro=None, paths=self.paths)
457
1219
        self.assertFalse(dsa._should_reprovision((None, None, {}, None)))
458
1220
459
1110
# vi: ts=4 expandtab
1221
# vi: ts=4 expandtab
Reviewer	Review Type	Date Requested	Status
Server Team CI bot	continuous-integration		Approve 11 hours ago
Scott Moser			Needs Fixing on 2018-01-03
Chad Smith			Approve on 2017-12-14
Paul Meyer (community)		2017-11-28	Approve on 2017-12-01
Review via email: mp+334341@code.launchpad.net