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_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")