def test_get_uplink_mac_raises_NoDefaultGateway_when_gateway_is_empty( self, mock_gateways): mock_gateways.return_value = {} mock_arp_cache = mock.Mock(spec=arp.ArpCache) mock_arp_cache.ip_to_mac.side_effect = Exception( 'Should not be called.') with self.assertRaises(network.NoDefaultGateway): network.get_uplink_mac(mock_arp_cache)
def test_get_uplink_mac(self, mock_gateways): mock_gateways.return_value = { 'default': {route.AF_INET: ('5.4.3.2', 'eth0')} } mock_arp_cache = mock.Mock(spec=arp.ArpCache) mock_arp_cache.ip_to_mac.return_value = '00:11:22:33:44:55' actual_mac = network.get_uplink_mac(mock_arp_cache) self.assertEqual('00:11:22:33:44:55', actual_mac) mock_arp_cache.ip_to_mac.assert_called_once_with('5.4.3.2')
def test_get_uplink_mac(self, mock_gateways): mock_gateways.return_value = { 'default': { route.AF_INET: ('5.4.3.2', 'eth0') } } mock_arp_cache = mock.Mock(spec=arp.ArpCache) mock_arp_cache.ip_to_mac.return_value = '00:11:22:33:44:55' actual_mac = network.get_uplink_mac(mock_arp_cache) self.assertEqual('00:11:22:33:44:55', actual_mac) mock_arp_cache.ip_to_mac.assert_called_once_with('5.4.3.2')