org.ow2.petals.camel.junit.PetalsCamelTestSupport Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of petals-se-camel-junit Show documentation
Service Engine based on Apache Camel
There is a newer version: 1.3.0
Show newest version
/**
 * Copyright (c) 2017-2020 Linagora
 * 
 * This program/library is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 2.1 of the License, or (at your
 * option) any later version.
 * 
 * This program/library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
 * for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program/library; If not, see http://www.gnu.org/licenses/
 * for the GNU Lesser General Public License version 2.1.
 */
package org.ow2.petals.camel.junit;

import java.util.Collection;

import javax.xml.bind.JAXBException;

import org.apache.camel.Exchange;
import org.apache.camel.builder.AdviceWithRouteBuilder;
import org.apache.camel.component.mock.MockComponent;
import org.apache.camel.component.mock.MockEndpoint;
import org.apache.camel.model.RouteDefinition;
import org.apache.camel.test.junit4.CamelTestSupport;
import org.junit.Before;
import org.ow2.petals.camel.helpers.MarshallingHelper;
import org.ow2.petals.camel.helpers.PetalsRouteBuilder;

public abstract class PetalsCamelTestSupport extends CamelTestSupport {

    private final boolean tracing;

    public PetalsCamelTestSupport() {
        this(false);
    }

    public PetalsCamelTestSupport(boolean tracing) {
        this.tracing = tracing;
    }

    @Override
    public boolean isUseAdviceWith() {
        return true;
    }

    protected abstract Collection routesToMock();

    @Before
    public void mockEndpoints() throws Exception {

        if (tracing) {
            context.setTracing(true);
            context.setStreamCaching(true);
            context.getProperties().put(Exchange.LOG_DEBUG_BODY_STREAMS, Boolean.TRUE.toString());
        }

        context.removeComponent("petals");
        context.addComponent("petals", new MockComponent());
        for (final String from : routesToMock()) {
            RouteDefinition routeDef = context.getRouteDefinition(from);
            assertNotNull("You should set the routeId of one of the routes to '" + from + "' from mocking to work",
                    routeDef);
            routeDef.adviceWith(context, new AdviceWithRouteBuilder() {
                @Override
                public void configure() throws Exception {
                    replaceFromWith("direct:" + from);
                }
            });
        }

        context.start();
    }

    protected String getFrom(String service) {
        return "direct:" + service;
    }

    protected MockEndpoint getTo(String service) {
        return getMockEndpoint("petals:" + service);
    }

    /**
     * Sets the fault on the exchange's out
     */
    protected void setJbiFault(final MarshallingHelper marshalling, final Exchange exchange, final Object fault)
            throws JAXBException {
        marshalling.marshal(exchange.getOut(), fault);
        // set this only after we are sure we properly marshaled the body!
        PetalsRouteBuilder.setIsJbiFault(exchange, false);
    }

    protected void setJbiFault(final Exchange exchange, final Object fault) {
        exchange.getOut().setBody(fault);
        PetalsRouteBuilder.setIsJbiFault(exchange, false);
    }
}