Advertisement
Guest User

Untitled

a guest
Jan 12th, 2018
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.86 KB | None | 0 0
  1. @mock.patch('requests.Session.request')
  2. @mock.patch('cloudinit.net.dhcp.maybe_perform_dhcp_discovery')
  3. @mock.patch('cloudinit.net.EphemeralIPv4Network')
  4. @mock.patch('cloudinit.sources.DataSourceAzure.util.is_FreeBSD')
  5. def test_poll_imds_returns_ovf_env(self, fake_resp, m_dhcp, m_net, m_is_bsd, *args):
  6.     """The _poll_imds method should return the ovf_env.xml."""
  7.     fake_resp.return_value = mock.MagicMock(status_code=200, text="ovf")
  8.     m_dhcp.return_value = [{
  9.         'interface': 'eth9', 'fixed-address': '192.168.2.9',
  10.         'routers': '192.168.2.1', 'subnet-mask': '255.255.255.0',
  11.         'broadcast-address': '192.168.2.255'}]
  12.     m_is_bsd.return_value = False
  13.     url = 'http://{0}/metadata/reprovisiondata?api-version=2017-04-02'
  14.     host = "169.254.169.254"
  15.     full_url = url.format(host)
  16.     dsa = dsaz.DataSourceAzure({}, distro=None, paths=self.paths)
  17.     self.assertTrue(len(dsa._poll_imds()) > 0)
  18.     print(fake_resp.call_args_list)
  19.     self.assertEqual(fake_resp.call_args_list,
  20.                      [mock.call(allow_redirects=True,
  21.                                 headers={'Metadata': 'true',
  22.                                          'User-Agent':
  23.                                          'Cloud-Init/%s' % vs()
  24.                                          }, method='GET', timeout=60.0,
  25.                                 url=full_url),
  26.                       mock.call(allow_redirects=True,
  27.                                 headers={'Metadata': 'true',
  28.                                          'User-Agent':
  29.                                          'Cloud-Init/%s' % vs()
  30.                                          }, method='GET', url=full_url)])
  31.     m_dhcp.assert_called_once_with('eth9')
  32.     m_net.assert_called_once_with(
  33.         broadcast='192.168.2.255', interface='eth9', ip='192.168.2.9',
  34.         prefix_or_mask='255.255.255.0', router='192.168.2.1')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement