def test_get_primary_ip(netif_gateways, netif_ifaddresses):
with mock.patch('netifaces.gateways') as gw, \
mock.patch('netifaces.ifaddresses') as ifaddr:
gw.return_value = netif_gateways
ifaddr.return_value = netif_ifaddresses
primary_ip = agent.get_primary_ip()
assert primary_ip == '192.168.1.3'
def test_get_primary_ip_none_on_exception(netif_gateways_invalid, netif_ifaddresses):
with mock.patch('netifaces.gateways') as gw, \
mock.patch('netifaces.ifaddresses') as ifaddr:
gw.return_value = netif_gateways_invalid
ifaddr.return_value = netif_ifaddresses
primary_ip = agent.get_primary_ip()
assert primary_ip is None
