X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=tests%2Funit%2Ftest_rate_policy.cpp;h=6977ea8025ee17bbe4adca451418f65b461987c7;hb=1f38e81c6b96ff42cce9b4c2d30e33cd9e82a99e;hp=d9ae4b7b6ba63fc94100c4bae3a3f9f126c8759a;hpb=f46376a14da2eb796690cb4e718e8b213839d6ea;p=lttng-tools.git

diff --git a/tests/unit/test_rate_policy.cpp b/tests/unit/test_rate_policy.cpp
index d9ae4b7b6..6977ea802 100644
--- a/tests/unit/test_rate_policy.cpp
+++ b/tests/unit/test_rate_policy.cpp
@@ -14,9 +14,9 @@
 
 #include <tap/tap.h>
 
-#include <common/payload-view.h>
-#include <common/payload.h>
-#include <lttng/action/rate-policy-internal.h>
+#include <common/payload-view.hpp>
+#include <common/payload.hpp>
+#include <lttng/action/rate-policy-internal.hpp>
 #include <lttng/action/rate-policy.h>
 
 /* For error.h. */
@@ -131,6 +131,7 @@ static void test_rate_policy_every_n(void)
 	lttng_rate_policy_destroy(policy_a);
 	lttng_rate_policy_destroy(policy_b);
 	lttng_rate_policy_destroy(policy_c);
+	lttng_rate_policy_destroy(policy_from_buffer);
 	lttng_payload_reset(&payload);
 }
 
@@ -240,6 +241,7 @@ static void test_rate_policy_once_after_n(void)
 	lttng_rate_policy_destroy(policy_a);
 	lttng_rate_policy_destroy(policy_b);
 	lttng_rate_policy_destroy(policy_c);
+	lttng_rate_policy_destroy(policy_from_buffer);
 	lttng_payload_reset(&payload);
 }