summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/main/java/com/att/nsa/mr/test/support/MRConsumerMock.java1
-rw-r--r--src/main/java/com/att/nsa/mr/tools/MessageCommand.java1
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 )
{