2 * Unit tests for the rate policy object API.
4 * Copyright (C) 2019 Jonathan Rajotte <jonathan.rajotte-julien@efficios.com>
6 * SPDX-License-Identifier: LGPL-2.1-only
17 #include <common/payload-view.hpp>
18 #include <common/payload.hpp>
19 #include <lttng/action/rate-policy-internal.hpp>
20 #include <lttng/action/rate-policy.h>
23 int lttng_opt_quiet
= 1;
24 int lttng_opt_verbose
;
29 static void test_rate_policy_every_n(void)
31 enum lttng_rate_policy_status status
;
32 struct lttng_rate_policy
*policy_a
= NULL
; /* Interval of 100. */
33 struct lttng_rate_policy
*policy_b
= NULL
; /* Interval of 100 */
34 struct lttng_rate_policy
*policy_c
= NULL
; /* Interval of 1 */
35 struct lttng_rate_policy
*policy_from_buffer
= NULL
;
36 uint64_t interval_a_b
= 100;
37 uint64_t interval_c
= 1;
38 uint64_t interval_query
= 0;
39 struct lttng_payload payload
;
41 lttng_payload_init(&payload
);
43 policy_a
= lttng_rate_policy_every_n_create(interval_a_b
);
44 policy_b
= lttng_rate_policy_every_n_create(interval_a_b
);
45 policy_c
= lttng_rate_policy_every_n_create(interval_c
);
47 "Rate policy every n A created: interval: %" PRIu64
,
50 "Rate policy every n B created: interval: %" PRIu64
,
53 "Rate policy every n C created: interval: %" PRIu64
,
56 ok(LTTNG_RATE_POLICY_TYPE_EVERY_N
==
57 lttng_rate_policy_get_type(policy_a
),
58 "Type is LTTNG_RATE_POLICY_TYPE_EVERY_N");
61 status
= lttng_rate_policy_every_n_get_interval(NULL
, NULL
);
62 ok(status
== LTTNG_RATE_POLICY_STATUS_INVALID
,
63 "Get interval returns INVALID");
65 status
= lttng_rate_policy_every_n_get_interval(NULL
, &interval_query
);
66 ok(status
== LTTNG_RATE_POLICY_STATUS_INVALID
,
67 "Get interval returns INVALID");
69 status
= lttng_rate_policy_every_n_get_interval(policy_a
, NULL
);
70 ok(status
== LTTNG_RATE_POLICY_STATUS_INVALID
,
71 "Get interval returns INVALID");
73 status
= lttng_rate_policy_every_n_get_interval(
74 policy_a
, &interval_query
);
75 ok(status
== LTTNG_RATE_POLICY_STATUS_OK
&&
76 interval_query
== interval_a_b
,
77 " Getting interval A");
79 status
= lttng_rate_policy_every_n_get_interval(
80 policy_b
, &interval_query
);
81 ok(status
== LTTNG_RATE_POLICY_STATUS_OK
&&
82 interval_query
== interval_a_b
,
83 " Getting interval B");
85 status
= lttng_rate_policy_every_n_get_interval(
86 policy_c
, &interval_query
);
87 ok(status
== LTTNG_RATE_POLICY_STATUS_OK
&&
88 interval_query
== interval_c
,
89 " Getting interval C");
92 /* TODO: this is the behaviour introduced by the
93 * lttng_condition_is_equal back in 2017 do we want to fix this and
94 * return true if both are NULL?
96 ok(false == lttng_rate_policy_is_equal(NULL
, NULL
),
97 "is equal (NULL,NULL)");
98 ok(false == lttng_rate_policy_is_equal(policy_a
, NULL
),
99 "is equal (object, NULL)");
100 ok(false == lttng_rate_policy_is_equal(NULL
, policy_a
),
101 " is equal (NULL, object)");
102 ok(true == lttng_rate_policy_is_equal(policy_a
, policy_a
),
103 "is equal (object A, object A)");
105 ok(true == lttng_rate_policy_is_equal(policy_a
, policy_b
),
106 "is equal (object A, object B");
107 ok(true == lttng_rate_policy_is_equal(policy_b
, policy_a
),
108 "is equal (object B, object A");
110 ok(false == lttng_rate_policy_is_equal(policy_a
, policy_c
),
111 "is equal (object A, object C)");
112 ok(false == lttng_rate_policy_is_equal(policy_c
, policy_a
),
113 "is equal (object C, object A)");
115 /* Serialization and create_from buffer. */
116 ok(lttng_rate_policy_serialize(policy_a
, &payload
) == 0, "Serializing");
118 struct lttng_payload_view view
=
119 lttng_payload_view_from_payload(
122 ok(lttng_rate_policy_create_from_payload(
123 &view
, &policy_from_buffer
) > 0 &&
124 policy_from_buffer
!= NULL
,
128 ok(lttng_rate_policy_is_equal(policy_a
, policy_from_buffer
),
129 "serialized and from buffer are equal");
131 lttng_rate_policy_destroy(policy_a
);
132 lttng_rate_policy_destroy(policy_b
);
133 lttng_rate_policy_destroy(policy_c
);
134 lttng_rate_policy_destroy(policy_from_buffer
);
135 lttng_payload_reset(&payload
);
138 static void test_rate_policy_once_after_n(void)
140 enum lttng_rate_policy_status status
;
141 struct lttng_rate_policy
*policy_a
= NULL
; /* Threshold of 100. */
142 struct lttng_rate_policy
*policy_b
= NULL
; /* threshold of 100 */
143 struct lttng_rate_policy
*policy_c
= NULL
; /* threshold of 1 */
144 struct lttng_rate_policy
*policy_from_buffer
= NULL
;
145 uint64_t threshold_a_b
= 100;
146 uint64_t threshold_c
= 1;
147 uint64_t threshold_query
= 0;
148 struct lttng_payload payload
;
150 lttng_payload_init(&payload
);
152 policy_a
= lttng_rate_policy_once_after_n_create(threshold_a_b
);
153 policy_b
= lttng_rate_policy_once_after_n_create(threshold_a_b
);
154 policy_c
= lttng_rate_policy_once_after_n_create(threshold_c
);
156 "Rate policy every n A created: threshold: %" PRIu64
,
159 "Rate policy every n B created: threshold: %" PRIu64
,
162 "Rate policy every n C created: threshold: %" PRIu64
,
165 ok(LTTNG_RATE_POLICY_TYPE_ONCE_AFTER_N
==
166 lttng_rate_policy_get_type(policy_a
),
167 "Type is LTTNG_RATE_POLICY_TYPE_once_after_n");
170 status
= lttng_rate_policy_once_after_n_get_threshold(NULL
, NULL
);
171 ok(status
== LTTNG_RATE_POLICY_STATUS_INVALID
,
172 "Get threshold returns INVALID");
174 status
= lttng_rate_policy_once_after_n_get_threshold(
175 NULL
, &threshold_query
);
176 ok(status
== LTTNG_RATE_POLICY_STATUS_INVALID
,
177 "Get threshold returns INVALID");
179 status
= lttng_rate_policy_once_after_n_get_threshold(policy_a
, NULL
);
180 ok(status
== LTTNG_RATE_POLICY_STATUS_INVALID
,
181 "Get threshold returns INVALID");
183 status
= lttng_rate_policy_once_after_n_get_threshold(
184 policy_a
, &threshold_query
);
185 ok(status
== LTTNG_RATE_POLICY_STATUS_OK
&&
186 threshold_query
== threshold_a_b
,
187 " Getting threshold A");
189 status
= lttng_rate_policy_once_after_n_get_threshold(
190 policy_b
, &threshold_query
);
191 ok(status
== LTTNG_RATE_POLICY_STATUS_OK
&&
192 threshold_query
== threshold_a_b
,
193 " Getting threshold B");
195 status
= lttng_rate_policy_once_after_n_get_threshold(
196 policy_c
, &threshold_query
);
197 ok(status
== LTTNG_RATE_POLICY_STATUS_OK
&&
198 threshold_query
== threshold_c
,
199 " Getting threshold C");
202 ok(false == lttng_rate_policy_is_equal(NULL
, NULL
),
203 "is equal (NULL,NULL)");
204 ok(false == lttng_rate_policy_is_equal(policy_a
, NULL
),
205 "is equal (object, NULL)");
206 ok(false == lttng_rate_policy_is_equal(NULL
, policy_a
),
207 " is equal (NULL, object)");
208 ok(true == lttng_rate_policy_is_equal(policy_a
, policy_a
),
209 "is equal (object A, object A)");
211 ok(true == lttng_rate_policy_is_equal(policy_a
, policy_b
),
212 "is equal (object A, object B");
213 ok(true == lttng_rate_policy_is_equal(policy_b
, policy_a
),
214 "is equal (object B, object A");
216 ok(false == lttng_rate_policy_is_equal(policy_a
, policy_c
),
217 "is equal (object A, object C)");
218 ok(false == lttng_rate_policy_is_equal(policy_c
, policy_a
),
219 "is equal (object C, object A)");
221 /* Serialization and create_from buffer. */
222 ok(lttng_rate_policy_serialize(policy_a
, &payload
) == 0, "Serializing");
224 struct lttng_payload_view view
=
225 lttng_payload_view_from_payload(
228 ok(lttng_rate_policy_create_from_payload(
229 &view
, &policy_from_buffer
) > 0 &&
230 policy_from_buffer
!= NULL
,
234 ok(lttng_rate_policy_is_equal(policy_a
, policy_from_buffer
),
235 "serialized and from buffer are equal");
237 lttng_rate_policy_destroy(policy_a
);
238 lttng_rate_policy_destroy(policy_b
);
239 lttng_rate_policy_destroy(policy_c
);
240 lttng_rate_policy_destroy(policy_from_buffer
);
241 lttng_payload_reset(&payload
);
246 plan_tests(NUM_TESTS
);
247 test_rate_policy_every_n();
248 test_rate_policy_once_after_n();
249 return exit_status();