def test_unisolate_endpoint_pending_isolation(requests_mock): from PaloAltoNetworks_XDR import unisolate_endpoint_command, Client requests_mock.post(f'{XDR_URL}/public_api/v1/endpoints/get_endpoint/', json={ 'reply': { 'endpoints': [ { 'endpoint_id': '1111', "is_isolated": "AGENT_PENDING_ISOLATION" } ] } }) unisolate_endpoint_response = load_test_data('./test_data/unisolate_endpoint.json') requests_mock.post(f'{XDR_URL}/public_api/v1/endpoints/unisolate', json=unisolate_endpoint_response) client = Client( base_url=f'{XDR_URL}/public_api/v1', headers={} ) args = { "endpoint_id": "1111" } with pytest.raises(ValueError, match='Error: Endpoint 1111 is pending isolation and therefore can not be' ' un-isolated.'): unisolate_endpoint_command(client, args)
def test_unisolate_endpoint(requests_mock): from PaloAltoNetworks_XDR import unisolate_endpoint_command, Client requests_mock.post(f'{XDR_URL}/public_api/v1/endpoints/get_endpoint/', json={ 'reply': { 'endpoints': [ { 'endpoint_id': '1111' } ] } }) unisolate_endpoint_response = load_test_data('./test_data/unisolate_endpoint.json') requests_mock.post(f'{XDR_URL}/public_api/v1/endpoints/unisolate', json=unisolate_endpoint_response) client = Client( base_url=f'{XDR_URL}/public_api/v1' ) args = { "endpoint_id": "1111" } readable_output, outputs, _ = unisolate_endpoint_command(client, args) assert outputs is None assert readable_output == 'Endpoint 1111 has un-isolated successfully'
def test_unisolate_endpoint(requests_mock): from PaloAltoNetworks_XDR import unisolate_endpoint_command, Client requests_mock.post(f'{XDR_URL}/public_api/v1/endpoints/get_endpoint/', json={ 'reply': { 'endpoints': [ { 'endpoint_id': '1111', "endpoint_status": "CONNECTED" } ] } }) unisolate_endpoint_response = load_test_data('./test_data/unisolate_endpoint.json') requests_mock.post(f'{XDR_URL}/public_api/v1/endpoints/unisolate', json=unisolate_endpoint_response) client = Client( base_url=f'{XDR_URL}/public_api/v1', headers={} ) args = { "endpoint_id": "1111" } readable_output, _, _ = unisolate_endpoint_command(client, args) assert readable_output == 'The un-isolation request has been submitted successfully on Endpoint 1111.\n' \ 'To check the endpoint isolation status please run:' \ ' !xdr-get-endpoints endpoint_id_list=1111 and look at the [is_isolated] field.'