[glib] Add some tests for mimeapps.list handling

[Matthias Clasen <matthiasc src gnome org>]

commit 011c0b488f0b2461e079a58cc4eacb4b92f23990
Author: Matthias Clasen <mclasen redhat com>
Date:   Fri Sep 9 23:43:02 2011 -0400

    Add some tests for mimeapps.list handling
    
    These tests directly look at mimeapps.list to verify that
    we make the expected changes to the database.

 gio/tests/Makefile.am |    7 ++
 gio/tests/mimeapps.c  |  212 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 219 insertions(+), 0 deletions(-)
---
diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am
index a85ea4f..1df90c7 100644
--- a/gio/tests/Makefile.am
+++ b/gio/tests/Makefile.am
@@ -74,6 +74,7 @@ TEST_PROGS +=			\
 	gdbus-bz627724		\
 	appinfo			\
 	contenttype		\
+	mimeapps		\
 	file			\
 	$(NULL)
 endif
@@ -368,6 +369,9 @@ appinfo_test_LDADD   = $(progs_ldadd)
 contenttype_SOURCES = contenttype.c
 contenttype_LDADD   = $(progs_ldadd)
 
+mimeapps_SOURCES = mimeapps.c
+mimeapps_LDADD   = $(progs_ldadd)
+
 file_SOURCES = file.c
 file_LDADD   = $(progs_ldadd)
 
@@ -527,3 +531,6 @@ DISTCLEANFILES = \
 	test.mo				\
 	gsettings.store			\
 	gschemas.compiled
+
+distclean-local:
+	rm -r xdgdatahome
diff --git a/gio/tests/mimeapps.c b/gio/tests/mimeapps.c
new file mode 100644
index 0000000..ba8b5db
--- /dev/null
+++ b/gio/tests/mimeapps.c
@@ -0,0 +1,212 @@
+#include <glib/gstdio.h>
+#include <gio/gio.h>
+#include <gio/gdesktopappinfo.h>
+
+static gboolean
+strv_equal (gchar **strv, ...)
+{
+  gint count;
+  va_list list;
+  const gchar *str;
+  gboolean res;
+
+  res = TRUE;
+  count = 0;
+  va_start (list, strv);
+  while (1)
+    {
+      str = va_arg (list, const gchar *);
+      if (str == NULL)
+        break;
+      if (g_strcmp0 (str, strv[count]) != 0)
+        {
+          res = FALSE;
+          break;
+        }
+      count++;
+    }
+  va_end (list);
+
+  if (res)
+    res = g_strv_length (strv) == count;
+
+  return res;
+}
+
+
+const gchar *myapp_data =
+  "[Desktop Entry]\n"
+  "Encoding=UTF-8\n"
+  "Version=1.0\n"
+  "Type=Application\n"
+  "Exec=my_app %f\n"
+  "Name=my app\n";
+
+const gchar *myapp2_data =
+  "[Desktop Entry]\n"
+  "Encoding=UTF-8\n"
+  "Version=1.0\n"
+  "Type=Application\n"
+  "Exec=my_app2 %f\n"
+  "Name=my app 2\n";
+
+/* Test that we handle mimeapps.list as expected.
+ * These tests are different from the ones in appinfo.c
+ * in that we directly parse mimeapps.list here
+ * to verify the results.
+ *
+ * We need to keep this test in a separate binary, since
+ * g_get_user_data_dir() doesn't get updated at runtime.
+ */
+static void
+test_mimeapps (void)
+{
+  gchar *dir;
+  gchar *xdgdir;
+  gchar *appdir;
+  gchar *mimeapps;
+  gchar *name;
+  gchar **assoc;
+  GAppInfo *appinfo;
+  GAppInfo *appinfo2;
+  GError *error = NULL;
+  GKeyFile *keyfile;
+  gchar *str;
+  gboolean res;
+
+  dir = g_get_current_dir ();
+  xdgdir = g_build_filename (dir, "xdgdatahome", NULL);
+  g_test_message ("setting XDG_DATA_HOME to '%s'\n", xdgdir);
+  g_setenv ("XDG_DATA_HOME", xdgdir, TRUE);
+  g_setenv ("XDG_DATA_DIRS", " ", TRUE);
+
+  appdir = g_build_filename (xdgdir, "applications", NULL);
+  g_test_message ("creating '%s'\n", appdir);
+  res = g_mkdir_with_parents (appdir, 0700);
+  g_assert (res == 0);
+
+  name = g_build_filename (appdir, "myapp.desktop", NULL);
+  g_test_message ("creating '%s'\n", name);
+  g_file_set_contents (name, myapp_data, -1, &error);
+  g_assert_no_error (error);
+  g_free (name);
+
+  name = g_build_filename (appdir, "myapp2.desktop", NULL);
+  g_test_message ("creating '%s'\n", name);
+  g_file_set_contents (name, myapp2_data, -1, &error);
+  g_assert_no_error (error);
+  g_free (name);
+
+  mimeapps = g_build_filename (appdir, "mimeapps.list", NULL);
+  g_test_message ("removing '%s'\n", mimeapps);
+  g_remove (mimeapps);
+
+  /* 1. add a non-default association */
+  appinfo = (GAppInfo*)g_desktop_app_info_new ("myapp.desktop");
+  g_app_info_add_supports_type (appinfo, "application/pdf", &error);
+  g_assert_no_error (error);
+
+  keyfile = g_key_file_new ();
+  g_key_file_load_from_file (keyfile, mimeapps, G_KEY_FILE_NONE, &error);
+  g_assert_no_error (error);
+
+  assoc = g_key_file_get_string_list (keyfile, "Added Associations", "application/pdf", NULL, &error);
+  g_assert_no_error (error);
+  g_assert (strv_equal (assoc, "myapp.desktop", NULL));
+  g_strfreev (assoc);
+
+  /* we've unset XDG_DATA_DIRS so there should be no default */
+  assoc = g_key_file_get_string_list (keyfile, "Default Applications", "application/pdf", NULL, &error);
+  g_assert_error (error, G_KEY_FILE_ERROR, G_KEY_FILE_ERROR_GROUP_NOT_FOUND);
+  g_clear_error (&error);
+
+  g_key_file_free (keyfile);
+
+  /* 2. add another non-default association */
+  appinfo2 = (GAppInfo*)g_desktop_app_info_new ("myapp2.desktop");
+  g_app_info_add_supports_type (appinfo2, "application/pdf", &error);
+  g_assert_no_error (error);
+
+  keyfile = g_key_file_new ();
+  g_key_file_load_from_file (keyfile, mimeapps, G_KEY_FILE_NONE, &error);
+  g_assert_no_error (error);
+
+  assoc = g_key_file_get_string_list (keyfile, "Added Associations", "application/pdf", NULL, &error);
+  g_assert_no_error (error);
+  g_assert (strv_equal (assoc, "myapp.desktop", "myapp2.desktop", NULL));
+  g_strfreev (assoc);
+
+  assoc = g_key_file_get_string_list (keyfile, "Default Applications", "application/pdf", NULL, &error);
+  g_assert_error (error, G_KEY_FILE_ERROR, G_KEY_FILE_ERROR_GROUP_NOT_FOUND);
+  g_clear_error (&error);
+
+  g_key_file_free (keyfile);
+
+  /* 3. make the first app the default */
+  g_app_info_set_as_default_for_type (appinfo, "application/pdf", &error);
+  g_assert_no_error (error);
+
+  keyfile = g_key_file_new ();
+  g_key_file_load_from_file (keyfile, mimeapps, G_KEY_FILE_NONE, &error);
+  g_assert_no_error (error);
+
+  assoc = g_key_file_get_string_list (keyfile, "Added Associations", "application/pdf", NULL, &error);
+  g_assert_no_error (error);
+  g_assert (strv_equal (assoc, "myapp.desktop", "myapp2.desktop", NULL));
+  g_strfreev (assoc);
+
+  str = g_key_file_get_string (keyfile, "Default Applications", "application/pdf", &error);
+  g_assert_no_error (error);
+  g_assert_cmpstr (str, ==, "myapp.desktop");
+
+  g_key_file_free (keyfile);
+
+  /* 4. make the second app the last used one */
+  g_app_info_set_as_last_used_for_type (appinfo2, "application/pdf", &error);
+  g_assert_no_error (error);
+
+  keyfile = g_key_file_new ();
+  g_key_file_load_from_file (keyfile, mimeapps, G_KEY_FILE_NONE, &error);
+  g_assert_no_error (error);
+
+  assoc = g_key_file_get_string_list (keyfile, "Added Associations", "application/pdf", NULL, &error);
+  g_assert_no_error (error);
+  g_assert (strv_equal (assoc, "myapp2.desktop", "myapp.desktop", NULL));
+  g_strfreev (assoc);
+
+  g_key_file_free (keyfile);
+
+  /* 5. reset everything */
+  g_app_info_reset_type_associations ("application/pdf");
+
+  keyfile = g_key_file_new ();
+  g_key_file_load_from_file (keyfile, mimeapps, G_KEY_FILE_NONE, &error);
+  g_assert_no_error (error);
+
+  res = g_key_file_has_key (keyfile, "Added Associations", "application/pdf", NULL);
+  g_assert (!res);
+
+  res = g_key_file_has_key (keyfile, "Default Applications", "application/pdf", NULL);
+  g_assert (!res);
+
+  g_key_file_free (keyfile);
+
+  g_object_unref (appinfo);
+  g_object_unref (appinfo2);
+
+  g_free (dir);
+  g_free (xdgdir);
+  g_free (mimeapps);
+  g_free (appdir);
+}
+
+int
+main (int argc, char *argv[])
+{
+  g_type_init ();
+  g_test_init (&argc, &argv, NULL);
+
+  g_test_add_func ("/appinfo/mimeapps", test_mimeapps);
+
+  return g_test_run ();
+}