Make ast_taskprocessor_listener opaque.
[asterisk/asterisk.git] / tests / test_taskprocessor.c
index 1d6cbb8..c717ad0 100644 (file)
 #include "asterisk/module.h"
 #include "asterisk/astobj2.h"
 
+/*!
+ * \brief userdata associated with baseline taskprocessor test
+ */
 struct task_data {
+       /* Condition used to signal to queuing thread that task was executed */
        ast_cond_t cond;
+       /* Lock protecting the condition */
        ast_mutex_t lock;
+       /*! Boolean indicating that the task was run */
        int task_complete;
 };
 
+/*!
+ * \brief Queued task for baseline test.
+ *
+ * The task simply sets a boolean to indicate the
+ * task has been run and then signals a condition
+ * saying it's complete
+ */
 static int task(void *data)
 {
        struct task_data *task_data = data;
@@ -51,6 +64,13 @@ static int task(void *data)
        return 0;
 }
 
+/*!
+ * \brief Baseline test for default taskprocessor
+ *
+ * This test ensures that when a task is added to a taskprocessor that
+ * has been allocated with a default listener that the task gets executed
+ * as expected
+ */
 AST_TEST_DEFINE(default_taskprocessor)
 {
        struct ast_taskprocessor *tps;
@@ -66,7 +86,7 @@ AST_TEST_DEFINE(default_taskprocessor)
                info->category = "/main/taskprocessor/";
                info->summary = "Test of default taskproccesor";
                info->description =
-                       "Ensures that queued tasks are executed.";
+                       "Ensures that a queued task gets executed.";
                return AST_TEST_NOT_RUN;
        case TEST_EXECUTE:
                break;
@@ -96,6 +116,7 @@ AST_TEST_DEFINE(default_taskprocessor)
                        break;
                }
        }
+       ast_mutex_unlock(&task_data.lock);
 
        if (!task_data.task_complete) {
                ast_test_status_update(test, "Queued task did not execute!\n");
@@ -112,13 +133,26 @@ test_end:
 
 #define NUM_TASKS 20000
 
+/*!
+ * \brief Relevant data associated with taskprocessor load test
+ */
 static struct load_task_data {
+       /*! Condition used to indicate a task has completed executing */
        ast_cond_t cond;
+       /*! Lock used to protect the condition */
        ast_mutex_t lock;
+       /*! Counter of the number of completed tasks */
        int tasks_completed;
+       /*! Storage for task-specific data */
        int task_rand[NUM_TASKS];
 } load_task_results;
 
+/*!
+ * \brief a queued task to be used in the taskprocessor load test
+ *
+ * The task increments the number of tasks executed and puts the passed-in
+ * data into the next slot in the array of random data.
+ */
 static int load_task(void *data)
 {
        int *randdata = data;
@@ -128,6 +162,13 @@ static int load_task(void *data)
        return 0;
 }
 
+/*!
+ * \brief Load test for taskprocessor with default listener
+ *
+ * This test queues a large number of tasks, each with random data associated.
+ * The test ensures that all of the tasks are run and that the tasks are executed
+ * in the same order that they were queued
+ */
 AST_TEST_DEFINE(default_taskprocessor_load)
 {
        struct ast_taskprocessor *tps;
@@ -144,7 +185,7 @@ AST_TEST_DEFINE(default_taskprocessor_load)
                info->category = "/main/taskprocessor/";
                info->summary = "Load test of default taskproccesor";
                info->description =
-                       "Ensure that many queued tasks are executed.";
+                       "Ensure that a large number of queued tasks are executed in the proper order.";
                return AST_TEST_NOT_RUN;
        case TEST_EXECUTE:
                break;
@@ -178,6 +219,7 @@ AST_TEST_DEFINE(default_taskprocessor_load)
                        break;
                }
        }
+       ast_mutex_unlock(&load_task_results.lock);
 
        if (load_task_results.tasks_completed != NUM_TASKS) {
                ast_test_status_update(test, "Unexpected number of tasks executed. Expected %d but got %d\n",
@@ -201,14 +243,24 @@ test_end:
        return res;
 }
 
+/*!
+ * \brief Private data for the test taskprocessor listener
+ */
 struct test_listener_pvt {
+       /* Counter of number of tasks pushed to the queue */
        int num_pushed;
+       /* Counter of number of times the queue was emptied */
        int num_emptied;
+       /* Counter of number of times that a pushed task occurred on an empty queue */
        int num_was_empty;
+       /* Boolean indicating whether the shutdown callback was called */
        int shutdown;
 };
 
-static void *test_alloc(struct ast_taskprocessor_listener *listener)
+/*!
+ * \brief test taskprocessor listener's alloc callback
+ */
+static void *test_listener_pvt_alloc(void)
 {
        struct test_listener_pvt *pvt;
 
@@ -216,46 +268,74 @@ static void *test_alloc(struct ast_taskprocessor_listener *listener)
        return pvt;
 }
 
+/*!
+ * \brief test taskprocessor listener's start callback
+ */
+static int test_start(struct ast_taskprocessor_listener *listener)
+{
+       return 0;
+}
+
+/*!
+ * \brief test taskprocessor listener's task_pushed callback
+ *
+ * Adjusts private data's stats as indicated by the parameters.
+ */
 static void test_task_pushed(struct ast_taskprocessor_listener *listener, int was_empty)
 {
-       struct test_listener_pvt *pvt = listener->private_data;
+       struct test_listener_pvt *pvt = ast_taskprocessor_listener_get_user_data(listener);
        ++pvt->num_pushed;
        if (was_empty) {
                ++pvt->num_was_empty;
        }
 }
 
+/*!
+ * \brief test taskprocessor listener's emptied callback.
+ */
 static void test_emptied(struct ast_taskprocessor_listener *listener)
 {
-       struct test_listener_pvt *pvt = listener->private_data;
+       struct test_listener_pvt *pvt = ast_taskprocessor_listener_get_user_data(listener);
        ++pvt->num_emptied;
 }
 
+/*!
+ * \brief test taskprocessor listener's shutdown callback.
+ */
 static void test_shutdown(struct ast_taskprocessor_listener *listener)
 {
-       struct test_listener_pvt *pvt = listener->private_data;
+       struct test_listener_pvt *pvt = ast_taskprocessor_listener_get_user_data(listener);
        pvt->shutdown = 1;
 }
 
-static void test_destroy(void *private_data)
-{
-       struct test_listener_pvt *pvt = private_data;
-       ast_free(pvt);
-}
-
 static const struct ast_taskprocessor_listener_callbacks test_callbacks = {
-       .alloc = test_alloc,
+       .start = test_start,
        .task_pushed = test_task_pushed,
        .emptied = test_emptied,
        .shutdown = test_shutdown,
-       .destroy = test_destroy,
 };
 
+/*!
+ * \brief Queued task for taskprocessor listener test.
+ *
+ * Does nothing.
+ */
 static int listener_test_task(void *ignore)
 {
        return 0;
 }
 
+/*!
+ * \brief helper to ensure that statistics the listener is keeping are what we expect
+ *
+ * \param test The currently-running test
+ * \param pvt The private data for the taskprocessor listener
+ * \param num_pushed The expected current number of tasks pushed to the processor
+ * \param num_emptied The expected current number of times the taskprocessor has become empty
+ * \param num_was_empty The expected current number of times that tasks were pushed to an empty taskprocessor
+ * \retval -1 Stats were not as expected
+ * \retval 0 Stats were as expected
+ */
 static int check_stats(struct ast_test *test, const struct test_listener_pvt *pvt, int num_pushed, int num_emptied, int num_was_empty)
 {
        if (pvt->num_pushed != num_pushed) {
@@ -279,11 +359,20 @@ static int check_stats(struct ast_test *test, const struct test_listener_pvt *pv
        return 0;
 }
 
+/*!
+ * \brief Test for a taskprocessor with custom listener.
+ *
+ * This test pushes tasks to a taskprocessor with a custom listener, executes the taskss,
+ * and destroys the taskprocessor.
+ *
+ * The test ensures that the listener's callbacks are called when expected and that the data
+ * being passed in is accurate.
+ */
 AST_TEST_DEFINE(taskprocessor_listener)
 {
-       struct ast_taskprocessor *tps;
-       struct ast_taskprocessor_listener *listener;
-       struct test_listener_pvt *pvt;
+       struct ast_taskprocessor *tps = NULL;
+       struct ast_taskprocessor_listener *listener = NULL;
+       struct test_listener_pvt *pvt = NULL;
        enum ast_test_result_state res = AST_TEST_PASS;
 
        switch (cmd) {
@@ -298,10 +387,17 @@ AST_TEST_DEFINE(taskprocessor_listener)
                break;
        }
 
-       listener = ast_taskprocessor_listener_alloc(&test_callbacks);
+       pvt = test_listener_pvt_alloc();
+       if (!pvt) {
+               ast_test_status_update(test, "Unable to allocate test taskprocessor listener user data\n");
+               return AST_TEST_FAIL;
+       }
+
+       listener = ast_taskprocessor_listener_alloc(&test_callbacks, pvt);
        if (!listener) {
                ast_test_status_update(test, "Unable to allocate test taskprocessor listener\n");
-               return AST_TEST_FAIL;
+               res = AST_TEST_FAIL;
+               goto test_exit;
        }
 
        tps = ast_taskprocessor_create_with_listener("test_listener", listener);
@@ -311,8 +407,6 @@ AST_TEST_DEFINE(taskprocessor_listener)
                goto test_exit;
        }
 
-       pvt = listener->private_data;
-
        ast_taskprocessor_push(tps, listener_test_task, NULL);
 
        if (check_stats(test, pvt, 1, 0, 1) < 0) {
@@ -349,9 +443,10 @@ AST_TEST_DEFINE(taskprocessor_listener)
        }
 
 test_exit:
-       ao2_ref(listener, -1);
+       ao2_cleanup(listener);
        /* This is safe even if tps is NULL */
        ast_taskprocessor_unreference(tps);
+       ast_free(pvt);
        return res;
 }