Examples with MessageToEventMapper org.talend.esb.sam.agent.eventproducer.MessageToEventMapper used on opensource projects

Example 1 with MessageToEventMapper

use of org.talend.esb.sam.agent.eventproducer.MessageToEventMapper in project tesb-rt-se by Talend.

the class EventFeatureImpl method setQueue.

/**
 * Sets the queue.
 *
 * @param queue the new queue
 */
@Inject
public void setQueue(EventQueue queue) {
    if (epi == null) {
        MessageToEventMapper mapper = new MessageToEventMapper();
        mapper.setMaxContentLength(maxContentLength);
        epi = new EventProducerInterceptor(mapper, queue);
    }
}

Example 2 with MessageToEventMapper

use of org.talend.esb.sam.agent.eventproducer.MessageToEventMapper in project tesb-rt-se by Talend.

the class MessageToEventMapperTest method testMapEventRest.

@Test
public void testMapEventRest() throws IOException, EndpointException {
    QName portType = new QName("PORT_TYPE");
    EndpointInfo info = EasyMock.createMock(EndpointInfo.class);
    EasyMock.expect(info.getName()).andReturn(portType).anyTimes();
    EasyMock.expect(info.getAddress()).andReturn(null).anyTimes();
    EasyMock.replay(info);
    Endpoint endpoint = EasyMock.createMock(Endpoint.class);
    EasyMock.expect(endpoint.getEndpointInfo()).andReturn(info).anyTimes();
    Map<String, String> samProperties = new HashMap<String, String>();
    EasyMock.expect(endpoint.get(EventFeature.SAM_PROPERTIES)).andReturn(samProperties).anyTimes();
    EasyMock.replay(endpoint);
    Message outMessage = EasyMock.createMock(Message.class);
    EasyMock.expect(outMessage.containsKey(Message.HTTP_REQUEST_METHOD)).andReturn(true).anyTimes();
    EasyMock.expect(outMessage.get(Message.HTTP_REQUEST_METHOD)).andReturn("POST").anyTimes();
    EasyMock.expect(outMessage.containsKey(Message.REQUEST_URI)).andReturn(true).anyTimes();
    EasyMock.expect(outMessage.get(Message.REQUEST_URI)).andReturn("REQUEST_URI").anyTimes();
    EasyMock.expect(outMessage.containsKey(Message.BASE_PATH)).andReturn(true).anyTimes();
    EasyMock.expect(outMessage.get(Message.BASE_PATH)).andReturn("REQUEST_URI").anyTimes();
    EasyMock.replay(outMessage);
    Exchange e = EasyMock.createMock(Exchange.class);
    EasyMock.expect(e.getOutMessage()).andReturn(outMessage).anyTimes();
    EasyMock.expect(e.getOutFaultMessage()).andReturn(null).anyTimes();
    EasyMock.expect(e.getInFaultMessage()).andReturn(null).anyTimes();
    EasyMock.expect(e.getBinding()).andReturn(null).anyTimes();
    EasyMock.expect(e.getEndpoint()).andReturn(endpoint).anyTimes();
    EasyMock.expect(e.get("org.apache.cxf.resource.operation.name")).andReturn("operationName").anyTimes();
    EasyMock.replay(e);
    AuthorizationPolicy authPolicy = EasyMock.createMock(AuthorizationPolicy.class);
    EasyMock.expect(authPolicy.getUserName()).andReturn("USERNAME").anyTimes();
    EasyMock.replay(authPolicy);
    CachedOutputStream cos = new CachedOutputStream();
    cos.write(1);
    cos.write(2);
    cos.write(3);
    Message message = EasyMock.createNiceMock(Message.class);
    EasyMock.expect(message.entrySet()).andReturn(null).anyTimes();
    EasyMock.expect(message.get(Message.REQUESTOR_ROLE)).andReturn(true).anyTimes();
    EasyMock.expect(message.getExchange()).andReturn(e).anyTimes();
    EasyMock.expect(message.get(Message.ENCODING)).andReturn("UTF-8").anyTimes();
    EasyMock.expect(message.getContent(CachedOutputStream.class)).andReturn(cos).anyTimes();
    EasyMock.expect(message.get("FlowId")).andReturn(FlowID).anyTimes();
    EasyMock.expect(message.get(CorrelationIdHelper.CORRELATION_ID_KEY)).andReturn("CORRELATION_ID_KEY").anyTimes();
    EasyMock.expect(message.get(Message.ACCEPT_CONTENT_TYPE)).andReturn("XML").anyTimes();
    EasyMock.expect(message.get(Message.CONTENT_TYPE)).andReturn("XML").anyTimes();
    EasyMock.expect(message.get(Message.RESPONSE_CODE)).andReturn(0).anyTimes();
    EasyMock.expect(message.get(SecurityContext.class)).andReturn(null).anyTimes();
    EasyMock.expect(message.get(AuthorizationPolicy.class)).andReturn(authPolicy).anyTimes();
    EasyMock.expect(message.get(CustomInfo.class)).andReturn(EasyMock.createMock(CustomInfo.class)).anyTimes();
    EasyMock.replay(message);
    MessageToEventMapper mapper = new MessageToEventMapper();
    mapper.setMaxContentLength(2);
    Event event = mapper.mapToEvent(message);
    Assert.assertEquals(EventTypeEnum.RESP_IN, event.getEventType());
    Assert.assertEquals("PORT_TYPE", event.getMessageInfo().getPortType());
    Assert.assertEquals("POST[/]", event.getMessageInfo().getOperationName());
    Assert.assertEquals("http://cxf.apache.org/transports/http", event.getMessageInfo().getTransportType());
    Assert.assertEquals(FlowID, event.getMessageInfo().getFlowId());
    Assert.assertNull(event.getMessageInfo().getMessageId());
}

Example 3 with MessageToEventMapper

use of org.talend.esb.sam.agent.eventproducer.MessageToEventMapper in project tesb-rt-se by Talend.

the class MessageToEventMapperTest method testMaxContentLength.

@Test
public void testMaxContentLength() throws IOException, EndpointException {
    Message message = getTestMessage();
    MessageToEventMapper mapper = new MessageToEventMapper();
    mapper.setMaxContentLength(MAXCONTENTLENGTH);
    Event event = mapper.mapToEvent(message);
    // System.out.println(event.getContent());
    Assert.assertEquals(MAXCONTENTLENGTH, event.getContent().length());
    Assert.assertEquals("<cut><![CDATA[" + TESTCONTENT.substring(0, MAXCONTENTLENGTH - 23) + "]]></cut>", event.getContent());
    Assert.assertTrue(event.isContentCut());
}

Example 4 with MessageToEventMapper

use of org.talend.esb.sam.agent.eventproducer.MessageToEventMapper in project tesb-rt-se by Talend.

the class MessageToEventMapperTest method testMapEvent.

@Test
public void testMapEvent() throws IOException, EndpointException {
    Message message = getTestMessage();
    Event event = new MessageToEventMapper().mapToEvent(message);
    Assert.assertEquals(EventTypeEnum.REQ_IN, event.getEventType());
    Assert.assertEquals("{interfaceNs}interfaceName", event.getMessageInfo().getPortType());
    Assert.assertEquals("{namespace}opName", event.getMessageInfo().getOperationName());
    Assert.assertEquals(TransportType, event.getMessageInfo().getTransportType());
    Assert.assertEquals(FlowID, event.getMessageInfo().getFlowId());
    Assert.assertNull(event.getMessageInfo().getMessageId());
    // By default the content should not be cut
    Assert.assertEquals(TESTCONTENT, event.getContent());
    Assert.assertFalse(event.isContentCut());
    // Principal
    Assert.assertEquals(PrincipalString, event.getOriginator().getPrincipal());
    Map<String, String> customInfo = event.getCustomInfo();
    // Assert.assertEquals(2, customInfo.keySet().size());
    // Assert.assertEquals(Address, customInfo.get("address"));
    Assert.assertEquals("value1", customInfo.get("key1"));
}