Add multi-session tests

[lttng-ust-java-tests.git] / src / org / lttng / ust / agent / benchmarks / jul / handler / lttng / LttngJulHandlerTracingEnabledBenchmark.java

diff --git a/src/org/lttng/ust/agent/benchmarks/jul/handler/lttng/LttngJulHandlerTracingEnabledBenchmark.java b/src/org/lttng/ust/agent/benchmarks/jul/handler/lttng/LttngJulHandlerTracingEnabledBenchmark.java
index eb0514ea023247343afc00661fc95834e73321d1..02c7b05ac138f5976c346743c18b045d370fee67 100644 (file)
--- a/src/org/lttng/ust/agent/benchmarks/jul/handler/lttng/LttngJulHandlerTracingEnabledBenchmark.java
+++ b/src/org/lttng/ust/agent/benchmarks/jul/handler/lttng/LttngJulHandlerTracingEnabledBenchmark.java
@@ -8,21 +8,25 @@ import org.junit.After;
 import org.junit.Before;
 import org.lttng.ust.agent.benchmarks.jul.handler.AbstractJulBenchmark;
 import org.lttng.ust.agent.jul.LttngLogHandler;
-import org.lttng.ust.agent.utils.LttngSessionControl;
-import org.lttng.ust.agent.utils.LttngSessionControl.Domain;
+import org.lttng.ust.agent.utils.LttngSession;
+import org.lttng.ust.agent.utils.LttngSession.Domain;
 
 public class LttngJulHandlerTracingEnabledBenchmark extends AbstractJulBenchmark {
 
+    private LttngSession session;
+
     @Before
     public void testSetup() throws IOException {
         handler = new LttngLogHandler();
 
-        assertTrue(LttngSessionControl.setupSessionAllEvents(null, Domain.JUL));
+        session = new LttngSession(null, Domain.JUL);
+        assertTrue(session.enableAllEvents());
+        assertTrue(session.start());
     }
 
     @After
     public void testTeardown() {
-        assertTrue(LttngSessionControl.stopSession());
-        assertTrue(LttngSessionControl.destroySession());
+        assertTrue(session.stop());
+        session.close();
     }
 }