Add a few unit tests for sr_session_trigger_{get,set}.

[libsigrok.git] / tests / check_session.c

diff --git a/tests/check_session.c b/tests/check_session.c
index e48e1d468af08f186a5966c9bebd9652c868e96b..699f58360c4f7bafd5e1151004bae88bf9c57d44 100644 (file)
--- a/tests/check_session.c
+++ b/tests/check_session.c
@@ -23,24 +23,6 @@
 #include "../include/libsigrok/libsigrok.h"
 #include "lib.h"
 
-static struct sr_context *sr_ctx;
-
-static void setup(void)
-{
-       int ret;
-
-       ret = sr_init(&sr_ctx);
-       fail_unless(ret == SR_OK, "sr_init() failed: %d.", ret);
-}
-
-static void teardown(void)
-{
-       int ret;
-
-       ret = sr_exit(sr_ctx);
-       fail_unless(ret == SR_OK, "sr_exit() failed: %d.", ret);
-}
-
 /*
  * Check whether sr_session_new() works.
  * If it returns != SR_OK (or segfaults) this test will fail.
@@ -136,6 +118,73 @@ START_TEST(test_session_destroy_bogus)
 }
 END_TEST
 
+START_TEST(test_session_trigger_set_get)
+{
+       int ret;
+       struct sr_session *sess;
+       struct sr_trigger *t1, *t2;
+
+       sr_session_new(&sess);
+       t1 = sr_trigger_new("T1");
+
+       /* Set a trigger and see if getting it works OK. */
+       ret = sr_session_trigger_set(sess, t1);
+       fail_unless(ret == SR_OK);
+       t2 = sr_session_trigger_get(sess);
+       fail_unless(t2 != NULL);
+       fail_unless(t1 == t2);
+       fail_unless(g_slist_length(t1->stages) == g_slist_length(t2->stages));
+       fail_unless(!strcmp(t1->name, t2->name));
+
+       sr_session_destroy(sess);
+}
+END_TEST
+
+START_TEST(test_session_trigger_set_get_null)
+{
+       int ret;
+       struct sr_session *sess;
+       struct sr_trigger *t;
+
+       sr_session_new(&sess);
+
+       /* Adding a NULL trigger is allowed. */
+       ret = sr_session_trigger_set(sess, NULL);
+       fail_unless(ret == SR_OK);
+       t = sr_session_trigger_get(sess);
+       fail_unless(t == NULL);
+
+       sr_session_destroy(sess);
+}
+END_TEST
+
+START_TEST(test_session_trigger_set_null)
+{
+       int ret;
+       struct sr_trigger *t;
+
+       t = sr_trigger_new("T1");
+
+       /* NULL session, must not segfault. */
+       ret = sr_session_trigger_set(NULL, t);
+       fail_unless(ret == SR_ERR_ARG);
+
+       /* NULL session and NULL trigger, must not segfault. */
+       ret = sr_session_trigger_set(NULL, NULL);
+       fail_unless(ret == SR_ERR_ARG);
+}
+END_TEST
+
+START_TEST(test_session_trigger_get_null)
+{
+       struct sr_trigger *t;
+
+       /* NULL session, must not segfault. */
+       t = sr_session_trigger_get(NULL);
+       fail_unless(t == NULL);
+}
+END_TEST
+
 Suite *suite_session(void)
 {
        Suite *s;
@@ -144,7 +193,7 @@ Suite *suite_session(void)
        s = suite_create("session");
 
        tc = tcase_create("new_destroy");
-       tcase_add_checked_fixture(tc, setup, teardown);
+       tcase_add_checked_fixture(tc, srtest_setup, srtest_teardown);
        tcase_add_test(tc, test_session_new);
        tcase_add_test(tc, test_session_new_bogus);
        tcase_add_test(tc, test_session_new_multiple);
@@ -152,5 +201,13 @@ Suite *suite_session(void)
        tcase_add_test(tc, test_session_destroy_bogus);
        suite_add_tcase(s, tc);
 
+       tc = tcase_create("trigger");
+       tcase_add_checked_fixture(tc, srtest_setup, srtest_teardown);
+       tcase_add_test(tc, test_session_trigger_set_get);
+       tcase_add_test(tc, test_session_trigger_set_get_null);
+       tcase_add_test(tc, test_session_trigger_set_null);
+       tcase_add_test(tc, test_session_trigger_get_null);
+       suite_add_tcase(s, tc);
+
        return s;
 }