Introduce lttng_event_rule_python_logging

[lttng-tools.git] / tests / unit / test_event_rule.c

diff --git a/tests/unit/test_event_rule.c b/tests/unit/test_event_rule.c
index 5ef1bf992b55476ea429496e44ab1d449a5dd435..4a6740b6fefefd5cf40eaaef737f79ad02dbd812 100644 (file)
--- a/tests/unit/test_event_rule.c
+++ b/tests/unit/test_event_rule.c
@@ -24,6 +24,8 @@
 #include <lttng/event-rule/kernel-kprobe.h>
 #include <lttng/event-rule/kernel-syscall-internal.h>
 #include <lttng/event-rule/kernel-syscall.h>
+#include <lttng/event-rule/python-logging-internal.h>
+#include <lttng/event-rule/python-logging.h>
 #include <lttng/event-rule/tracepoint-internal.h>
 #include <lttng/event-rule/tracepoint.h>
 #include <lttng/event-rule/kernel-tracepoint-internal.h>
@@ -44,7 +46,7 @@ int lttng_opt_quiet = 1;
 int lttng_opt_verbose;
 int lttng_opt_mi;
 
-#define NUM_TESTS 304
+#define NUM_TESTS 317
 
 struct tracepoint_test {
        enum lttng_domain_type type;
@@ -498,6 +500,71 @@ void test_event_rule_log4j_logging(void)
        lttng_log_level_rule_destroy(log_level_rule);
 }
 
+static
+void test_event_rule_python_logging(void)
+{
+       struct lttng_event_rule *python_logging = NULL;
+       struct lttng_event_rule *python_logging_from_buffer = NULL;
+       enum lttng_event_rule_status status;
+       const char *pattern="my_event_*";
+       const char *filter="msg_id == 23 && size >= 2048";
+       const char *tmp;
+       struct lttng_log_level_rule *log_level_rule = NULL;
+       const struct lttng_log_level_rule *log_level_rule_return = NULL;
+       struct lttng_payload payload;
+
+       diag("Testing lttng_event_rule_user_python_logging.");
+
+       lttng_payload_init(&payload);
+
+       log_level_rule = lttng_log_level_rule_exactly_create(LTTNG_LOGLEVEL_INFO);
+       assert(log_level_rule);
+
+       python_logging = lttng_event_rule_python_logging_create();
+       ok(python_logging, "python_logging object.");
+
+       status = lttng_event_rule_python_logging_set_name_pattern(python_logging, pattern);
+       ok(status == LTTNG_EVENT_RULE_STATUS_OK, "setting pattern.");
+       status = lttng_event_rule_python_logging_get_name_pattern(python_logging, &tmp);
+       ok(status == LTTNG_EVENT_RULE_STATUS_OK, "getting pattern.");
+       ok(!strncmp(pattern, tmp, strlen(pattern)), "pattern is equal.");
+
+       status = lttng_event_rule_python_logging_set_filter(python_logging, filter);
+       ok(status == LTTNG_EVENT_RULE_STATUS_OK, "setting filter.");
+       status = lttng_event_rule_python_logging_get_filter(python_logging, &tmp);
+       ok(status == LTTNG_EVENT_RULE_STATUS_OK, "getting filter.");
+       ok(!strncmp(filter, tmp, strlen(filter)), "filter is equal.");
+
+       status = lttng_event_rule_python_logging_get_log_level_rule(python_logging, &log_level_rule_return);
+       ok(status == LTTNG_EVENT_RULE_STATUS_UNSET, "get unset log level rule.");
+
+       status = lttng_event_rule_python_logging_set_log_level_rule(
+                       python_logging, log_level_rule);
+       ok(status == LTTNG_EVENT_RULE_STATUS_OK, "setting log level rule.");
+       status = lttng_event_rule_python_logging_get_log_level_rule(
+                       python_logging, &log_level_rule_return);
+       ok(status == LTTNG_EVENT_RULE_STATUS_OK, "get log level rule.");
+
+       ok(lttng_event_rule_serialize(python_logging, &payload) == 0, "Serializing.");
+
+       {
+               struct lttng_payload_view view =
+                               lttng_payload_view_from_payload(
+                                               &payload, 0, -1);
+
+               ok(lttng_event_rule_create_from_payload(
+                               &view, &python_logging_from_buffer) > 0,
+                               "Deserializing.");
+       }
+
+       ok(lttng_event_rule_is_equal(python_logging, python_logging_from_buffer), "serialized and from buffer are equal.");
+
+       lttng_payload_reset(&payload);
+       lttng_event_rule_destroy(python_logging);
+       lttng_event_rule_destroy(python_logging_from_buffer);
+       lttng_log_level_rule_destroy(log_level_rule);
+}
+
 static void test_event_rule_userspace_probe(void)
 {
        struct lttng_event_rule *uprobe = NULL;
@@ -860,6 +927,7 @@ int main(int argc, const char *argv[])
        test_event_rule_kernel_probe();
        test_event_rule_log4j_logging();
        test_event_rule_jul_logging();
+       test_event_rule_python_logging();
        test_event_rule_log_level_ust();
        test_event_rule_log_level_jul();
        test_event_rule_log_level_log4j();