def test_exception_get_vip_by_portpool_id(self): requests_mock = patch( 'networkapiclient.ClientFactory.ApiVipRequest.search').start() requests_mock.side_effect = NetworkAPIClientError('') napi = NetworkAPI() data = napi.get_vip_by_portpool_id(3) self.assertEqual(data, [])
def test_get_vip_by_portpool_id(self): """Test assert called get vip by port""" requests_mock = patch( 'networkapiclient.ClientFactory.ApiVipRequest.search').start() requests_mock.return_value = {'vips': [{'id': 1}]} napi = NetworkAPI() data = napi.get_vip_by_portpool_id(1) requests_mock.assert_called_once_with( search={'extends_search': [ {'viprequestport__viprequestportpool': 1}]}, kind='details') self.assertDictEqual(data, {'id': 1})