PepPortTypeImpl implements PepPortType { @Override public RetrieveDocumentSetResponse retrieveDocumentSet( RetrieveDocumentSetRequest input) { return pep.retrieveDocumentSet(input); } PepPortTypeImpl(PolicyEnforcementPoint pep); @Override DirectEmailSendResponse directEmailSend(
			DirectEmailSendRequest parameters); @Override AdhocQueryResponse registryStoredQuery(AdhocQueryRequest input); @Override RetrieveDocumentSetResponse retrieveDocumentSet(
			RetrieveDocumentSetRequest input);  }
@Test public void testRetrieveDocumentSet() { RetrieveDocumentSetRequest req = new RetrieveDocumentSetRequest(); RetrieveDocumentSetResponse resp = new RetrieveDocumentSetResponse(); RegistryResponseType regResp = new RegistryResponseType(); regResp.setRequestId(TEST_REQUEST_ID); resp.setRegistryResponse(regResp); PepService service = new PepService(wsdlURL, serviceName); PepPortType port = service.getXDSHTTPEndpoint(); when(pep.retrieveDocumentSet(isA(RetrieveDocumentSetRequest.class))).thenReturn(resp); RetrieveDocumentSetResponse response = port.retrieveDocumentSet(req); assertEquals(resp.getRegistryResponse().getRequestId(), response.getRegistryResponse().getRequestId()); }