You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@camel.apache.org by da...@apache.org on 2016/12/15 19:20:32 UTC
[3/6] camel git commit: CAMEL-10511: Updated MllpTcpClientProducer
and MllpTcpServerConsumer to consume all available data on socket
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerConsumerMessageValidationTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerConsumerMessageValidationTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerConsumerMessageValidationTest.java
new file mode 100644
index 0000000..5c431e6
--- /dev/null
+++ b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerConsumerMessageValidationTest.java
@@ -0,0 +1,317 @@
+/**
+ * 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.mllp;
+
+import org.apache.camel.CamelContext;
+import org.apache.camel.EndpointInject;
+import org.apache.camel.LoggingLevel;
+import org.apache.camel.builder.NotifyBuilder;
+import org.apache.camel.builder.RouteBuilder;
+import org.apache.camel.component.mock.MockEndpoint;
+import org.apache.camel.impl.DefaultCamelContext;
+import org.apache.camel.test.AvailablePortFinder;
+import org.apache.camel.test.junit.rule.mllp.MllpClientResource;
+import org.apache.camel.test.junit.rule.mllp.MllpJUnitResourceException;
+import org.apache.camel.test.junit.rule.mllp.MllpJUnitResourceTimeoutException;
+import org.apache.camel.test.junit4.CamelTestSupport;
+import org.junit.Ignore;
+import org.junit.Rule;
+import org.junit.Test;
+
+import java.util.concurrent.TimeUnit;
+
+import static org.apache.camel.component.mllp.MllpEndpoint.END_OF_BLOCK;
+import static org.apache.camel.component.mllp.MllpEndpoint.START_OF_BLOCK;
+import static org.apache.camel.test.mllp.Hl7MessageGenerator.generateMessage;
+
+public class MllpTcpServerConsumerMessageValidationTest extends CamelTestSupport {
+ @Rule
+ public MllpClientResource mllpClient = new MllpClientResource();
+
+ @EndpointInject(uri = "mock://result")
+ MockEndpoint result;
+
+ @EndpointInject(uri = "mock://timeout-ex")
+ MockEndpoint timeout;
+
+ @EndpointInject(uri = "mock://invalid-ex")
+ MockEndpoint invalid;
+
+ @Override
+ protected CamelContext createCamelContext() throws Exception {
+ DefaultCamelContext context = (DefaultCamelContext) super.createCamelContext();
+
+ context.setUseMDCLogging(true);
+ context.setName(this.getClass().getSimpleName());
+
+ return context;
+ }
+
+ @Override
+ protected RouteBuilder createRouteBuilder() {
+
+ mllpClient.setMllpHost("localhost");
+ mllpClient.setMllpPort(AvailablePortFinder.getNextAvailable());
+
+ return new RouteBuilder() {
+ int connectTimeout = 500;
+ int responseTimeout = 5000;
+
+ @Override
+ public void configure() throws Exception {
+ String routeId = "mllp-test-receiver-route";
+
+ onException(MllpTimeoutException.class)
+ .to(timeout);
+
+ onException(MllpInvalidMessageException.class)
+ .to(invalid);
+
+ fromF("mllp://%s:%d?validatePayload=true&autoAck=true&connectTimeout=%d&receiveTimeout=%d",
+ mllpClient.getMllpHost(), mllpClient.getMllpPort(), connectTimeout, responseTimeout)
+ .routeId(routeId)
+ .log(LoggingLevel.INFO, routeId, "Test route received message")
+ .to(result);
+
+ }
+ };
+ }
+
+ @Test
+ public void testReceiveSingleMessage() throws Exception {
+ result.expectedMessageCount(1);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(0);
+ invalid.expectedMessageCount(0);
+
+ mllpClient.connect();
+
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(), 10000);
+
+ assertMockEndpointsSatisfied(10, TimeUnit.SECONDS);
+ }
+
+ @Test
+ public void testReceiveSingleMessageWithDelayAfterConnection() throws Exception {
+ result.expectedMinimumMessageCount(1);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(0);
+ invalid.expectedMessageCount(0);
+
+ mllpClient.connect();
+
+ Thread.sleep(5000);
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(), 10000);
+
+ assertMockEndpointsSatisfied(10, TimeUnit.SECONDS);
+ }
+
+ @Test
+ public void testReceiveMultipleMessages() throws Exception {
+ int sendMessageCount = 5;
+ result.expectedMinimumMessageCount(5);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(0);
+ invalid.expectedMessageCount(0);
+
+ mllpClient.connect();
+
+ for (int i = 1; i <= sendMessageCount; ++i) {
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(i));
+ }
+
+ assertMockEndpointsSatisfied(10, TimeUnit.SECONDS);
+ }
+
+ @Test
+ public void testOpenMllpEnvelopeWithReset() throws Exception {
+ result.expectedMessageCount(4);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(1);
+ invalid.expectedMessageCount(0);
+
+ NotifyBuilder notify1 = new NotifyBuilder(context).whenDone(2).create();
+ NotifyBuilder notify2 = new NotifyBuilder(context).whenDone(5).create();
+
+ mllpClient.connect();
+ mllpClient.setSoTimeout(10000);
+
+ log.info("Sending TEST_MESSAGE_1");
+ String acknowledgement1 = mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(1));
+
+ log.info("Sending TEST_MESSAGE_2");
+ String acknowledgement2 = mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(2));
+
+ assertTrue("First two normal exchanges did not complete", notify1.matches(10, TimeUnit.SECONDS));
+
+ log.info("Sending TEST_MESSAGE_3");
+ mllpClient.setSendEndOfBlock(false);
+ mllpClient.setSendEndOfData(false);
+ // Acknowledgement won't come here
+ try {
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(3));
+ } catch (MllpJUnitResourceException resourceEx) {
+ log.info("Expected exception reading response");
+ }
+ mllpClient.disconnect();
+ Thread.sleep(1000);
+ mllpClient.connect();
+
+ log.info("Sending TEST_MESSAGE_4");
+ mllpClient.setSendEndOfBlock(true);
+ mllpClient.setSendEndOfData(true);
+ String acknowledgement4 = mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(4));
+
+ log.info("Sending TEST_MESSAGE_5");
+ String acknowledgement5 = mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(5));
+
+ assertTrue("Remaining exchanges did not complete", notify2.matches(10, TimeUnit.SECONDS));
+
+ assertMockEndpointsSatisfied(10, TimeUnit.SECONDS);
+
+ assertTrue("Should be acknowledgment for message 1", acknowledgement1.contains("MSA|AA|00001"));
+ assertTrue("Should be acknowledgment for message 2", acknowledgement2.contains("MSA|AA|00002"));
+ // assertTrue("Should be acknowledgment for message 3", acknowledgement3.contains("MSA|AA|00003"));
+ assertTrue("Should be acknowledgment for message 4", acknowledgement4.contains("MSA|AA|00004"));
+ assertTrue("Should be acknowledgment for message 5", acknowledgement5.contains("MSA|AA|00005"));
+ }
+
+ @Test
+ public void testMessageReadTimeout() throws Exception {
+ result.expectedMessageCount(0);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(1);
+ invalid.expectedMessageCount(0);
+
+ NotifyBuilder notify = new NotifyBuilder(context).whenDone(1).create();
+
+ mllpClient.setSendEndOfBlock(false);
+ mllpClient.setSendEndOfData(false);
+
+ mllpClient.sendFramedData(generateMessage());
+
+ assertTrue("One exchange should have completed", notify.matches(15, TimeUnit.SECONDS));
+
+ assertMockEndpointsSatisfied();
+ }
+
+ @Test
+ public void testInvalidMessage() throws Exception {
+ result.expectedMessageCount(0);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(0);
+ invalid.expectedMessageCount(1);
+
+ mllpClient.sendFramedData("INVALID PAYLOAD");
+
+ assertMockEndpointsSatisfied(15, TimeUnit.SECONDS);
+ }
+
+ @Test
+ public void testNthInvalidMessage() throws Exception {
+ int messageCount = 10;
+
+ result.expectedMessageCount(messageCount - 1);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(0);
+ invalid.expectedMessageCount(1);
+
+ for (int i = 0; i < messageCount; ++i) {
+ if (i == messageCount / 2) {
+ try {
+ mllpClient.sendMessageAndWaitForAcknowledgement("INVALID PAYLOAD");
+ fail("An acknowledgement should not be received for an invalid HL7 message");
+ } catch (MllpJUnitResourceTimeoutException timeoutEx) {
+ // expected - eat this
+ }
+ } else {
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(i + 1));
+ }
+ }
+
+ assertMockEndpointsSatisfied();
+ }
+
+ @Test
+ public void testMessageContainingEmbeddedStartOfBlock() throws Exception {
+ result.expectedMessageCount(0);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(0);
+ invalid.expectedMessageCount(1);
+
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage().replaceFirst("EVN", "EVN" + START_OF_BLOCK));
+
+ assertMockEndpointsSatisfied();
+ }
+
+ @Test
+ public void testNthMessageContainingEmbeddedStartOfBlock() throws Exception {
+ int messageCount = 10;
+
+ result.expectedMessageCount(messageCount - 1);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(0);
+ invalid.expectedMessageCount(1);
+
+ for (int i = 0; i < messageCount; ++i) {
+ String message = (i == (messageCount/2))
+ ? generateMessage(i + 1).replaceFirst("EVN", "EVN" + START_OF_BLOCK)
+ : generateMessage(i + 1);
+
+ log.debug("Sending message {}", MllpComponent.covertToPrintFriendlyString(message));
+
+ mllpClient.sendMessageAndWaitForAcknowledgement(message);
+ }
+
+ assertMockEndpointsSatisfied(15, TimeUnit.SECONDS);
+ }
+
+ @Test
+ public void testMessageContainingEmbeddedEndOfBlock() throws Exception {
+ result.expectedMessageCount(0);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(0);
+ invalid.expectedMessageCount(1);
+
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage().replaceFirst("EVN", "EVN" + END_OF_BLOCK));
+
+ assertMockEndpointsSatisfied();
+ }
+
+ @Test
+ public void testInvalidMessageContainingEmbeddedEndOfBlock() throws Exception {
+ int messageCount = 10;
+
+ result.expectedMessageCount(messageCount - 1);
+ result.setAssertPeriod(1000);
+ timeout.expectedMessageCount(0);
+ invalid.expectedMessageCount(1);
+
+ for (int i = 0; i < messageCount; ++i) {
+ String message = (i == (messageCount/2))
+ ? generateMessage(i + 1).replaceFirst("EVN", "EVN" + END_OF_BLOCK)
+ : generateMessage(i + 1);
+
+ log.debug("Sending message {}", MllpComponent.covertToPrintFriendlyString(message));
+
+ mllpClient.sendMessageAndWaitForAcknowledgement(message);
+ }
+
+ assertMockEndpointsSatisfied(15, TimeUnit.SECONDS);
+ }
+}
+
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerConsumerTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerConsumerTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerConsumerTest.java
index c77c35f..4060626 100644
--- a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerConsumerTest.java
+++ b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerConsumerTest.java
@@ -16,8 +16,6 @@
*/
package org.apache.camel.component.mllp;
-import java.util.concurrent.TimeUnit;
-
import org.apache.camel.CamelContext;
import org.apache.camel.EndpointInject;
import org.apache.camel.LoggingLevel;
@@ -28,10 +26,15 @@ import org.apache.camel.impl.DefaultCamelContext;
import org.apache.camel.test.AvailablePortFinder;
import org.apache.camel.test.junit.rule.mllp.MllpClientResource;
import org.apache.camel.test.junit.rule.mllp.MllpJUnitResourceException;
+import org.apache.camel.test.junit.rule.mllp.MllpJUnitResourceTimeoutException;
import org.apache.camel.test.junit4.CamelTestSupport;
import org.junit.Rule;
import org.junit.Test;
+import java.util.concurrent.TimeUnit;
+
+import static org.apache.camel.component.mllp.MllpEndpoint.END_OF_BLOCK;
+import static org.apache.camel.component.mllp.MllpEndpoint.START_OF_BLOCK;
import static org.apache.camel.test.mllp.Hl7MessageGenerator.generateMessage;
public class MllpTcpServerConsumerTest extends CamelTestSupport {
@@ -41,6 +44,9 @@ public class MllpTcpServerConsumerTest extends CamelTestSupport {
@EndpointInject(uri = "mock://result")
MockEndpoint result;
+ @EndpointInject(uri = "mock://timeout-ex")
+ MockEndpoint timeout;
+
@Override
protected CamelContext createCamelContext() throws Exception {
DefaultCamelContext context = (DefaultCamelContext) super.createCamelContext();
@@ -65,9 +71,8 @@ public class MllpTcpServerConsumerTest extends CamelTestSupport {
public void configure() throws Exception {
String routeId = "mllp-test-receiver-route";
- onCompletion()
- .toF("log:%s?level=INFO&showAll=true", routeId)
- .log(LoggingLevel.INFO, routeId, "Test route complete");
+ onException(MllpTimeoutException.class)
+ .to(timeout);
fromF("mllp://%s:%d?autoAck=true&connectTimeout=%d&receiveTimeout=%d",
mllpClient.getMllpHost(), mllpClient.getMllpPort(), connectTimeout, responseTimeout)
@@ -82,6 +87,7 @@ public class MllpTcpServerConsumerTest extends CamelTestSupport {
@Test
public void testReceiveSingleMessage() throws Exception {
result.expectedMessageCount(1);
+ timeout.expectedMessageCount(0);
mllpClient.connect();
@@ -93,6 +99,7 @@ public class MllpTcpServerConsumerTest extends CamelTestSupport {
@Test
public void testReceiveSingleMessageWithDelayAfterConnection() throws Exception {
result.expectedMinimumMessageCount(1);
+ timeout.expectedMessageCount(0);
mllpClient.connect();
@@ -106,6 +113,7 @@ public class MllpTcpServerConsumerTest extends CamelTestSupport {
public void testReceiveMultipleMessages() throws Exception {
int sendMessageCount = 5;
result.expectedMinimumMessageCount(5);
+ timeout.expectedMessageCount(0);
mllpClient.connect();
@@ -119,6 +127,7 @@ public class MllpTcpServerConsumerTest extends CamelTestSupport {
@Test
public void testOpenMllpEnvelopeWithReset() throws Exception {
result.expectedMessageCount(4);
+ timeout.expectedMessageCount(1);
NotifyBuilder notify1 = new NotifyBuilder(context).whenDone(2).create();
NotifyBuilder notify2 = new NotifyBuilder(context).whenDone(5).create();
@@ -165,5 +174,114 @@ public class MllpTcpServerConsumerTest extends CamelTestSupport {
assertTrue("Should be acknowledgment for message 5", acknowledgement5.contains("MSA|AA|00005"));
}
+ @Test
+ public void testMessageReadTimeout() throws Exception {
+ result.expectedMessageCount(0);
+ timeout.expectedMessageCount(1);
+
+ NotifyBuilder notify = new NotifyBuilder(context).whenDone(1).create();
+
+ mllpClient.setSendEndOfBlock(false);
+ mllpClient.setSendEndOfData(false);
+
+ mllpClient.sendFramedData(generateMessage());
+
+ assertTrue("One exchange should have completed", notify.matches(15, TimeUnit.SECONDS));
+
+ assertMockEndpointsSatisfied();
+ }
+
+ @Test
+ public void testInvalidMessage() throws Exception {
+ result.expectedMessageCount(1);
+ timeout.expectedMessageCount(0);
+
+ mllpClient.sendFramedData("INVALID PAYLOAD");
+
+ assertMockEndpointsSatisfied(15, TimeUnit.SECONDS);
+ }
+
+ @Test
+ public void testNthInvalidMessage() throws Exception {
+ int messageCount = 10;
+
+ result.expectedMessageCount(messageCount);
+ timeout.expectedMessageCount(0);
+
+ for (int i = 0; i < messageCount; ++i) {
+ if (i == messageCount / 2) {
+ try {
+ mllpClient.sendMessageAndWaitForAcknowledgement("INVALID PAYLOAD");
+ fail("An acknowledgement should not be received for an invalid HL7 message");
+ } catch (MllpJUnitResourceTimeoutException timeoutEx) {
+ // expected - eat this
+ }
+ } else {
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage(i + 1));
+ }
+ }
+
+ assertMockEndpointsSatisfied(15, TimeUnit.SECONDS);
+ }
+
+ @Test
+ public void testMessageContainingEmbeddedStartOfBlock() throws Exception {
+ result.expectedMessageCount(1);
+ timeout.expectedMessageCount(0);
+
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage().replaceFirst("EVN", "EVN" + START_OF_BLOCK));
+
+ assertMockEndpointsSatisfied();
+ }
+
+ @Test
+ public void testNthMessageContainingEmbeddedStartOfBlock() throws Exception {
+ int messageCount = 10;
+
+ result.expectedMessageCount(messageCount);
+ timeout.expectedMessageCount(0);
+
+ for (int i = 0; i < messageCount; ++i) {
+ String message = (i == (messageCount/2))
+ ? generateMessage(i + 1).replaceFirst("EVN", "EVN" + START_OF_BLOCK)
+ : generateMessage(i + 1);
+
+ log.debug("Sending message {}", MllpComponent.covertToPrintFriendlyString(message));
+
+ mllpClient.sendMessageAndWaitForAcknowledgement(message);
+ }
+
+ assertMockEndpointsSatisfied(15, TimeUnit.SECONDS);
+ }
+
+ @Test
+ public void testMessageContainingEmbeddedEndOfBlock() throws Exception {
+ result.expectedMessageCount(1);
+ timeout.expectedMessageCount(0);
+
+ mllpClient.sendMessageAndWaitForAcknowledgement(generateMessage().replaceFirst("EVN", "EVN" + END_OF_BLOCK));
+
+ assertMockEndpointsSatisfied();
+ }
+
+ @Test
+ public void testInvalidMessageContainingEmbeddedEndOfBlock() throws Exception {
+ int messageCount = 10;
+
+ result.expectedMessageCount(messageCount);
+ timeout.expectedMessageCount(0);
+
+ for (int i = 0; i < messageCount; ++i) {
+ String message = (i == (messageCount/2))
+ ? generateMessage(i + 1).replaceFirst("EVN", "EVN" + END_OF_BLOCK)
+ : generateMessage(i + 1);
+
+ log.debug("Sending message {}", MllpComponent.covertToPrintFriendlyString(message));
+
+ mllpClient.sendMessageAndWaitForAcknowledgement(message);
+ }
+
+ assertMockEndpointsSatisfied(15, TimeUnit.SECONDS);
+ }
}
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerProducerBlueprintTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerProducerBlueprintTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerProducerBlueprintTest.java
deleted file mode 100644
index dc796a5..0000000
--- a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTcpServerProducerBlueprintTest.java
+++ /dev/null
@@ -1,35 +0,0 @@
-/**
- * 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.mllp;
-
-import org.apache.camel.EndpointInject;
-import org.apache.camel.component.mock.MockEndpoint;
-import org.apache.camel.test.blueprint.CamelBlueprintTestSupport;
-import org.junit.Ignore;
-
-@Ignore(value = "Not Yet Implemented")
-// TODO: Implement this
-public class MllpTcpServerProducerBlueprintTest extends CamelBlueprintTestSupport {
- @EndpointInject(uri = "mock://target")
- MockEndpoint target;
-
- @Override
- protected String getBlueprintDescriptor() {
- return "OSGI-INF/blueprint/mllp-tcp-server-producer.xml";
- }
-
-}
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTimeoutExceptionTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTimeoutExceptionTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTimeoutExceptionTest.java
deleted file mode 100644
index da4ceb7..0000000
--- a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpTimeoutExceptionTest.java
+++ /dev/null
@@ -1,101 +0,0 @@
-/**
- * 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.mllp;
-
-import org.junit.After;
-import org.junit.Before;
-import org.junit.Test;
-
-import static org.junit.Assert.assertEquals;
-
-public class MllpTimeoutExceptionTest {
- static final String EXCEPTION_MESSAGE = "Test Timeoute Exception";
-
- static final String HL7_MESSAGE =
- "MSH|^~\\&|APP_A|FAC_A|^org^sys||||ADT^A04^ADT_A04|||2.6" + '\r'
- + "PID|1||1100832^^^^PI||TEST^FIG||98765432|U||R|435 MAIN STREET^^LONGMONT^CO^80503||123-456-7890|||S" + '\r'
- + '\r' + '\n';
-
- static final String EXCEPTION_MESSAGE_WITH_LOG_PHI_DISABLED = EXCEPTION_MESSAGE;
- static final String EXCEPTION_MESSAGE_WITH_LOG_PHI_ENABLED =
- String.format(String.format("%s:\n\tHL7 Message: %s",
- EXCEPTION_MESSAGE,
- new String(HL7_MESSAGE).replaceAll("\r", "<CR>").replaceAll("\n", "<LF>"))
- );
-
- Exception exception;
-
- @Before
- public void setUp() throws Exception {
- exception = new MllpTimeoutException(EXCEPTION_MESSAGE, HL7_MESSAGE.getBytes());
- }
-
- @After
- public void tearDown() throws Exception {
- System.clearProperty(MllpComponent.MLLP_LOG_PHI_PROPERTY);
- }
-
- @Test
- public void testLogPhiDefault() throws Exception {
- String exceptionMessage = exception.getMessage();
-
- assertEquals(EXCEPTION_MESSAGE_WITH_LOG_PHI_ENABLED, exceptionMessage);
- }
-
- @Test
- public void testLogPhiDisabled() throws Exception {
- System.setProperty(MllpComponent.MLLP_LOG_PHI_PROPERTY, "false");
-
- String exceptionMessage = exception.getMessage();
-
- assertEquals(EXCEPTION_MESSAGE_WITH_LOG_PHI_DISABLED, exceptionMessage);
- }
-
- @Test
- public void testLogPhiEnabled() throws Exception {
- System.setProperty(MllpComponent.MLLP_LOG_PHI_PROPERTY, "true");
-
- String exceptionMessage = exception.getMessage();
-
- assertEquals(EXCEPTION_MESSAGE_WITH_LOG_PHI_ENABLED, exceptionMessage);
- }
-
- @Test
- public void testNullPayload() throws Exception {
- final String expectedMessage = String.format("%s:\n\tHL7 Message: null", EXCEPTION_MESSAGE);
-
- exception = new MllpTimeoutException(EXCEPTION_MESSAGE, null);
-
- System.setProperty(MllpComponent.MLLP_LOG_PHI_PROPERTY, "true");
- String exceptionMessage = exception.getMessage();
-
- assertEquals(expectedMessage, exceptionMessage);
- }
-
- @Test
- public void testToString() throws Exception {
- final String expectedString =
- "org.apache.camel.component.mllp.MllpTimeoutException: "
- + "{hl7Message="
- + "MSH|^~\\&|APP_A|FAC_A|^org^sys||||ADT^A04^ADT_A04|||2.6<CR>"
- + "PID|1||1100832^^^^PI||TEST^FIG||98765432|U||R|435 MAIN STREET^^LONGMONT^CO^80503||123-456-7890|||S<CR><CR><LF>"
- + "}";
-
- assertEquals(expectedString, exception.toString());
- }
-
-}
\ No newline at end of file
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpWriteExceptionTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpWriteExceptionTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpWriteExceptionTest.java
deleted file mode 100644
index daf7fe9..0000000
--- a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/MllpWriteExceptionTest.java
+++ /dev/null
@@ -1,101 +0,0 @@
-/**
- * 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.mllp;
-
-import org.junit.After;
-import org.junit.Before;
-import org.junit.Test;
-
-import static org.junit.Assert.assertEquals;
-
-public class MllpWriteExceptionTest {
- static final String EXCEPTION_MESSAGE = "Test Write Exception";
-
- static final String HL7_MESSAGE =
- "MSH|^~\\&|APP_A|FAC_A|^org^sys||||ADT^A04^ADT_A04|||2.6" + '\r'
- + "PID|1||1100832^^^^PI||TEST^FIG||98765432|U||R|435 MAIN STREET^^LONGMONT^CO^80503||123-456-7890|||S" + '\r'
- + '\r' + '\n';
-
- static final String EXCEPTION_MESSAGE_WITH_LOG_PHI_DISABLED = EXCEPTION_MESSAGE;
- static final String EXCEPTION_MESSAGE_WITH_LOG_PHI_ENABLED =
- String.format(String.format("%s:\n\tMLLP Payload: %s",
- EXCEPTION_MESSAGE,
- new String(HL7_MESSAGE).replaceAll("\r", "<CR>").replaceAll("\n", "<LF>"))
- );
-
- Exception exception;
-
- @Before
- public void setUp() throws Exception {
- exception = new MllpWriteException(EXCEPTION_MESSAGE, HL7_MESSAGE.getBytes());
- }
-
- @After
- public void tearDown() throws Exception {
- System.clearProperty(MllpComponent.MLLP_LOG_PHI_PROPERTY);
- }
-
- @Test
- public void testLogPhiDefault() throws Exception {
- String exceptionMessage = exception.getMessage();
-
- assertEquals(EXCEPTION_MESSAGE_WITH_LOG_PHI_ENABLED, exceptionMessage);
- }
-
- @Test
- public void testLogPhiDisabled() throws Exception {
- System.setProperty(MllpComponent.MLLP_LOG_PHI_PROPERTY, "false");
-
- String exceptionMessage = exception.getMessage();
-
- assertEquals(EXCEPTION_MESSAGE_WITH_LOG_PHI_DISABLED, exceptionMessage);
- }
-
- @Test
- public void testLogPhiEnabled() throws Exception {
- System.setProperty(MllpComponent.MLLP_LOG_PHI_PROPERTY, "true");
-
- String exceptionMessage = exception.getMessage();
-
- assertEquals(EXCEPTION_MESSAGE_WITH_LOG_PHI_ENABLED, exceptionMessage);
- }
-
- @Test
- public void testNullPayload() throws Exception {
- final String expectedMessage = String.format("%s:\n\tMLLP Payload: null", EXCEPTION_MESSAGE);
-
- exception = new MllpWriteException(EXCEPTION_MESSAGE, null);
-
- System.setProperty(MllpComponent.MLLP_LOG_PHI_PROPERTY, "true");
- String exceptionMessage = exception.getMessage();
-
- assertEquals(expectedMessage, exceptionMessage);
- }
-
- @Test
- public void testToString() throws Exception {
- final String expectedString =
- "org.apache.camel.component.mllp.MllpWriteException: "
- + "{mllpPayload="
- + "MSH|^~\\&|APP_A|FAC_A|^org^sys||||ADT^A04^ADT_A04|||2.6<CR>"
- + "PID|1||1100832^^^^PI||TEST^FIG||98765432|U||R|435 MAIN STREET^^LONGMONT^CO^80503||123-456-7890|||S<CR><CR><LF>"
- + "}";
-
- assertEquals(expectedString, exception.toString());
- }
-
-}
\ No newline at end of file
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/Hl7UtilTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/Hl7UtilTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/Hl7UtilTest.java
new file mode 100644
index 0000000..e7d443c
--- /dev/null
+++ b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/Hl7UtilTest.java
@@ -0,0 +1,126 @@
+/**
+ * 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.mllp.impl;
+
+import java.io.ByteArrayOutputStream;
+
+import org.junit.Test;
+
+import static org.apache.camel.component.mllp.MllpEndpoint.END_OF_BLOCK;
+import static org.apache.camel.component.mllp.MllpEndpoint.START_OF_BLOCK;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+
+public class Hl7UtilTest {
+ static final String TEST_MESSAGE =
+ "MSH|^~\\&|ADT|EPIC|JCAPS|CC|20161206193919|RISTECH|ADT^A08|00001|D|2.3^^|||||||" + '\r'
+ + "EVN|A08|20150107161440||REG_UPDATE_SEND_VISIT_MESSAGES_ON_PATIENT_CHANGES|RISTECH^RADIOLOGY^TECHNOLOGIST^^^^^^UCLA^^^^^RRMC||" + '\r'
+ + "PID|1|2100355^^^MRN^MRN|2100355^^^MRN^MRN||MDCLS9^MC9||19700109|F||U|111 HOVER STREET^^LOS ANGELES^CA^90032^USA^P^^LOS ANGELE|"
+ + "LOS ANGELE|(310)725-6952^P^PH^^^310^7256952||ENGLISH|U||60000013647|565-33-2222|||U||||||||N||" + '\r'
+ + "PD1|||UCLA HEALTH SYSTEM^^10|10002116^ADAMS^JOHN^D^^^^^EPIC^^^^PROVID||||||||||||||" + '\r'
+ + "NK1|1|DOE^MC9^^|OTH|^^^^^USA|(310)888-9999^^^^^310^8889999|(310)999-2222^^^^^310^9992222|Emergency Contact 1|||||||||||||||||||||||||||" + '\r'
+ + "PV1|1|OUTPATIENT|RR CT^^^1000^^^^^^^DEPID|EL|||017511^TOBIAS^JONATHAN^^^^^^EPIC^^^^PROVID|017511^TOBIAS^JONATHAN^^^^^^EPIC^^^^PROVID||||||"
+ + "CLR|||||60000013647|SELF|||||||||||||||||||||HOV_CONF|^^^1000^^^^^^^||20150107161438||||||||||" + '\r'
+ + "PV2||||||||20150107161438||||CT BRAIN W WO CONTRAST||||||||||N|||||||||||||||||||||||||||" + '\r'
+ + "ZPV||||||||||||20150107161438|||||||||" + '\r'
+ + "AL1|1||33361^NO KNOWN ALLERGIES^^NOTCOMPUTRITION^NO KNOWN ALLERGIES^EXTELG||||||" + '\r'
+ + "DG1|1|DX|784.0^Headache^DX|Headache||VISIT" + '\r'
+ + "GT1|1|1000235129|MDCLS9^MC9^^||111 HOVER STREET^^LOS ANGELES^CA^90032^USA^^^LOS ANGELE|(310)725-6952^^^^^310^7256952||19700109|F|P/F|SLF|"
+ + "565-33-2222|||||^^^^^USA|||UNKNOWN|||||||||||||||||||||||||||||" + '\r'
+ + "UB2||||||||" + '\r'
+ + '\n';
+
+ @Test
+ public void testGenerateInvalidPayloadExceptionMessage() throws Exception {
+ String message = Hl7Util.generateInvalidPayloadExceptionMessage(TEST_MESSAGE.getBytes());
+
+ assertNull("Valid payload should result in a null message", message);
+ }
+
+ @Test
+ public void testGenerateInvalidPayloadExceptionMessageWithLengthLargerThanArraySize() throws Exception {
+ byte[] payload = TEST_MESSAGE.getBytes();
+ String message = Hl7Util.generateInvalidPayloadExceptionMessage(payload, payload.length * 2);
+
+ assertNull("Valid payload should result in a null message", message);
+ }
+
+ @Test
+ public void testGenerateInvalidPayloadExceptionMessageWithLengthSmallerThanArraySize() throws Exception {
+ byte[] payload = TEST_MESSAGE.getBytes();
+ String message = Hl7Util.generateInvalidPayloadExceptionMessage(payload, 10);
+
+ assertEquals("The HL7 payload terminating bytes [0x7c, 0x41] are incorrect - expected [0xd, 0xa] {ASCII [<CR>, <LF>]}", message);
+ }
+
+ @Test
+ public void testGenerateInvalidPayloadExceptionMessageWithNullPayload() throws Exception {
+ assertEquals("HL7 payload is null", Hl7Util.generateInvalidPayloadExceptionMessage(null));
+ }
+
+ @Test
+ public void testGenerateInvalidPayloadExceptionMessageWithInvalidStartingSegment() throws Exception {
+ byte[] invalidStartingSegment = "MSA|AA|00001|\r".getBytes();
+ byte[] basePayload = TEST_MESSAGE.getBytes();
+
+ ByteArrayOutputStream payloadStream = new ByteArrayOutputStream(invalidStartingSegment.length + basePayload.length);
+ payloadStream.write(invalidStartingSegment);
+ payloadStream.write(basePayload.length);
+
+ assertEquals("The first segment of the HL7 payload {MSA} is not an MSH segment", Hl7Util.generateInvalidPayloadExceptionMessage(payloadStream.toByteArray()));
+ }
+
+ @Test
+ public void testGenerateInvalidPayloadExceptionMessageWithEmptyPayload() throws Exception {
+ byte[] payload = new byte[0];
+
+ assertEquals("HL7 payload is empty", Hl7Util.generateInvalidPayloadExceptionMessage(payload));
+ assertEquals("HL7 payload is empty", Hl7Util.generateInvalidPayloadExceptionMessage(payload, payload.length));
+ }
+
+ @Test
+ public void testGenerateInvalidPayloadExceptionMessageWithEmbeddedStartOfBlock() throws Exception {
+ byte[] basePayload = TEST_MESSAGE.getBytes();
+
+ ByteArrayOutputStream payloadStream = new ByteArrayOutputStream(basePayload.length + 1);
+
+ int embeddedStartOfBlockIndex = basePayload.length / 2;
+ payloadStream.write(basePayload, 0, embeddedStartOfBlockIndex);
+ payloadStream.write(START_OF_BLOCK);
+ payloadStream.write(basePayload, embeddedStartOfBlockIndex, basePayload.length - embeddedStartOfBlockIndex);
+
+ String expected = "HL7 payload contains an embedded START_OF_BLOCK {0xb, ASCII <VT>} at index " + embeddedStartOfBlockIndex;
+
+ assertEquals(expected, Hl7Util.generateInvalidPayloadExceptionMessage(payloadStream.toByteArray()));
+ }
+
+ @Test
+ public void testGenerateInvalidPayloadExceptionMessageWithEmbeddedEndOfBlock() throws Exception {
+ byte[] basePayload = TEST_MESSAGE.getBytes();
+
+ ByteArrayOutputStream payloadStream = new ByteArrayOutputStream(basePayload.length + 1);
+
+ int embeddedEndOfBlockIndex = basePayload.length / 2;
+ payloadStream.write(basePayload, 0, embeddedEndOfBlockIndex);
+ payloadStream.write(END_OF_BLOCK);
+ payloadStream.write(basePayload, embeddedEndOfBlockIndex, basePayload.length - embeddedEndOfBlockIndex);
+
+ String expected = "HL7 payload contains an embedded END_OF_BLOCK {0x1c, ASCII <FS>} at index " + embeddedEndOfBlockIndex;
+
+ assertEquals(expected, Hl7Util.generateInvalidPayloadExceptionMessage(payloadStream.toByteArray()));
+ }
+}
\ No newline at end of file
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpBufferedSocketAcknowledgementWriterTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpBufferedSocketAcknowledgementWriterTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpBufferedSocketAcknowledgementWriterTest.java
new file mode 100644
index 0000000..41fa6e5
--- /dev/null
+++ b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpBufferedSocketAcknowledgementWriterTest.java
@@ -0,0 +1,125 @@
+/**
+ * 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.mllp.impl;
+
+import org.apache.camel.component.mllp.MllpAcknowledgementDeliveryException;
+import org.apache.camel.test.util.PayloadBuilder;
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.apache.camel.component.mllp.MllpEndpoint.END_OF_BLOCK;
+import static org.apache.camel.component.mllp.MllpEndpoint.END_OF_DATA;
+import static org.apache.camel.component.mllp.MllpEndpoint.START_OF_BLOCK;
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertNotNull;
+
+public class MllpBufferedSocketAcknowledgementWriterTest extends MllpSocketWriterTestSupport {
+ MllpSocketWriter mllpSocketWriter;
+
+ @Before
+ public void setUp() throws Exception {
+ mllpSocketWriter = new MllpBufferedSocketWriter(fakeSocket, true);
+ }
+
+ @Test
+ public void testWriteAcknowledgement() throws Exception {
+ byte[] expected = PayloadBuilder.build(START_OF_BLOCK, TEST_ACKNOWLEDGEMENT, END_OF_BLOCK, END_OF_DATA);
+
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+
+ assertArrayEquals(expected, fakeSocket.payload());
+ }
+
+ @Test
+ public void testWriteNullAcknowledgement() throws Exception {
+ byte[] acknowledgement = null;
+
+ byte[] expected = PayloadBuilder.build(START_OF_BLOCK, END_OF_BLOCK, END_OF_DATA);
+
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), acknowledgement);
+
+ assertArrayEquals(expected, fakeSocket.payload());
+ }
+
+ @Test
+ public void testWriteEmptyAcknowledgement() throws Exception {
+ byte[] acknowledgement = new byte[0];
+
+ byte[] expected = PayloadBuilder.build(START_OF_BLOCK, END_OF_BLOCK, END_OF_DATA);
+
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), acknowledgement);
+
+ assertArrayEquals(expected, fakeSocket.payload());
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testGetOutputStreamFailure() throws Exception {
+ fakeSocket.fakeSocketOutputStream = null;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testWriteToUnconnectedSocket() throws Exception {
+ fakeSocket.connected = false;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testWriteToClosedSocket() throws Exception {
+ fakeSocket.closed = true;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testWriteAcknowledgementFailure() throws Exception {
+ fakeSocket.fakeSocketOutputStream.failOnWriteArray = true;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ private void verifyException(MllpAcknowledgementDeliveryException expectedEx) throws Exception {
+ assertNotNull(expectedEx.getMessage());
+ assertArrayEquals(TEST_MESSAGE.getBytes(), expectedEx.getHl7Message());
+ assertArrayEquals(TEST_ACKNOWLEDGEMENT.getBytes(), expectedEx.getHl7Acknowledgement());
+ assertArrayEquals(TEST_ACKNOWLEDGEMENT.getBytes(), expectedEx.getMllpPayload());
+ }
+
+
+}
\ No newline at end of file
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpBufferedSocketMessageWriterTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpBufferedSocketMessageWriterTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpBufferedSocketMessageWriterTest.java
new file mode 100644
index 0000000..1f2c7bf
--- /dev/null
+++ b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpBufferedSocketMessageWriterTest.java
@@ -0,0 +1,126 @@
+/**
+ * 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.mllp.impl;
+
+import org.apache.camel.component.mllp.MllpWriteException;
+import org.apache.camel.test.util.PayloadBuilder;
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.apache.camel.component.mllp.MllpEndpoint.END_OF_BLOCK;
+import static org.apache.camel.component.mllp.MllpEndpoint.END_OF_DATA;
+import static org.apache.camel.component.mllp.MllpEndpoint.START_OF_BLOCK;
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+
+public class MllpBufferedSocketMessageWriterTest extends MllpSocketWriterTestSupport {
+ MllpSocketWriter mllpSocketWriter;
+
+ @Before
+ public void setUp() throws Exception {
+ mllpSocketWriter = new MllpBufferedSocketWriter(fakeSocket, false);
+ }
+
+ @Test
+ public void testWriteMessage() throws Exception {
+ byte[] expected = PayloadBuilder.build(START_OF_BLOCK, TEST_MESSAGE, END_OF_BLOCK, END_OF_DATA);
+
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), null);
+
+ assertArrayEquals(expected, fakeSocket.payload());
+ }
+
+ @Test
+ public void testWriteNullMessage() throws Exception {
+ byte[] message = null;
+
+ byte[] expected = PayloadBuilder.build(START_OF_BLOCK, END_OF_BLOCK, END_OF_DATA);
+
+ mllpSocketWriter.writeEnvelopedPayload(message, null);
+
+ assertArrayEquals(expected, fakeSocket.payload());
+ }
+
+ @Test
+ public void testWriteEmptyMessage() throws Exception {
+ byte[] message = new byte[0];
+
+ byte[] expected = PayloadBuilder.build(START_OF_BLOCK, END_OF_BLOCK, END_OF_DATA);
+
+ mllpSocketWriter.writeEnvelopedPayload(message, null);
+
+ assertArrayEquals(expected, fakeSocket.payload());
+ }
+
+ @Test(expected = MllpWriteException.class)
+ public void testGetOutputStreamFailure() throws Exception {
+ fakeSocket.fakeSocketOutputStream = null;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), null);
+ } catch (MllpWriteException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpWriteException.class)
+ public void testWriteToUnconnectedSocket() throws Exception {
+ fakeSocket.connected = false;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), null);
+ } catch (MllpWriteException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpWriteException.class)
+ public void testWriteToClosedSocket() throws Exception {
+ fakeSocket.closed = true;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), null);
+ } catch (MllpWriteException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpWriteException.class)
+ public void testWriteFailure() throws Exception {
+ fakeSocket.fakeSocketOutputStream.failOnWriteArray = true;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), null);
+ } catch (MllpWriteException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+
+ private void verifyException(MllpWriteException expectedEx) throws Exception {
+ assertNotNull(expectedEx.getMessage());
+ assertArrayEquals(TEST_MESSAGE.getBytes(), expectedEx.getHl7Message());
+ assertNull(expectedEx.getHl7Acknowledgement());
+ assertArrayEquals(TEST_MESSAGE.getBytes(), expectedEx.getMllpPayload());
+ }
+
+}
\ No newline at end of file
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpSocketAcknowledgementReaderTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpSocketAcknowledgementReaderTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpSocketAcknowledgementReaderTest.java
new file mode 100644
index 0000000..fa3b687
--- /dev/null
+++ b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpSocketAcknowledgementReaderTest.java
@@ -0,0 +1,533 @@
+/**
+ * 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.mllp.impl;
+
+
+import java.io.IOException;
+import java.net.SocketException;
+import java.net.SocketTimeoutException;
+
+import org.apache.camel.component.mllp.MllpAcknowledgementException;
+import org.apache.camel.component.mllp.MllpAcknowledgementTimeoutException;
+import org.apache.camel.component.mllp.MllpReceiveAcknowledgementException;
+
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.apache.camel.component.mllp.MllpEndpoint.SEGMENT_DELIMITER;
+import static org.apache.camel.component.mllp.MllpEndpoint.START_OF_BLOCK;
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.CoreMatchers.sameInstance;
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertThat;
+
+public class MllpSocketAcknowledgementReaderTest extends MllpSocketReaderTestSupport {
+ MllpSocketReader mllpSocketReader;
+
+ @Before
+ public void setUp() throws Exception {
+ mllpSocketReader = new MllpSocketReader(fakeSocket, 5000, 1000, true);
+ }
+
+ @Test
+ public void testReadMessage() throws Exception {
+ byte[] expected = TEST_ACKNOWLEDGEMENT.getBytes();
+
+ fakeSocket.fakeSocketInputStream.addPacket(START_PACKET, expected, END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testReadMessageWithSeparateEnvelopingAndMessagePackets() throws Exception {
+ byte[] expected = TEST_MESSAGE.getBytes();
+
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, TEST_MESSAGE.getBytes(), END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload();
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testReadMessageWithMultipleMessagePackets() throws Exception {
+ byte[] expected = TEST_MESSAGE.getBytes();
+
+ fakeSocket.fakeSocketInputStream
+ .addPacket(START_PACKET)
+ .addPackets(TEST_MESSAGE, SEGMENT_DELIMITER)
+ .addPacket(END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload();
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testReadEmptyMessage() throws Exception {
+ byte[] expected = new byte[0];
+ fakeSocket.fakeSocketInputStream.addPacket(START_PACKET, END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testReadEmptyMessageWithSeparateEnvelopingPackets() throws Exception {
+ byte[] expected = new byte[0];
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload();
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test()
+ public void testGetInputStreamFailure() throws Exception {
+ fakeSocket.fakeSocketInputStream = null;
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause(), instanceOf(IOException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testEndOfStreamOnInitialRead() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPackets(EMPTY_PACKET, TEST_MESSAGE);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNull(expectedEx.getCause());
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testTimeoutOnInitialRead() throws Exception {
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpAcknowledgementTimeoutException.class);
+ } catch (MllpAcknowledgementTimeoutException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(SocketTimeoutException.class));
+ assertSocketOpen();
+ }
+ }
+
+ @Test
+ public void testTimeoutOnInitialReadWithStartOfBlock() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPacket(START_OF_BLOCK);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpAcknowledgementTimeoutException.class);
+ } catch (MllpAcknowledgementTimeoutException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(SocketTimeoutException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testSocketExceptionOnInitialRead() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPacket(EXCEPTION_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(SocketException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testIOExceptionOnInitialRead() throws Exception {
+ fakeSocket.fakeSocketInputStream.useSocketExceptionOnNullPacket = false;
+ fakeSocket.fakeSocketInputStream.addPacket(EXCEPTION_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(IOException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testEndOfStreamOnFirstAdditionalRead() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, EMPTY_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNull(expectedEx.getCause());
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testEndOfStreamOnFirstAdditionalReadWithPartialPayload() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPacket(START_PACKET, TEST_ACKNOWLEDGEMENT.getBytes()).addPacket(EMPTY_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertExpectedException(expectedEx);
+ assertNull(expectedEx.getCause());
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testTimeoutOnFirstAdditionalRead() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpAcknowledgementTimeoutException.class);
+ } catch (MllpAcknowledgementTimeoutException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(SocketTimeoutException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testTimeoutOnFirstAdditionalReadWithPartialPayload() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPacket(START_PACKET, TEST_ACKNOWLEDGEMENT.getBytes());
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpAcknowledgementTimeoutException.class);
+ } catch (MllpAcknowledgementTimeoutException expectedEx) {
+ assertExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(SocketTimeoutException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testSocketExceptionOnFirstAdditionalRead() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, EXCEPTION_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(SocketException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testSocketExceptionOnFirstAdditionalReadWithPartialPayload() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPacket(START_PACKET, TEST_ACKNOWLEDGEMENT.getBytes()).addPacket(EXCEPTION_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(SocketException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testIOExceptionOnFirstAdditionalRead() throws Exception {
+ fakeSocket.fakeSocketInputStream.useSocketExceptionOnNullPacket = false;
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, EXCEPTION_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertEmptyExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(IOException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testIOExceptionOnFirstAdditionalReadWithPartialPayload() throws Exception {
+ fakeSocket.fakeSocketInputStream.useSocketExceptionOnNullPacket = false;
+ fakeSocket.fakeSocketInputStream.addPacket(START_PACKET, TEST_ACKNOWLEDGEMENT.getBytes()).addPacket(EXCEPTION_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(IOException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testEndOfStreamOnSecondAdditionalReadWithPartialPayload() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, TEST_ACKNOWLEDGEMENT.getBytes(), EMPTY_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertExpectedException(expectedEx);
+ assertNull(expectedEx.getCause());
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testTimeoutOnSecondAdditionalReadWithPartialPayload() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, TEST_ACKNOWLEDGEMENT.getBytes());
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpAcknowledgementTimeoutException.class);
+ } catch (MllpAcknowledgementTimeoutException expectedEx) {
+ assertExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(SocketTimeoutException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testSocketExceptionOnSecondAdditionalReadWithPartialPayload() throws Exception {
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, TEST_ACKNOWLEDGEMENT.getBytes(), EXCEPTION_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(SocketException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testIOExceptionOnSecondAdditionalReadWithPartialPayload() throws Exception {
+ fakeSocket.fakeSocketInputStream.useSocketExceptionOnNullPacket = false;
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, TEST_ACKNOWLEDGEMENT.getBytes(), EXCEPTION_PACKET);
+
+ try {
+ mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ expectedExceptionFailure(MllpReceiveAcknowledgementException.class);
+ } catch (MllpReceiveAcknowledgementException expectedEx) {
+ assertExpectedException(expectedEx);
+ assertNotNull(expectedEx.getCause());
+ assertThat(expectedEx.getCause().getClass(), sameInstance(IOException.class));
+ assertSocketReset();
+ }
+ }
+
+ @Test
+ public void testLeadingOutOfBandBytes() throws Exception {
+ byte[] expected = TEST_ACKNOWLEDGEMENT.getBytes();
+
+ fakeSocket.fakeSocketInputStream.addPacket("Junk".getBytes(), START_PACKET, expected, END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ }
+
+ @Test
+ public void testLeadingOutOfBandBytesWithEmptyMessage() throws Exception {
+ byte[] expected = new byte[0];
+ fakeSocket.fakeSocketInputStream.addPacket("Junk".getBytes(), START_PACKET, END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testLeadingOutOfBandBytesWithEmptyMessageWithSeparateEnvelopingPackets() throws Exception {
+ byte[] expected = new byte[0];
+ fakeSocket.fakeSocketInputStream.addPackets("Junk".getBytes(), START_PACKET, END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testLeadingOutOfBandBytesSeparateEnvelopingAndMessagePackets() throws Exception {
+ byte[] expected = TEST_ACKNOWLEDGEMENT.getBytes();
+
+ fakeSocket.fakeSocketInputStream.addPackets("Junk".getBytes(), START_PACKET, expected, END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testLeadingOutOfBandBytesWithMultipleMessagePackets() throws Exception {
+ byte[] expected = TEST_ACKNOWLEDGEMENT.getBytes();
+
+ fakeSocket.fakeSocketInputStream
+ .addPacket("Junk")
+ .addPacket(START_PACKET)
+ .addPackets(TEST_ACKNOWLEDGEMENT, SEGMENT_DELIMITER)
+ .addPacket(END_PACKET);
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testTrailingOutOfBandBytes() throws Exception {
+ byte[] expected = TEST_ACKNOWLEDGEMENT.getBytes();
+
+ fakeSocket.fakeSocketInputStream.addPacket(START_PACKET, expected, END_PACKET, "Junk".getBytes());
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ }
+
+ @Test
+ public void testTrailingOutOfBandBytesWithEmptyMessage() throws Exception {
+ byte[] expected = new byte[0];
+ fakeSocket.fakeSocketInputStream.addPacket(START_PACKET, END_PACKET, "Junk".getBytes());
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testTrailingOutOfBandBytesWithEmptyMessageWithSeparateEnvelopingPackets() throws Exception {
+ byte[] expected = new byte[0];
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, END_PACKET, "Junk".getBytes());
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testTrailingOutOfBandBytesSeparateEnvelopingAndMessagePackets() throws Exception {
+ byte[] expected = TEST_ACKNOWLEDGEMENT.getBytes();
+
+ fakeSocket.fakeSocketInputStream.addPackets(START_PACKET, expected, END_PACKET, "Junk".getBytes());
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ @Test
+ public void testTrailingOutOfBandBytesWithMultipleMessagePackets() throws Exception {
+ byte[] expected = TEST_ACKNOWLEDGEMENT.getBytes();
+
+ fakeSocket.fakeSocketInputStream
+ .addPacket(START_PACKET)
+ .addPackets(TEST_ACKNOWLEDGEMENT, SEGMENT_DELIMITER)
+ .addPacket(END_PACKET)
+ .addPacket("Junk");
+
+ byte[] actual = mllpSocketReader.readEnvelopedPayload(TEST_MESSAGE.getBytes());
+
+ assertArrayEquals(expected, actual);
+ assertSocketOpen();
+ }
+
+ private void assertEmptyExpectedException(MllpAcknowledgementException expectedEx) {
+ assertNotNull(expectedEx);
+ assertNotNull(expectedEx.getMessage());
+ assertArrayEquals(TEST_MESSAGE.getBytes(), expectedEx.getHl7Message());
+ assertNull(expectedEx.getHl7Acknowledgement());
+ assertArrayEquals(TEST_MESSAGE.getBytes(), expectedEx.getMllpPayload());
+ }
+
+ private void assertExpectedException(MllpAcknowledgementException expectedEx) {
+ assertNotNull(expectedEx);
+ assertNotNull(expectedEx.getMessage());
+ assertArrayEquals(TEST_MESSAGE.getBytes(), expectedEx.getHl7Message());
+ assertArrayEquals(TEST_ACKNOWLEDGEMENT.getBytes(), expectedEx.getHl7Acknowledgement());
+ assertArrayEquals(TEST_ACKNOWLEDGEMENT.getBytes(), expectedEx.getMllpPayload());
+ }
+
+}
\ No newline at end of file
http://git-wip-us.apache.org/repos/asf/camel/blob/91121843/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpSocketAcknowledgementWriterTest.java
----------------------------------------------------------------------
diff --git a/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpSocketAcknowledgementWriterTest.java b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpSocketAcknowledgementWriterTest.java
new file mode 100644
index 0000000..9c218a7
--- /dev/null
+++ b/components/camel-mllp/src/test/java/org/apache/camel/component/mllp/impl/MllpSocketAcknowledgementWriterTest.java
@@ -0,0 +1,150 @@
+/**
+ * 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.mllp.impl;
+
+import org.apache.camel.component.mllp.MllpAcknowledgementDeliveryException;
+import org.apache.camel.test.util.PayloadBuilder;
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.apache.camel.component.mllp.MllpEndpoint.END_OF_BLOCK;
+import static org.apache.camel.component.mllp.MllpEndpoint.END_OF_DATA;
+import static org.apache.camel.component.mllp.MllpEndpoint.START_OF_BLOCK;
+import static org.apache.camel.component.mllp.impl.MllpSocketWriter.PAYLOAD_TERMINATOR;
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertNotNull;
+
+public class MllpSocketAcknowledgementWriterTest extends MllpSocketWriterTestSupport {
+ MllpSocketWriter mllpSocketWriter;
+
+ @Before
+ public void setUp() throws Exception {
+ mllpSocketWriter = new MllpSocketWriter(fakeSocket, true);
+ }
+
+ @Test
+ public void testWriteAcknowledgement() throws Exception {
+ byte[] expected = PayloadBuilder.build(START_OF_BLOCK, TEST_ACKNOWLEDGEMENT, END_OF_BLOCK, END_OF_DATA);
+
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+
+ assertArrayEquals(expected, fakeSocket.payload());
+ }
+
+ @Test
+ public void testWriteNullAcknowledgement() throws Exception {
+ byte[] acknowledgement = null;
+
+ byte[] expected = PayloadBuilder.build(START_OF_BLOCK, END_OF_BLOCK, END_OF_DATA);
+
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), acknowledgement);
+
+ assertArrayEquals(expected, fakeSocket.payload());
+ }
+
+ @Test
+ public void testWriteEmptyAcknowledgement() throws Exception {
+ byte[] acknowledgement = new byte[0];
+
+ byte[] expected = PayloadBuilder.build(START_OF_BLOCK, END_OF_BLOCK, END_OF_DATA);
+
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), acknowledgement);
+
+ assertArrayEquals(expected, fakeSocket.payload());
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testGetOutputStreamFailure() throws Exception {
+ fakeSocket.fakeSocketOutputStream = null;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testWriteToUnconnectedSocket() throws Exception {
+ fakeSocket.connected = false;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testWriteToClosedSocket() throws Exception {
+ fakeSocket.closed = true;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testWriteStartOfBlockFailure() throws Exception {
+ fakeSocket.fakeSocketOutputStream.writeFailOn = new Byte((byte) START_OF_BLOCK);
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testWriteAcknowledgementFailure() throws Exception {
+ fakeSocket.fakeSocketOutputStream.writeArrayFailOn = TEST_ACKNOWLEDGEMENT.getBytes();
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ @Test(expected = MllpAcknowledgementDeliveryException.class)
+ public void testWriteEndOfMessageFailure() throws Exception {
+ fakeSocket.fakeSocketOutputStream.writeArrayFailOn = PAYLOAD_TERMINATOR;
+
+ try {
+ mllpSocketWriter.writeEnvelopedPayload(TEST_MESSAGE.getBytes(), TEST_ACKNOWLEDGEMENT.getBytes());
+ } catch (MllpAcknowledgementDeliveryException expectedEx) {
+ verifyException(expectedEx);
+ throw expectedEx;
+ }
+ }
+
+ private void verifyException(MllpAcknowledgementDeliveryException expectedEx) throws Exception {
+ assertNotNull(expectedEx.getMessage());
+ assertArrayEquals(TEST_MESSAGE.getBytes(), expectedEx.getHl7Message());
+ assertArrayEquals(TEST_ACKNOWLEDGEMENT.getBytes(), expectedEx.getHl7Acknowledgement());
+ assertArrayEquals(TEST_ACKNOWLEDGEMENT.getBytes(), expectedEx.getMllpPayload());
+ }
+
+
+}
\ No newline at end of file