examples/camel-example-transformer-demo/src/test/java/org/apache/camel/example/transformer/OrderRouteSpringTest.java

/**
 * 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.example.transformer;

import com.fasterxml.jackson.databind.ObjectMapper;
import org.apache.camel.EndpointInject;
import org.apache.camel.Exchange;
import org.apache.camel.Processor;
import org.apache.camel.Produce;
import org.apache.camel.ProducerTemplate;
import org.apache.camel.component.mock.MockEndpoint;
import org.apache.camel.example.transformer.demo.Order;
import org.apache.camel.example.transformer.demo.OrderResponse;
import org.apache.camel.spi.DataType;
import org.apache.camel.spi.DataTypeAware;
import org.apache.camel.test.spring.CamelSpringDelegatingTestContextLoader;
import org.apache.camel.test.spring.CamelSpringRunner;
import org.apache.camel.test.spring.MockEndpointsAndSkip;
import org.custommonkey.xmlunit.XMLUnit;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.test.context.ContextConfiguration;

import static org.junit.Assert.assertEquals;

@RunWith(CamelSpringRunner.class)
@ContextConfiguration(value = "/META-INF/spring/camel-context.xml", loader = CamelSpringDelegatingTestContextLoader.class)
@MockEndpointsAndSkip("direct:csv")
public class OrderRouteSpringTest {
    @Produce(uri = "direct:java")
    protected ProducerTemplate javaProducer;

    @Produce(uri = "direct:xml")
    protected ProducerTemplate xmlProducer;

    @Produce(uri = "direct:json")
    protected ProducerTemplate jsonProducer;

    @EndpointInject(uri = "mock:direct:csv")
    private MockEndpoint mockCsv;

    @Before
    public void before() {
        mockCsv.reset();
    }

    @Test
    public void testJava() throws Exception {
        mockCsv.whenAnyExchangeReceived(new Processor() {
            public void process(Exchange exchange) {
                Object mockBody = exchange.getIn().getBody();
                assertEquals(Order.class, mockBody.getClass());
                Order mockOrder = (Order)mockBody;
                assertEquals("Order-Java-0001", mockOrder.getOrderId());
                assertEquals("MILK", mockOrder.getItemId());
                assertEquals(3, mockOrder.getQuantity());
            }
        });
        mockCsv.setExpectedMessageCount(1);

        Order order = new Order()
            .setOrderId("Order-Java-0001")
            .setItemId("MILK")
            .setQuantity(3);
        Object answer = javaProducer.requestBody(order);
        assertEquals(OrderResponse.class, answer.getClass());
        OrderResponse or = (OrderResponse)answer;
        assertEquals("Order-Java-0001", or.getOrderId());
        assertEquals(true, or.isAccepted());
        assertEquals("Order accepted:[item='MILK' quantity='3']", or.getDescription());
        mockCsv.assertIsSatisfied();
    }

    @Test
    public void testXML() throws Exception {
        mockCsv.whenAnyExchangeReceived(new Processor() {
            public void process(Exchange exchange) {
                Object mockBody = exchange.getIn().getBody();
                assertEquals(Order.class, mockBody.getClass());
                Order mockOrder = (Order)mockBody;
                assertEquals("Order-XML-0001", mockOrder.getOrderId());
                assertEquals("MIKAN", mockOrder.getItemId());
                assertEquals(365, mockOrder.getQuantity());
            }
        });
        mockCsv.setExpectedMessageCount(1);

        String order = "<order orderId=\"Order-XML-0001\" itemId=\"MIKAN\" quantity=\"365\"/>";
        String expectedAnswer = "<orderResponse orderId=\"Order-XML-0001\" accepted=\"true\" description=\"Order accepted:[item='MIKAN' quantity='365']\"/>";
        Exchange answer = xmlProducer.send("direct:xml", ex -> {
            ((DataTypeAware)ex.getIn()).setBody(order, new DataType("xml:XMLOrder"));
        });
        XMLUnit.compareXML(expectedAnswer, answer.getOut().getBody(String.class));
        mockCsv.assertIsSatisfied();
    }

    @Test
    public void testJSON() throws Exception {
        mockCsv.whenAnyExchangeReceived(new Processor() {
            public void process(Exchange exchange) {
                Object mockBody = exchange.getIn().getBody();
                assertEquals(Order.class, mockBody.getClass());
                Order mockOrder = (Order)mockBody;
                assertEquals("Order-JSON-0001", mockOrder.getOrderId());
                assertEquals("MIZUYO-KAN", mockOrder.getItemId());
                assertEquals(16350, mockOrder.getQuantity());
            }
        });
        mockCsv.setExpectedMessageCount(1);

        String order = "{\"orderId\":\"Order-JSON-0001\", \"itemId\":\"MIZUYO-KAN\", \"quantity\":\"16350\"}";
        OrderResponse expected = new OrderResponse()
            .setAccepted(true)
            .setOrderId("Order-JSON-0001")
            .setDescription("Order accepted:[item='MIZUYO-KAN' quantity='16350']");
        ObjectMapper jsonMapper = new ObjectMapper();
        String expectedJson = jsonMapper.writeValueAsString(expected);
        Exchange answer = jsonProducer.send("direct:json", ex -> {
            ((DataTypeAware)ex.getIn()).setBody(order, new DataType("json"));
        });
        assertEquals(expectedJson, answer.getOut().getBody(String.class));
        mockCsv.assertIsSatisfied();
    }
}