diff options
-rw-r--r-- | src/main/java/com/att/nsa/mr/test/support/MRConsumerMock.java | 1 | ||||
-rw-r--r-- | src/main/java/com/att/nsa/mr/tools/MessageCommand.java | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/src/main/java/com/att/nsa/mr/test/support/MRConsumerMock.java b/src/main/java/com/att/nsa/mr/test/support/MRConsumerMock.java index 3373710..7d0e20c 100644 --- a/src/main/java/com/att/nsa/mr/test/support/MRConsumerMock.java +++ b/src/main/java/com/att/nsa/mr/test/support/MRConsumerMock.java @@ -71,6 +71,7 @@ public class MRConsumerMock implements MRConsumer } catch ( InterruptedException e ) { + Thread.currentThread().interrupt(); throw new IOException ( e ); } } diff --git a/src/main/java/com/att/nsa/mr/tools/MessageCommand.java b/src/main/java/com/att/nsa/mr/tools/MessageCommand.java index 97216c4..354dc18 100644 --- a/src/main/java/com/att/nsa/mr/tools/MessageCommand.java +++ b/src/main/java/com/att/nsa/mr/tools/MessageCommand.java @@ -116,6 +116,7 @@ public class MessageCommand implements Command<MRCommandContext> { out.println ( "Problem sending message: " + e.getMessage() ); logger.error("Problem sending message: ", e); + Thread.currentThread().interrupt(); } if ( left != null && left.size () > 0 ) { |