commit 94b9b539c032f66b5fcf6627ec3bed8189b02b04
author Masatake Iwasaki <iwasakims@apache.org> Fri Feb 06 14:36:32 2015 -0800
committer Masatake Iwasaki <iwasakims@apache.org> Fri Feb 06 14:36:32 2015 -0800
tree fbd3f4193410cb10b306dac9bfa6a908809b6c8d
parent cab75ceea74d26f4e7438c15845bfa375858ede3

HTRACE-103. Reset unclosed spans after TestBadClient (iwasakims)

htrace-core/src/test/java/org/apache/htrace/TestBadClient.java

diff --git a/htrace-core/src/test/java/org/apache/htrace/TestBadClient.java b/htrace-core/src/test/java/org/apache/htrace/TestBadClient.java
index cf060d5..e13a0f8 100644
--- a/htrace-core/src/test/java/org/apache/htrace/TestBadClient.java
+++ b/htrace-core/src/test/java/org/apache/htrace/TestBadClient.java
@@ -23,12 +23,14 @@
 import org.apache.htrace.HTraceConfiguration;
 import org.apache.htrace.Span;
 import org.apache.htrace.SpanReceiver;
+import org.apache.htrace.Tracer;
 import org.apache.htrace.TraceTree.SpansByParent;
 import org.apache.htrace.TraceTree;
 import org.apache.htrace.impl.AlwaysSampler;
 import org.apache.htrace.impl.LocalFileSpanReceiver;
 import org.apache.htrace.impl.POJOSpanReceiver;
 import org.apache.htrace.impl.StandardOutSpanReceiver;
+import org.junit.After;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -57,7 +59,6 @@
     assertTrue("Expected to get exception because of improper " +
         "scope closure.", gotException);
     innerScope.close();
-    Tracer.getInstance().setCurrentSpan(null);
   }
 
   /**
@@ -147,4 +148,9 @@
     assertTrue("Expected to get exception because of incorrect startSpan.",
         gotException);
   }
+
+  @After
+  public void resetCurrentSpan() {
+    Tracer.getInstance().setCurrentSpan(null);
+  }
 }