private Response sendEvent(String operation, Metacard... metacards) {
try {
List<Result> results =
Arrays.asList(metacards).stream().map(ResultImpl::new).collect(Collectors.toList());
SourceResponse queryResponse = new QueryResponseImpl(query, results, true, metacards.length);
LOGGER.debug(
"Attempting to transform an Event with mime-type: {} & outputSchema: {}",
mimeType,
outputSchema);
QueryResponseTransformer transformer;
Map<String, Serializable> arguments = new HashMap<>();
if (StringUtils.isBlank(outputSchema)
&& StringUtils.isNotBlank(mimeType)
&& !XML_MIME_TYPES.contains(mimeType)) {
transformer = transformerManager.getTransformerByMimeType(mimeType);
} else {
transformer = transformerManager.getTransformerBySchema(CswConstants.CSW_OUTPUT_SCHEMA);
if (elementName != null) {
arguments.put(CswConstants.ELEMENT_NAMES, elementName.toArray());
}
arguments.put(CswConstants.OUTPUT_SCHEMA_PARAMETER, outputSchema);
arguments.put(CswConstants.ELEMENT_SET_TYPE, elementSetType);
arguments.put(CswConstants.IS_BY_ID_QUERY, false);
arguments.put(CswConstants.GET_RECORDS, request);
arguments.put(CswConstants.RESULT_TYPE_PARAMETER, resultType);
arguments.put(CswConstants.WRITE_NAMESPACES, false);
}
if (transformer == null) {
throw new WebApplicationException(
new CatalogTransformerException("Unable to locate Transformer."));
}
BinaryContent binaryContent = transformer.transform(queryResponse, arguments);
if (binaryContent == null) {
throw new WebApplicationException(
new CatalogTransformerException("Transformer returned null."));
}
return webClient.invoke(operation, binaryContent.getByteArray());
} catch (IOException | CatalogTransformerException | RuntimeException e) {
LOGGER.error("Error sending event to {}.", callbackUrl, e);
}
return null;
}
@Before
public void setUp() throws Exception {
System.setProperty("ddf.home", ".");
callbackURI = new URL("https://localhost:12345/services/csw/subscription/event");
ObjectFactory objectFactory = new ObjectFactory();
request = new GetRecordsType();
request.setOutputSchema(CswConstants.CSW_OUTPUT_SCHEMA);
request.setResultType(ResultType.RESULTS);
request.getResponseHandler().add(callbackURI.toString());
queryType = new QueryType();
elementSetNameType = new ElementSetNameType();
elementSetNameType.setValue(ElementSetType.BRIEF);
queryType.setElementSetName(elementSetNameType);
request.setAbstractQuery(objectFactory.createAbstractQuery(queryType));
transformerManager = mock(TransformerManager.class);
transformer = mock(QueryResponseTransformer.class);
binaryContent = mock(BinaryContent.class);
when(transformerManager.getTransformerBySchema(
Matchers.contains(CswConstants.CSW_OUTPUT_SCHEMA)))
.thenReturn(transformer);
when(transformer.transform(any(SourceResponse.class), anyMap())).thenReturn(binaryContent);
when(binaryContent.getByteArray()).thenReturn("byte array with message contents".getBytes());
query = mock(QueryRequest.class);
metacard = mock(Metacard.class);
webclient = mock(WebClient.class);
mockCxfClientFactory = mock(SecureCxfClientFactory.class);
response = mock(Response.class);
subject = mock(Subject.class);
mockSecurity = mock(Security.class);
headers.put(Subject.class.toString(), Arrays.asList(new Subject[] {subject}));
AccessPlugin accessPlugin = mock(AccessPlugin.class);
accessPlugins.add(accessPlugin);
when(mockCxfClientFactory.getWebClient()).thenReturn(webclient);
// when(webclient.head()).thenReturn(response);
when(webclient.invoke(anyString(), any(QueryResponse.class))).thenReturn(response);
when(response.getHeaders()).thenReturn(headers);
when(accessPlugin.processPostQuery(any(QueryResponse.class)))
.thenAnswer(
new Answer<QueryResponse>() {
@Override
public QueryResponse answer(InvocationOnMock invocationOnMock) throws Throwable {
return (QueryResponse) invocationOnMock.getArguments()[0];
}
});
sendEvent = new SendEventExtension(transformerManager, request, query, mockCxfClientFactory);
sendEvent.setSubject(subject);
}