From ce1c2b5dda5971aabc3af709ff0cd581c67642b0 Mon Sep 17 00:00:00 2001 From: Jonathan Rajotte Date: Wed, 7 Apr 2021 21:17:13 -0400 Subject: [PATCH] tests: unit: action: rotate_session MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jonathan Rajotte Signed-off-by: Jérémie Galarneau Change-Id: Ib412196dea5cb9617b4c56e8eb93698bcfaa40d7 --- tests/unit/test_action.c | 109 ++++++++++++++++++++++++++++++++++++++- 1 file changed, 108 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_action.c b/tests/unit/test_action.c index e0f1aff24..30fe06e70 100644 --- a/tests/unit/test_action.c +++ b/tests/unit/test_action.c @@ -24,13 +24,14 @@ #include #include #include +#include /* For error.h */ int lttng_opt_quiet = 1; int lttng_opt_verbose; int lttng_opt_mi; -#define NUM_TESTS 8 +#define NUM_TESTS 21 static void test_action_notify(void) { @@ -106,9 +107,115 @@ static void test_action_notify(void) lttng_payload_reset(&payload); } +static void test_action_rotate_session(void) +{ + int ret; + enum lttng_action_status status; + struct lttng_action *rotate_session_action = NULL, + *rotate_session_action_from_buffer = NULL; + struct lttng_firing_policy *policy = NULL, *default_policy; + struct lttng_payload payload; + const char *session_name = "my_session_name"; + const char *get_session_name; + + lttng_payload_init(&payload); + + /* To set. */ + policy = lttng_firing_policy_every_n_create(100); + /* For comparison. */ + default_policy = lttng_firing_policy_every_n_create(1); + + assert(policy && default_policy); + + rotate_session_action = lttng_action_rotate_session_create(); + ok(rotate_session_action, "Create rotate_session action"); + ok(lttng_action_get_type(rotate_session_action) == + LTTNG_ACTION_TYPE_ROTATE_SESSION, + "Action has type LTTNG_ACTION_TYPE_ROTATE_SESSION"); + + /* Session name setter. */ + status = lttng_action_rotate_session_set_session_name(NULL, NULL); + ok(status == LTTNG_ACTION_STATUS_INVALID, + "Set session name (NULL,NULL) expect invalid"); + status = lttng_action_rotate_session_set_session_name( + rotate_session_action, NULL); + ok(status == LTTNG_ACTION_STATUS_INVALID, + "Set session name (object,NULL) expect invalid"); + status = lttng_action_rotate_session_set_session_name( + NULL, session_name); + ok(status == LTTNG_ACTION_STATUS_INVALID, + "Set session name (NULL,object) expect invalid"); + + /* Set the session name */ + status = lttng_action_rotate_session_set_session_name( + rotate_session_action, session_name); + ok(status == LTTNG_ACTION_STATUS_OK, "Set session name"); + + status = lttng_action_rotate_session_get_session_name( + rotate_session_action, &get_session_name); + ok(status == LTTNG_ACTION_STATUS_OK && + !strcmp(session_name, get_session_name), + "Get session name, expected `%s` got `%s`", + session_name, get_session_name); + + /* Validate the default policy for a rotate_session action. */ + { + const struct lttng_firing_policy *cur_policy = NULL; + status = lttng_action_rotate_session_get_firing_policy( + rotate_session_action, &cur_policy); + ok(status == LTTNG_ACTION_STATUS_OK && + lttng_firing_policy_is_equal( + default_policy, + cur_policy), + "Default policy is every n=1"); + } + + /* Set a custom policy. */ + status = lttng_action_rotate_session_set_firing_policy( + rotate_session_action, policy); + ok(status == LTTNG_ACTION_STATUS_OK, "Set firing policy"); + + /* Validate the custom policy for a rotate_session action. */ + { + const struct lttng_firing_policy *cur_policy = NULL; + status = lttng_action_rotate_session_get_firing_policy( + rotate_session_action, &cur_policy); + ok(status == LTTNG_ACTION_STATUS_OK && + lttng_firing_policy_is_equal( + policy, + cur_policy), + "rotate_session action policy get"); + } + + /* Ser/des tests. */ + ret = lttng_action_serialize(rotate_session_action, &payload); + ok(ret == 0, "Action rotate_session serialized"); + + { + struct lttng_payload_view view = + lttng_payload_view_from_payload( + &payload, 0, -1); + (void) lttng_action_create_from_payload( + &view, &rotate_session_action_from_buffer); + } + ok(rotate_session_action_from_buffer, + "rotate_session action created from payload is non-null"); + + ok(lttng_action_is_equal(rotate_session_action, + rotate_session_action_from_buffer), + "Serialized and de-serialized rotate_session action are equal"); + + lttng_firing_policy_destroy(default_policy); + lttng_firing_policy_destroy(policy); + lttng_action_destroy(rotate_session_action); + lttng_action_destroy(rotate_session_action_from_buffer); + lttng_payload_reset(&payload); +} + int main(int argc, const char *argv[]) { plan_tests(NUM_TESTS); test_action_notify(); + test_action_rotate_session(); return exit_status(); } -- 2.34.1