org.jboss.fuse.examples.HL7ConsumerRouteTest.java Source code

Introduction

Here is the source code for org.jboss.fuse.examples.HL7ConsumerRouteTest.java

Source

/*
 * Copyright (C) Red Hat, Inc.
 * http://www.redhat.com
 *
 * Licensed 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.jboss.fuse.examples;

import org.apache.activemq.broker.BrokerService;
import org.apache.camel.CamelContext;
import org.apache.camel.EndpointInject;
import org.apache.camel.Produce;
import org.apache.camel.ProducerTemplate;
import org.apache.camel.builder.NotifyBuilder;
import org.apache.camel.component.mock.MockEndpoint;
import org.apache.camel.test.spring.CamelSpringJUnit4ClassRunner;
import org.apache.camel.util.FileUtil;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;

import java.io.File;
import java.util.concurrent.TimeUnit;

@RunWith(CamelSpringJUnit4ClassRunner.class)
@ContextConfiguration({ "classpath*:/META-INF/spring/*.xml" })
public class HL7ConsumerRouteTest {

    @Autowired(required = true)
    private CamelContext testRoutes;

    @Autowired(required = true)
    private CamelContext consumer1Camel;

    @Produce(uri = "{{camel.activemq.endpoint}}")
    private ProducerTemplate activeMQProducer;

    @EndpointInject(uri = "mock:hl7Mock")
    private MockEndpoint hl7Mock;

    @EndpointInject(uri = "mock:fileMock")
    private MockEndpoint fileMock;

    private static BrokerService broker;

    @BeforeClass
    public static void bootActiveMQ() throws Exception {
        broker = new BrokerService();
        broker.addConnector("tcp://localhost:61616");
        broker.setPersistent(false);
        broker.deleteAllMessages();
        broker.start();
        broker.waitUntilStarted();
    }

    @BeforeClass
    public static void cleanDirectories() {
        File auditDir = new File("./target/audit");
        if (auditDir.exists()) {
            FileUtil.removeDir(auditDir);
        }
        auditDir.mkdir();
    }

    @AfterClass
    public static void stopActiveMQ() throws Exception {
        if (broker != null) {
            broker.stop();
            broker.waitUntilStopped();
        }
    }

    private String createValidHl7Message() {
        StringBuilder hl7Message = new StringBuilder();
        hl7Message.append("MSH|^~\\&|FUSEDEMO|ORG|TEST|JBOSS|20061019172719||ADT^A01^ADT_A01|MSGID12349876|P|2.4")
                .append("\r");
        hl7Message.append(
                "PID|||20301||Durden^Tyler^^^Mr.||19700312|M|||88 Punchward Dr.^^Los Angeles^CA^11221^USA|||||||")
                .append("\r");
        hl7Message.append(
                "PV1||O|OP^^||||4652^Paulson^Robert|||OP|||||||||9|||||||||||||||||||||||||20061019172717|20061019172718")
                .append("\r");
        return hl7Message.toString();
    }

    @Test
    public void testSendHL7ToActiveMQMessage() throws Exception {
        hl7Mock.expectedMessageCount(1);
        fileMock.expectedMessageCount(1);
        NotifyBuilder notify = new NotifyBuilder(consumer1Camel).whenCompleted(2).create();
        activeMQProducer.sendBody(createValidHl7Message());
        notify.matches(2, TimeUnit.SECONDS);

        MockEndpoint.assertIsSatisfied(testRoutes, 2, TimeUnit.SECONDS);
        hl7Mock.message(1).body().contains("FUSEDEMO|ORG|TEST");
    }

}