Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- @mock.patch('requests.Session.request')
- @mock.patch('cloudinit.net.dhcp.maybe_perform_dhcp_discovery')
- @mock.patch('cloudinit.net.EphemeralIPv4Network')
- @mock.patch('cloudinit.sources.DataSourceAzure.util.is_FreeBSD')
- def test_poll_imds_returns_ovf_env(self, fake_resp, m_dhcp, m_net, m_is_bsd, *args):
- """The _poll_imds method should return the ovf_env.xml."""
- fake_resp.return_value = mock.MagicMock(status_code=200, text="ovf")
- m_dhcp.return_value = [{
- 'interface': 'eth9', 'fixed-address': '192.168.2.9',
- 'routers': '192.168.2.1', 'subnet-mask': '255.255.255.0',
- 'broadcast-address': '192.168.2.255'}]
- m_is_bsd.return_value = False
- url = 'http://{0}/metadata/reprovisiondata?api-version=2017-04-02'
- host = "169.254.169.254"
- full_url = url.format(host)
- dsa = dsaz.DataSourceAzure({}, distro=None, paths=self.paths)
- self.assertTrue(len(dsa._poll_imds()) > 0)
- print(fake_resp.call_args_list)
- self.assertEqual(fake_resp.call_args_list,
- [mock.call(allow_redirects=True,
- headers={'Metadata': 'true',
- 'User-Agent':
- 'Cloud-Init/%s' % vs()
- }, method='GET', timeout=60.0,
- url=full_url),
- mock.call(allow_redirects=True,
- headers={'Metadata': 'true',
- 'User-Agent':
- 'Cloud-Init/%s' % vs()
- }, method='GET', url=full_url)])
- m_dhcp.assert_called_once_with('eth9')
- m_net.assert_called_once_with(
- broadcast='192.168.2.255', interface='eth9', ip='192.168.2.9',
- prefix_or_mask='255.255.255.0', router='192.168.2.1')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement