Test: Add the new log4j agent

[lttng-tools.git] / tests / regression / ust / java-log4j / JTestLTTng.java

diff --git a/tests/regression/ust/java-log4j/JTestLTTng.java b/tests/regression/ust/java-log4j/JTestLTTng.java
new file mode 100644 (file)
index 0000000..0019cb8
--- /dev/null
+++ b/tests/regression/ust/java-log4j/JTestLTTng.java
@@ -0,0 +1,63 @@
+/*
+ * Copyright (C) 2014 - David Goulet <dgoulet@efficios.com>
+ *                      Christian Babeux <christian.babeux@efficios.com>
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License, version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along with
+ * this program; if not, write to the Free Software Foundation, Inc., 51
+ * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+import java.io.IOException;
+import java.lang.Integer;
+
+import org.apache.log4j.Logger;
+import org.apache.log4j.BasicConfigurator;
+
+import org.lttng.ust.agent.LTTngAgent;
+
+public class JTestLTTng
+{
+       private static LTTngAgent lttngAgent;
+
+       public static void main(String args[]) throws Exception
+       {
+               Logger lttng = Logger.getLogger("log4j-event");
+               Logger lttng2 = Logger.getLogger("log4j-event-2");
+               int nrIter = Integer.parseInt(args[0]);
+               int waitTime = Integer.parseInt(args[1]);
+               int fire_debug_tp = 0;
+               int fire_second_tp = 0;
+
+               if (args.length > 2) {
+                       fire_debug_tp = Integer.parseInt(args[2]);
+               }
+               if (args.length > 3) {
+                       fire_second_tp = Integer.parseInt(args[3]);
+               }
+
+               BasicConfigurator.configure();
+               lttngAgent = LTTngAgent.getLTTngAgent();
+
+               for (int iter = 0; iter < nrIter; iter++) {
+                       lttng.info("LOG4J tp fired!");
+                       if (fire_debug_tp == 1) {
+                               /* Third arg, trigger debug TP. */
+                               lttng.debug("LOG4J DEBUG tp fired");
+                       }
+                       Thread.sleep(waitTime);
+               }
+
+               if (fire_second_tp == 1) {
+                       lttng2.info("LOG4J second logger fired");
+               }
+       }
+}