| /** |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.camel.component.test; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| import org.apache.camel.Component; |
| import org.apache.camel.Endpoint; |
| import org.apache.camel.Exchange; |
| import org.apache.camel.Processor; |
| import org.apache.camel.component.mock.MockEndpoint; |
| import org.apache.camel.util.EndpointHelper; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| /** |
| * A <a href="http://camel.apache.org/test.html">Test Endpoint</a> is a |
| * <a href="http://camel.apache.org/mock.html">Mock Endpoint</a> for testing but it will |
| * pull all messages from the nested endpoint and use those as expected message body assertions. |
| * |
| * @version |
| */ |
| public class TestEndpoint extends MockEndpoint { |
| private static final transient Logger LOG = LoggerFactory.getLogger(TestEndpoint.class); |
| private final Endpoint expectedMessageEndpoint; |
| private long timeout = 2000L; |
| |
| public TestEndpoint(String endpointUri, Component component, Endpoint expectedMessageEndpoint) { |
| super(endpointUri, component); |
| this.expectedMessageEndpoint = expectedMessageEndpoint; |
| } |
| |
| @Override |
| protected void doStart() throws Exception { |
| LOG.debug("Consuming expected messages from: {}", expectedMessageEndpoint); |
| final List<Object> expectedBodies = new ArrayList<Object>(); |
| EndpointHelper.pollEndpoint(expectedMessageEndpoint, new Processor() { |
| public void process(Exchange exchange) throws Exception { |
| Object body = getInBody(exchange); |
| expectedBodies.add(body); |
| } |
| }, timeout); |
| |
| LOG.debug("Received: {} expected message(s) from: {}", expectedBodies.size(), expectedMessageEndpoint); |
| expectedBodiesReceived(expectedBodies); |
| } |
| |
| /** |
| * This method allows us to convert or coerce the expected message body into some other type |
| */ |
| protected Object getInBody(Exchange exchange) { |
| return exchange.getIn().getBody(); |
| } |
| } |