[tracker/tracker-0.12] tests/libtracker-miner: unittest for TrackerTaskPool

[Martyn James Russell <mr src gnome org>]

commit 8e249ab27e258acdcdeec6643039ecb1d4cf5ea2
Author: Ivan Frade <ivan frade gmail com>
Date:   Tue Sep 13 17:22:16 2011 +0300

    tests/libtracker-miner: unittest for TrackerTaskPool

 tests/libtracker-miner/Makefile.am              |    6 +-
 tests/libtracker-miner/tracker-task-pool-test.c |  180 +++++++++++++++++++++++
 2 files changed, 185 insertions(+), 1 deletions(-)
---
diff --git a/tests/libtracker-miner/Makefile.am b/tests/libtracker-miner/Makefile.am
index 75c8737..6a00089 100644
--- a/tests/libtracker-miner/Makefile.am
+++ b/tests/libtracker-miner/Makefile.am
@@ -22,7 +22,8 @@ TEST_PROGS +=                                          \
 	tracker-password-provider-test                 \
 	tracker-thumbnailer-test                       \
 	tracker-monitor-test			       \
-	tracker-priority-queue-test
+	tracker-priority-queue-test		       \
+	tracker-task-pool-test
 
 AM_CPPFLAGS =                                          \
 	$(BUILD_CFLAGS)                                \
@@ -82,6 +83,9 @@ tracker_monitor_test_SOURCES =                         \
 tracker_priority_queue_test_SOURCES = 		       \
 	tracker-priority-queue-test.c
 
+tracker_task_pool_test_SOURCES = 		       \
+	tracker-task-pool-test.c
+
 noinst_HEADERS =                                       \
 	$(libtracker_miner_marshal_headers)            \
 	$(libtracker_miner_monitor_headers)            \
diff --git a/tests/libtracker-miner/tracker-task-pool-test.c b/tests/libtracker-miner/tracker-task-pool-test.c
new file mode 100644
index 0000000..578c144
--- /dev/null
+++ b/tests/libtracker-miner/tracker-task-pool-test.c
@@ -0,0 +1,180 @@
+/*
+ * Copyright (C) 2011, Nokia <ivan frade nokia com>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+ * Boston, MA  02110-1301, USA.
+ *
+ */
+#include <glib.h>
+#include <libtracker-miner/tracker-task-pool.h>
+
+void
+test_task_pool_limit_set (void)
+{
+        TrackerTaskPool *pool;
+
+        pool = tracker_task_pool_new (5);
+        g_assert_cmpint (tracker_task_pool_get_limit(pool), ==, 5);
+
+        tracker_task_pool_set_limit (pool, 3);
+        g_assert_cmpint (tracker_task_pool_get_limit(pool), ==, 3);
+
+        g_assert (tracker_task_pool_limit_reached (pool) == FALSE);
+        g_object_unref (pool);
+}
+
+void
+add_task (TrackerTaskPool *pool, const gchar *filename, gint expected_size, gboolean hit_limit)
+{
+        TrackerTask *task;
+
+        task = tracker_task_new (g_file_new_for_path (filename), NULL, NULL);
+        tracker_task_pool_add (pool, task);
+
+        g_assert_cmpint (tracker_task_pool_get_size (pool), ==, expected_size);
+        g_assert (tracker_task_pool_limit_reached (pool) == hit_limit);
+
+        g_object_unref (tracker_task_get_file (task));
+        tracker_task_unref (task);
+}
+
+void
+remove_task (TrackerTaskPool *pool, const gchar *filename, gint expected_size, gboolean hit_limit)
+{
+        TrackerTask *task;
+
+        task = tracker_task_new (g_file_new_for_path (filename), NULL, NULL);
+        tracker_task_pool_remove (pool, task);
+
+        g_assert_cmpint (tracker_task_pool_get_size (pool), ==, expected_size);
+        g_assert (hit_limit == tracker_task_pool_limit_reached (pool));
+
+        g_object_unref (tracker_task_get_file (task));
+        tracker_task_unref (task);
+}
+
+void
+test_task_pool_add_remove (void)
+{
+        TrackerTaskPool *pool;
+
+        pool = tracker_task_pool_new (3);
+
+        /* Additions ... */
+        add_task (pool, "/dev/null", 1, FALSE);
+        add_task (pool, "/dev/null2", 2, FALSE);
+        add_task (pool, "/dev/null3", 3, TRUE);
+
+        /* We can go over the limit */
+        add_task (pool, "/dev/null4", 4, TRUE);
+
+        /* Remove something that doesn't exist */
+        remove_task (pool, "/dev/null/imNotInThePool", 4, TRUE);
+
+        /* Removals ... (in different order)*/
+        remove_task (pool, "/dev/null4", 3, TRUE);
+        remove_task (pool, "/dev/null2", 2, FALSE);
+        remove_task (pool, "/dev/null3", 1, FALSE);
+        remove_task (pool, "/dev/null", 0, FALSE);
+
+        /* Remove in empty queue */
+        remove_task (pool, "/dev/null/random", 0, FALSE);
+
+        g_object_unref (pool);
+}
+
+void
+test_task_pool_find ()
+{
+        TrackerTaskPool *pool;
+        TrackerTask     *task;
+        GFile           *goal;
+
+        pool = tracker_task_pool_new (3);
+
+        add_task (pool, "/dev/null", 1, FALSE);
+        add_task (pool, "/dev/null2", 2, FALSE);
+        add_task (pool, "/dev/null3", 3, TRUE);
+
+        /* Search first, last, in the middle... */
+        goal = g_file_new_for_path ("/dev/null2");
+        task = tracker_task_pool_find (pool, goal);
+        g_assert (task);
+        g_object_unref (goal);
+
+        goal = g_file_new_for_path ("/dev/null");
+        task = tracker_task_pool_find (pool, goal);
+        g_assert (task);
+        g_object_unref (goal);
+
+        goal = g_file_new_for_path ("/dev/null3");
+        task = tracker_task_pool_find (pool, goal);
+        g_assert (task);
+        g_object_unref (goal);
+
+        goal = g_file_new_for_path ("/dev/thisDoesntExists");
+        task = tracker_task_pool_find (pool, goal);
+        g_assert (task == NULL);
+        g_object_unref (goal);
+
+        g_object_unref (pool);
+}
+
+void
+count_elements_cb (gpointer data, gpointer user_data)
+{
+        gint *counter = (gint*)user_data;
+        (*counter) += 1;
+}
+
+void
+test_task_pool_foreach ()
+{
+        TrackerTaskPool *pool;
+        int              counter = 0;
+
+        pool = tracker_task_pool_new (3);
+
+        add_task (pool, "/dev/null", 1, FALSE);
+        add_task (pool, "/dev/null2", 2, FALSE);
+        add_task (pool, "/dev/null3", 3, TRUE);
+
+        tracker_task_pool_foreach (pool, count_elements_cb, &counter);
+
+        g_assert_cmpint (counter, ==, 3);
+}
+
+
+gint
+main (gint argc, gchar **argv)
+{
+        g_type_init ();
+        g_thread_init (NULL);
+        g_test_init (&argc, &argv, NULL);
+
+        g_test_add_func ("/libtracker-miner/tracker-task-pool/limit_set",
+                         test_task_pool_limit_set);
+
+        g_test_add_func ("/libtracker-miner/tracker-task-pool/add_remove",
+                         test_task_pool_add_remove);
+
+        g_test_add_func ("/libtracker-miner/tracker-task-pool/find",
+                         test_task_pool_find);
+ 
+        g_test_add_func ("/libtracker-miner/tracker-task-pool/foreach",
+                         test_task_pool_foreach);
+
+        return g_test_run ();
+}