def test_latest_lease_used(self):
encoded_addresses = ['5:4:3:2', '4:3:2:1']
file_content = '\n'.join([self._build_lease_content(encoded_address)
for encoded_address in encoded_addresses])
self.load_file.return_value = file_content
self.assertEqual(encoded_addresses[-1].replace(':', '.'),
wa_shim.find_endpoint("foobar"))
def test_missing_special_azure_line(self):
"""wa_shim find_endpoint uses default endpoint if leasefile is found
but does not contain DHCP Option 245 (whose value is the endpoint)
"""
self.load_file.return_value = ''
self.dhcp_options.return_value = {'eth0': {'key': 'value'}}
self.assertEqual(wa_shim.find_endpoint(), "168.63.129.16")
def test_missing_special_azure_line(self):
"""wa_shim find_endpoint uses default endpoint if leasefile is found
but does not contain DHCP Option 245 (whose value is the endpoint)
"""
self.load_file.return_value = ''
self.dhcp_options.return_value = {'eth0': {'key': 'value'}}
self.assertEqual(wa_shim.find_endpoint(), "168.63.129.16")
def test_latest_lease_used(self, m_is_freebsd):
m_is_freebsd.return_value = False # To avoid hitting load_file
encoded_addresses = ['5:4:3:2', '4:3:2:1']
file_content = '\n'.join([self._build_lease_content(encoded_address)
for encoded_address in encoded_addresses])
self.load_file.return_value = file_content
self.assertEqual(encoded_addresses[-1].replace(':', '.'),
wa_shim.find_endpoint("foobar"))
def test_from_dhcp_client(self):
self.dhcp_options.return_value = {"eth0": {"unknown_245": "5:4:3:2"}}
self.assertEqual('5.4.3.2', wa_shim.find_endpoint(None))
def test_missing_file(self):
"""wa_shim find_endpoint uses default endpoint if leasefile not found
"""
self.assertEqual(wa_shim.find_endpoint(), "168.63.129.16")
def test_from_dhcp_client(self):
self.dhcp_options.return_value = {"eth0": {"unknown_245": "5:4:3:2"}}
self.assertEqual('5.4.3.2', wa_shim.find_endpoint(None))
def test_missing_file(self):
"""wa_shim find_endpoint uses default endpoint if leasefile not found
"""
self.assertEqual(wa_shim.find_endpoint(), "168.63.129.16")
