def test_exception_get_ipv4_by_ip_equipment_id(self):
requests_mock = patch(
'networkapiclient.ClientFactory.ApiIPv4.search').start()
requests_mock.side_effect = NetworkAPIClientError('')
napi = NetworkAPI()
data = napi.get_ipv4_by_ip_equipment_id(3)
self.assertEqual(data, [])
def test_get_ipv4_by_ip_equipment_id(self):
"""Test assert called get ipv4 by ip eqpt"""
requests_mock = patch(
'networkapiclient.ClientFactory.ApiIPv4.search').start()
requests_mock.return_value = {'ips': [{'id': 1}]}
napi = NetworkAPI()
data = napi.get_ipv4_by_ip_equipment_id(1)
requests_mock.assert_called_once_with(
search={'extends_search': [{'ipequipamento': 1}]},
fields=['networkipv4', 'ip_formated']
)
self.assertDictEqual(data, {'id': 1})
