Exemplo n.º 1
0
  public SendEvent(
      TransformerManager transformerManager, GetRecordsType request, QueryRequest query)
      throws CswException {

    URI deliveryMethodUrl;
    if (request.getResponseHandler() != null && !request.getResponseHandler().isEmpty()) {

      try {
        deliveryMethodUrl = new URI(request.getResponseHandler().get(0));
      } catch (URISyntaxException e) {
        throw new CswException("Invalid ResponseHandler URL", e);
      }
    } else {
      String msg = "Subscriptions require a ResponseHandler URL to be specified";
      LOGGER.error(msg);
      throw new CswException(msg);
    }
    this.transformerManager = transformerManager;
    this.query = query;
    this.callbackUrl = deliveryMethodUrl;
    this.request = request;
    this.outputSchema = request.getOutputSchema();
    this.mimeType = request.getOutputFormat();
    QueryType queryType = (QueryType) request.getAbstractQuery().getValue();
    this.elementName = queryType.getElementName();
    this.elementSetType =
        (queryType.getElementSetName() != null) ? queryType.getElementSetName().getValue() : null;
    this.resultType = request.getResultType() == null ? ResultType.HITS : request.getResultType();
    SecureCxfClientFactory<CswSubscribe> cxfClientFactory =
        new SecureCxfClientFactory<>(callbackUrl.toString(), CswSubscribe.class);
    webClient = cxfClientFactory.getWebClient();
  }
Exemplo n.º 2
0
  @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);
  }