Improve test stability

mp911de · Feb 18, 2016 · 1608d33 · 1608d33
1 parent af07f39
commit 1608d33
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/test/java/biz/paluch/spinach/ClientTest.java b/src/test/java/biz/paluch/spinach/ClientTest.java
@@ -180,7 +180,7 @@ public void connectConnectionRefused() throws Exception {
             client.connect();
         } catch (Exception e) {
             assertThat(e).hasRootCauseExactlyInstanceOf(ConnectException.class);
-            assertThat(e.getCause().getCause()).hasMessageContaining("Connection refused");
+            assertThat(e.getCause().getCause()).hasMessageContaining(TestSettings.host());
         }
         client.shutdown(0, 0, TimeUnit.MILLISECONDS);
     }

diff --git a/src/test/java/biz/paluch/spinach/impl/DisqueCommandTest.java b/src/test/java/biz/paluch/spinach/impl/DisqueCommandTest.java
@@ -1,12 +1,13 @@
 package biz.paluch.spinach.impl;
 
-import static org.assertj.core.api.Assertions.*;
+import static org.assertj.core.api.Assertions.assertThat;
 
 import java.nio.ByteBuffer;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 
+import org.junit.After;
 import org.junit.Test;
 
 import biz.paluch.spinach.api.CommandKeyword;
@@ -23,6 +24,12 @@ public class DisqueCommandTest {
     private DisqueCommand<String, String, String> sut = new DisqueCommand<String, String, String>(CommandKeyword.GET,
             new StatusOutput<String, String>(null), new DisqueCommandArgs<String, String>(null));
 
+    @After
+    public void tearDown() throws Exception {
+        // clear interrupted flag
+        Thread.interrupted();
+    }
+
     @Test
     public void sucessfulGet() throws Exception {