From 1f47c51e592563339c13256ea02b8fa768b8fde2 Mon Sep 17 00:00:00 2001

From: Robbie Harwood <rharwood@redhat.com>

Date: Thu, 30 Aug 2018 15:34:31 -0400

Subject: [PATCH] Add cmocka unit tests for ipa otpd queue code

Reviewed-By: Florence Blanc-Renaud <frenaud@redhat.com>

---

 daemons/ipa-otpd/Makefile.am                  |  12 +

 .../ipa-otpd/ipa_otpd_queue_cmocka_tests.c    | 212 ++++++++++++++++++

 2 files changed, 224 insertions(+)

 create mode 100644 daemons/ipa-otpd/ipa_otpd_queue_cmocka_tests.c

diff --git a/daemons/ipa-otpd/Makefile.am b/daemons/ipa-otpd/Makefile.am

index 9ba6237566634cfa6a1e84c545ee1fc9cf2dc4f6..9ccc450a5e00db90b698ff9b25a496d09f0ff6e5 100644

--- a/daemons/ipa-otpd/Makefile.am

+++ b/daemons/ipa-otpd/Makefile.am

@@ -20,3 +20,15 @@ ipa_otpd_SOURCES = bind.c forward.c main.c parse.c query.c queue.c stdio.c

 	     $< > $@

 CLEANFILES = $(systemdsystemunit_DATA)

+

+TESTS =

+check_PROGRAMS =

+

+if HAVE_CMOCKA

+TESTS += queue_tests

+check_PROGRAMS += queue_tests

+endif

+

+queue_tests_SOURCES = ipa_otpd_queue_cmocka_tests.c queue.c

+queue_tests_CFLAGS = $(CMOCKA_CFLAGS)

+queue_tests_LDADD = $(CMOCKA_LIBS)

diff --git a/daemons/ipa-otpd/ipa_otpd_queue_cmocka_tests.c b/daemons/ipa-otpd/ipa_otpd_queue_cmocka_tests.c

new file mode 100644

index 0000000000000000000000000000000000000000..068431e6475bb74b01acbcab22115915dec1a278

--- /dev/null

+++ b/daemons/ipa-otpd/ipa_otpd_queue_cmocka_tests.c

@@ -0,0 +1,212 @@

+/*

+ * FreeIPA 2FA companion daemon - internal queue tests

+ *

+ * Author: Robbie Harwood <rharwood@redhat.com>

+ *

+ * Copyright (C) 2018  Robbie Harwood, Red Hat

+ * see file 'COPYING' for use and warranty information

+ *

+ * This program is free software you can redistribute it and/or modify it

+ * under the terms of the GNU General Public License as published by the Free

+ * Software Foundation, either version 3 of the License, or (at your option)

+ * any later version.

+ *

+ * This program is distributed in the hope that it will be useful, but WITHOUT

+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for

+ * more details.

+ *

+ * You should have received a copy of the GNU General Public License along

+ * with this program.  If not, see <http://www.gnu.org/licenses/>.

+ */

+

+#include <setjmp.h>

+#include <stdarg.h>

+#include <stddef.h>

+

+#include <cmocka.h>

+

+#include "internal.h"

+

+/* Bypass otpd queue allocation/freeing to avoid calling into LDAP and

+ * krad.  No effort is made to make the types match. */

+static struct otpd_queue_item *new_elt(int id)

+{

+    krb5_error_code ret;

+    struct otpd_queue_item *e = NULL;

+

+    ret = otpd_queue_item_new(NULL, &e);

+    assert_int_equal(ret, 0);

+    assert_ptr_not_equal(e, NULL);

+

+    e->msgid = id;

+    return e;

+}

+static void free_elt(struct otpd_queue_item **e)

+{

+    assert_ptr_not_equal(e, NULL);

+    free(*e);

+    *e = NULL;

+}

+static void free_elts(struct otpd_queue *q)

+{

+    assert_ptr_not_equal(q, NULL);

+    for (struct otpd_queue_item *e = otpd_queue_pop(q); e != NULL;

+         e = otpd_queue_pop(q))

+        free_elt(&e);

+}

+#define otpd_queue_item_new new_elt

+#define otpd_queue_item_free free_elt

+#define otpd_queue_free_items free_elts

+

+static void assert_elt_equal(struct otpd_queue_item *e1,

+                             struct otpd_queue_item *e2)

+{

+    if (e1 == NULL && e2 == NULL)

+        return;

+    assert_ptr_not_equal(e1, NULL);

+    assert_ptr_not_equal(e2, NULL);

+    assert_int_equal(e1->msgid, e2->msgid);

+}

+

+static void test_single_insert()

+{

+    struct otpd_queue q = { NULL };

+    struct otpd_queue_item *ein, *eout;

+

+    ein = new_elt(0);

+    otpd_queue_push(&q, ein);

+

+    eout = otpd_queue_peek(&q);

+    assert_elt_equal(ein, eout);

+

+    eout = otpd_queue_pop(&q);

+    assert_elt_equal(ein, eout);

+    free_elt(&eout);

+

+    eout = otpd_queue_pop(&q);

+    assert_ptr_equal(eout, NULL);

+

+    free_elts(&q);

+}

+

+static void test_jump_insert()

+{

+    struct otpd_queue q = { NULL };

+    struct otpd_queue_item *echeck;

+

+    for (int i = 0; i < 3; i++) {

+        struct otpd_queue_item *e = new_elt(i);

+        otpd_queue_push_head(&q, e);

+

+        echeck = otpd_queue_peek(&q);

+        assert_elt_equal(e, echeck);

+    }

+

+    free_elts(&q);

+}

+

+static void test_garbage_insert()

+{

+    struct otpd_queue q = { NULL };

+    struct otpd_queue_item *e, *g;

+

+    g = new_elt(0);

+    g->next = g;

+    otpd_queue_push(&q, g);

+

+    e = otpd_queue_peek(&q);

+    assert_ptr_equal(e->next, NULL);

+

+    free_elts(&q);

+}

+

+static void test_removal()

+{

+    struct otpd_queue q = { NULL };

+

+    for (int i = 0; i < 3; i++) {

+        struct otpd_queue_item *e = new_elt(i);

+        otpd_queue_push(&q, e);

+    }

+    for (int i = 0; i < 3; i++) {

+        struct otpd_queue_item *e = otpd_queue_pop(&q);

+        assert_ptr_not_equal(e, NULL);

+        assert_ptr_equal(e->next, NULL);

+        assert_int_equal(e->msgid, i);

+        free_elt(&e);

+    }

+}

+

+static void pick_id(struct otpd_queue *q, int msgid)

+{

+    struct otpd_queue_item *e;

+

+    e = otpd_queue_pop_msgid(q, msgid);

+    assert_int_equal(e->msgid, msgid);

+    assert_ptr_equal(e->next, NULL);

+    free_elt(&e);

+    e = otpd_queue_pop_msgid(q, msgid);

+    assert_ptr_equal(e, NULL);

+}

+static void test_pick_removal()

+{

+    struct otpd_queue q = { NULL };

+

+    for (int i = 0; i < 4; i++) {

+        struct otpd_queue_item *e = new_elt(i);

+        otpd_queue_push(&q, e);

+    }

+

+    pick_id(&q, 0); /* first */

+    pick_id(&q, 2); /* middle */

+    pick_id(&q, 3); /* last */

+    pick_id(&q, 1); /* singleton */

+

+    free_elts(&q);

+}

+

+static void test_iter()

+{

+    krb5_error_code ret;

+    struct otpd_queue q = { NULL };

+    const struct otpd_queue *queues[3];

+    struct otpd_queue_iter *iter = NULL;

+    const krad_packet *p = NULL;

+

+    for (ptrdiff_t i = 1; i <= 3; i++) {

+        struct otpd_queue_item *e = new_elt(i);

+        e->req = (void *)i;

+        otpd_queue_push(&q, e);

+    }

+

+    queues[0] = &q;

+    queues[1] = &q;

+    queues[2] = NULL;

+    ret = otpd_queue_iter_new(queues, &iter);

+    assert_int_equal(ret, 0);

+    assert_ptr_not_equal(iter, NULL);

+

+    for (ptrdiff_t i = 0; i < 6; i++) {

+        p = otpd_queue_iter_func(iter, FALSE);

+        assert_ptr_equal(p, (void *) (i % 3 + 1));

+    }

+    p = otpd_queue_iter_func(iter, FALSE);

+    assert_ptr_equal(p, NULL);

+

+    free_elts(&q);

+}

+

+int main(int argc, char *argv[])

+{

+    const struct CMUnitTest tests[] = {

+        cmocka_unit_test(test_single_insert),

+        cmocka_unit_test(test_jump_insert),

+        cmocka_unit_test(test_garbage_insert),

+        cmocka_unit_test(test_removal),

+        cmocka_unit_test(test_pick_removal),

+        cmocka_unit_test(test_iter),

+    };

+

+    return cmocka_run_group_tests(tests, NULL, NULL);

+}

-- 

2.17.1