def test_get_endpoints(requests_mock):
from PaloAltoNetworks_XDR import get_endpoints_command, Client
get_endpoints_response = load_test_data('./test_data/get_endpoints.json')
requests_mock.post(f'{XDR_URL}/public_api/v1/endpoints/get_endpoint/',
json=get_endpoints_response)
client = Client(base_url=f'{XDR_URL}/public_api/v1', headers={})
args = {'hostname': 'foo', 'page': 1, 'limit': 3}
_, outputs, _ = get_endpoints_command(client, args)
assert get_endpoints_response.get('reply').get('endpoints') == \
outputs['PaloAltoNetworksXDR.Endpoint(val.endpoint_id == obj.endpoint_id)']
def test_get_all_endpoints_using_limit(requests_mock):
from PaloAltoNetworks_XDR import get_endpoints_command, Client
get_endpoints_response = load_test_data(
'./test_data/get_all_endpoints.json')
requests_mock.post(f'{XDR_URL}/public_api/v1/endpoints/get_endpoints/',
json=get_endpoints_response)
client = Client(base_url=f'{XDR_URL}/public_api/v1', headers={})
args = {'limit': 1, 'page': 0, 'sort_order': 'asc'}
_, outputs, _ = get_endpoints_command(client, args)
expected_endpoint = get_endpoints_response.get('reply')[0]
assert [expected_endpoint] == outputs[
'PaloAltoNetworksXDR.Endpoint(val.endpoint_id == obj.endpoint_id)']