Update tests for previous API change
diff --git a/src/main/java/org/cloudera/htrace/wrappers/TraceCallable.java b/src/main/java/org/cloudera/htrace/wrappers/TraceCallable.java
index 0a0fcff..a25b416 100644
--- a/src/main/java/org/cloudera/htrace/wrappers/TraceCallable.java
+++ b/src/main/java/org/cloudera/htrace/wrappers/TraceCallable.java
@@ -18,7 +18,6 @@
import java.util.concurrent.Callable;
-import org.cloudera.htrace.Sampler;
import org.cloudera.htrace.Span;
import org.cloudera.htrace.Trace;
@@ -42,8 +41,7 @@
@Override
public V call() throws Exception {
if (parent != null) {
- Span chunk = Trace.startSpan(Thread.currentThread().getName(), parent,
- Sampler.ALWAYS);
+ Span chunk = Trace.startSpan(Thread.currentThread().getName(), parent);
try {
return impl.call();
diff --git a/src/main/java/org/cloudera/htrace/wrappers/TraceRunnable.java b/src/main/java/org/cloudera/htrace/wrappers/TraceRunnable.java
index 4b2de8f..122c227 100644
--- a/src/main/java/org/cloudera/htrace/wrappers/TraceRunnable.java
+++ b/src/main/java/org/cloudera/htrace/wrappers/TraceRunnable.java
@@ -41,8 +41,7 @@
@Override
public void run() {
if (parent != null) {
- Span chunk = Trace.startSpan(Thread.currentThread().getName(), parent,
- Sampler.ALWAYS);
+ Span chunk = Trace.startSpan(Thread.currentThread().getName(), parent);
try {
runnable.run();
diff --git a/src/test/java/org/cloudera/htrace/TestSampler.java b/src/test/java/org/cloudera/htrace/TestSampler.java
index 6990873..be378e4 100644
--- a/src/test/java/org/cloudera/htrace/TestSampler.java
+++ b/src/test/java/org/cloudera/htrace/TestSampler.java
@@ -20,8 +20,7 @@
@Test
public void testAlwaysSampler() {
- Span cur = Trace.startSpan("test", new TraceInfo(0, 0),
- Sampler.ALWAYS);
+ Span cur = Trace.startSpan("test", new TraceInfo(0, 0));
assertFalse(cur.equals(NullSpan.getInstance()));
cur.stop();
}
diff --git a/src/test/java/org/cloudera/htrace/TraceCreator.java b/src/test/java/org/cloudera/htrace/TraceCreator.java
index afbd821..48fe77c 100644
--- a/src/test/java/org/cloudera/htrace/TraceCreator.java
+++ b/src/test/java/org/cloudera/htrace/TraceCreator.java
@@ -148,7 +148,7 @@
}
public void pretendRpcReceiveWithTraceInfo(TraceInfo traceInfo) {
- Span s = Trace.startSpan("received RPC", traceInfo, Sampler.ALWAYS);
+ Span s = Trace.startSpan("received RPC", traceInfo);
try {
importantWork1();
} finally {