diff --git a/unit/test-queue.c b/unit/test-queue.c
index 9fe15ba..95d14b7 100644
--- a/unit/test-queue.c
+++ b/unit/test-queue.c
queue_destroy(static_queue, destroy_remove);
}
+static void test_push_after(void)
+{
+ struct queue *queue;
+ unsigned int len, i;
+
+ queue = queue_new();
+ g_assert(queue != NULL);
+
+ /*
+ * Pre-populate queue. Initial elements are:
+ * [ NULL, 2, 5 ]
+ */
+ g_assert(queue_push_tail(queue, NULL));
+ g_assert(queue_push_tail(queue, UINT_TO_PTR(2)));
+ g_assert(queue_push_tail(queue, UINT_TO_PTR(5)));
+ g_assert(queue_length(queue) == 3);
+
+ /* Invalid insertion */
+ g_assert(!queue_push_after(queue, UINT_TO_PTR(6), UINT_TO_PTR(1)));
+
+ /* Valid insertions */
+ g_assert(queue_push_after(queue, NULL, UINT_TO_PTR(1)));
+ g_assert(queue_push_after(queue, UINT_TO_PTR(2), UINT_TO_PTR(3)));
+ g_assert(queue_push_after(queue, UINT_TO_PTR(3), UINT_TO_PTR(4)));
+ g_assert(queue_push_after(queue, UINT_TO_PTR(5), UINT_TO_PTR(6)));
+
+ g_assert(queue_peek_head(queue) == NULL);
+ g_assert(queue_peek_tail(queue) == UINT_TO_PTR(6));
+
+ /*
+ * Queue should contain 7 elements:
+ * [ NULL, 1, 2, 3, 4, 5, 6 ]
+ */
+ len = queue_length(queue);
+ g_assert(len == 7);
+
+ for (i = 0; i < 7; i++)
+ g_assert(queue_pop_head(queue) == UINT_TO_PTR(i));
+
+ /* Test with identical elements */
+ g_assert(queue_push_head(queue, UINT_TO_PTR(1)));
+ g_assert(queue_push_head(queue, UINT_TO_PTR(1)));
+ g_assert(queue_push_head(queue, UINT_TO_PTR(1)));
+ g_assert(queue_push_after(queue, UINT_TO_PTR(1), UINT_TO_PTR(0)));
+
+ g_assert(queue_pop_head(queue) == UINT_TO_PTR(1));
+ g_assert(queue_pop_head(queue) == UINT_TO_PTR(0));
+ g_assert(queue_pop_head(queue) == UINT_TO_PTR(1));
+ g_assert(queue_pop_head(queue) == UINT_TO_PTR(1));
+}
+
int main(int argc, char *argv[])
{
g_test_init(&argc, &argv, NULL);
g_test_add_func("/queue/foreach_destroy", test_foreach_destroy);
g_test_add_func("/queue/foreach_remove_all", test_foreach_remove_all);
g_test_add_func("/queue/destroy_remove", test_destroy_remove);
+ g_test_add_func("/queue/push_after", test_push_after);
return g_test_run();
}