ir: split event files into event and event-class files

[babeltrace.git] / tests / lib / test_ctf_ir_ref.c

diff --git a/tests/lib/test_ctf_ir_ref.c b/tests/lib/test_ctf_ir_ref.c
index 856715cb90857b6d1d7d2e579b7ee8fce8104d85..eecaad880ad0753fe67003ae8e150ec65e28e3cc 100644 (file)
--- a/tests/lib/test_ctf_ir_ref.c
+++ b/tests/lib/test_ctf_ir_ref.c
@@ -24,9 +24,12 @@
 #include <babeltrace/ctf-ir/stream-class.h>
 #include <babeltrace/ctf-ir/stream.h>
 #include <babeltrace/ctf-ir/event.h>
+#include <babeltrace/ctf-ir/event-class.h>
 #include <babeltrace/object-internal.h>
 #include <assert.h>
 
+#define NR_TESTS 41
+
 struct user {
        struct bt_ctf_trace *tc;
        struct bt_ctf_stream_class *sc;
@@ -358,6 +361,9 @@ int main(int argc, char **argv)
                        *weak_ec3 = NULL;
        struct user user_a = { 0 }, user_b = { 0 }, user_c = { 0 };
 
+        /* Initialize tap harness before any tests */
+       plan_tests(NR_TESTS);
+
        /* The only reference which exists at this point is on TC1. */
        tc1 = create_tc1();
         ok(tc1, "Initialize trace");
@@ -367,7 +373,6 @@ int main(int argc, char **argv)
 
        init_weak_refs(tc1, &weak_tc1, &weak_sc1, &weak_sc2, &weak_ec1,
                        &weak_ec2, &weak_ec3);
-       plan_no_plan();
 
        ok(bt_object_get_ref_count(weak_sc1) == 0,
                        "Initial SC1 reference count is 0");