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})