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