ru.yandex.qatools.camelot.test.service.TestHelperImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of camelot-test Show documentation
The newest version!
package ru.yandex.qatools.camelot.test.service;

import org.apache.camel.CamelContext;
import org.apache.camel.CamelContextAware;
import org.apache.camel.ProducerTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import ru.yandex.qatools.camelot.common.ProcessingEngine;
import ru.yandex.qatools.camelot.config.Plugin;
import ru.yandex.qatools.camelot.test.TestHelper;

import java.util.Map;

import static ru.yandex.qatools.camelot.util.MapUtil.map;

/**
 * @author Ilya Sadykov (mailto: [email protected])
 * @author Innokenty Shuvalov (mailto: [email protected])
 */
public class TestHelperImpl implements CamelContextAware, TestHelper {

    @Autowired
    ProcessingEngine pluginsService;

    ProducerTemplate producerTemplate;
    CamelContext camelContext;

    @Override
    public void sendTo(Class pluginClass, Object event) {
        pluginsService.getPluginContext(pluginClass).getInput().produce(event);
    }

    @Override
    public void sendTo(String pluginId, Object event) {
        pluginsService.getPluginContext(pluginId).getInput().produce(event);
    }

    @Override
    public void sendTo(Class pluginClass, Object event, Map headers) {
        pluginsService.getPluginContext(pluginClass).getInput().produce(event, headers);
    }

    @Override
    public void sendTo(Class pluginClass, Object event, String header, Object headerValue) {
        sendTo(pluginClass, event, map(header, headerValue));
    }

    @Override
    public void sendTo(String pluginId, Object event, Map headers) {
        pluginsService.getPluginContext(pluginId).getInput().produce(event, headers);
    }

    @Override
    public void sendTo(String pluginId, Object event, String header, Object headerValue) {
        sendTo(pluginId, event, map(header, headerValue));
    }

    @Override
    public void send(Object event) {
        pluginsService.getMainInput().produce(event);
    }

    @Override
    public void send(Object event, Map headers) {
        pluginsService.getMainInput().produce(event, headers);
    }

    @Override
    public void send(Object event, String header, Object headerValue) {
        send(event, map(header, headerValue));
    }

    @Override
    public void invokeTimersFor(Class pluginClass) {
        invokeTimers(pluginsService.getPlugin(pluginClass));
    }

    @Override
    public void invokeTimersFor(String pluginId) {
        invokeTimers(pluginsService.getPlugin(pluginId));
    }

    @Override
    public void invokeTimers(Plugin plugin){
        try {
            pluginsService.getBuildersFactory().newSchedulerBuildersFactory(
                    pluginsService.getScheduler(),
                    pluginsService.getCamelContext()
            ).build(plugin).invokeJobs();
        } catch (Exception e) {
            // that's an error of the test, rethrowing exception
            throw new RuntimeException(e);
        }
    }

    @Override
    public void invokeTimerFor(Class pluginClass, String method) {
        invokeTimer(pluginsService.getPlugin(pluginClass), method);
    }

    @Override
    public void invokeTimerFor(String pluginId, String method) {
        invokeTimer(pluginsService.getPlugin(pluginId), method);
    }

    @Override
    public void invokeTimer(Plugin plugin, String method) {
        try {
            pluginsService.getBuildersFactory().newSchedulerBuildersFactory(
                    pluginsService.getScheduler(),
                    pluginsService.getCamelContext()
            ).build(plugin).invokeJob(method);
        } catch (Exception e) {
            // that's an error of the test, rethrowing exception
            throw new RuntimeException(e);
        }
    }

    @Override
    public void setCamelContext(CamelContext camelContext) {
        this.camelContext = camelContext;
        producerTemplate = camelContext.createProducerTemplate();
        producerTemplate.setDefaultEndpointUri(pluginsService.getInputUri());
    }

    @Override
    public CamelContext getCamelContext() {
        return camelContext;
    }
}