X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Forg%2Flttng%2Fust%2Fagent%2Fintegration%2FEnabledEventsTestBase.java;h=fd72279768671ec41f0a7d29e5deff36b6bad23c;hb=ff620befefbb7be90965d845e76bc826eca2d951;hp=159b6f7f3a594b32bc29bfbfc9a6573c1d3e8fb8;hpb=2b408e85806bd695f6036371701dcb8691ab3c32;p=lttng-ust-java-tests.git

diff --git a/src/test/java/org/lttng/ust/agent/integration/EnabledEventsTestBase.java b/src/test/java/org/lttng/ust/agent/integration/EnabledEventsTestBase.java
index 159b6f7..fd72279 100644
--- a/src/test/java/org/lttng/ust/agent/integration/EnabledEventsTestBase.java
+++ b/src/test/java/org/lttng/ust/agent/integration/EnabledEventsTestBase.java
@@ -27,10 +27,17 @@ import java.util.List;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.lttng.tools.ILttngSession;
+import org.lttng.tools.ILttngSession.Domain;
 import org.lttng.ust.agent.ILttngHandler;
-import org.lttng.ust.agent.utils.LttngSession;
-import org.lttng.ust.agent.utils.LttngSession.Domain;
+import org.lttng.ust.agent.utils.TestPrintRunner;
 
+/**
+ * Base abstract class to implement all sorts of integration tests verifying the
+ * presence of enabled events in resulting traces.
+ */
+@RunWith(TestPrintRunner.class)
 public abstract class EnabledEventsTestBase {
 
     protected static final String EVENT_NAME_A = "EventA";
@@ -38,7 +45,7 @@ public abstract class EnabledEventsTestBase {
     protected static final String EVENT_NAME_C = "EventABC";
     protected static final String EVENT_NAME_D = "EventABCD";
 
-    private LttngSession session;
+    private ILttngSession session;
 
     /* Fields defined by the sub-class */
     protected ILttngHandler handlerA;
@@ -49,11 +56,17 @@ public abstract class EnabledEventsTestBase {
 
     protected abstract void sendEventsToLoggers();
 
+    /**
+     * Base test setup
+     */
     @Before
     public void testSetup() {
-        session = new LttngSession(null, getDomain());
+        session = ILttngSession.createSession(null, getDomain());
     }
 
+    /**
+     * Base test teardown
+     */
     @After
     public void testTeardown() {
         session.close();