def test_exception_get_equipment(self):
requests_mock = patch(
'networkapiclient.ClientFactory.ApiEquipment.get').start()
requests_mock.side_effect = NetworkAPIClientError('')
napi = NetworkAPI()
data = napi.get_equipment(3)
self.assertEqual(data, [])
def test_get_equipment(self):
"""Test assert called get equipment"""
requests_mock = patch(
'networkapiclient.ClientFactory.ApiEquipment.get').start()
requests_mock.return_value = {'equipments': [{'id': 1}]}
napi = NetworkAPI()
data = napi.get_equipment(1)
requests_mock.assert_called_once_with(
ids=[1],
include=['equipment_type__details',
'ipv4__basic__networkipv4', 'ipv6__basic__networkipv6']
)
self.assertDictEqual(data, {'id': 1})
