better tests

thorneny · Mar 16, 2016 · 8260051 · 8260051
1 parent fbb6e13
commit 8260051
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/hystrix-contrib/hystrix-junit/src/main/java/com/hystrix/junit/HystrixRequestContextRule.java b/hystrix-contrib/hystrix-junit/src/main/java/com/hystrix/junit/HystrixRequestContextRule.java
@@ -43,6 +43,7 @@ protected void before() throws Throwable {
     protected void after() {
         if (this.context != null) {
             this.context.shutdown();
+            this.context = null;
         }
     }
 

diff --git a/...-contrib/hystrix-junit/src/test/java/com/hystrix/junit/HystrixRequestContextRuleTest.java b/...-contrib/hystrix-junit/src/test/java/com/hystrix/junit/HystrixRequestContextRuleTest.java
@@ -17,5 +17,7 @@ public void initsContext() {
     @Test
     public void manuallyShutdownContextDontBreak() {
         this.request.after();
+        this.request.after();
+        MatcherAssert.assertThat(this.request.context(), CoreMatchers.nullValue());
     }
 }