[glib] Improve file measure testcase

[Matthias Clasen <matthiasc src gnome org>]

commit 72eae6edddf0d2200a64c1a0a68079ced5ff48a6
Author: Matthias Clasen <mclasen redhat com>
Date:   Sat Nov 30 22:51:03 2013 -0500

    Improve file measure testcase
    
    Make the testcase compare the byte size to what is reported
    by du. Also add a test for the async api, and mak eit test
    the progress reporting callback.

 gio/tests/file.c |  166 ++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 162 insertions(+), 4 deletions(-)
---
diff --git a/gio/tests/file.c b/gio/tests/file.c
index 8ca24ed..cab1835 100644
--- a/gio/tests/file.c
+++ b/gio/tests/file.c
@@ -806,11 +806,60 @@ test_copy_preserve_mode (void)
 }
 #endif
 
+static gchar *
+splice_to_string (GInputStream   *stream,
+                  GError        **error)
+{
+  GMemoryOutputStream *buffer = NULL;
+  char *ret = NULL;
+
+  buffer = (GMemoryOutputStream*)g_memory_output_stream_new (NULL, 0, g_realloc, g_free);
+  if (g_output_stream_splice ((GOutputStream*)buffer, stream, 0, NULL, error) < 0)
+    goto out;
+
+  if (!g_output_stream_write ((GOutputStream*)buffer, "\0", 1, NULL, error))
+    goto out;
+
+  if (!g_output_stream_close ((GOutputStream*)buffer, NULL, error))
+    goto out;
+
+  ret = g_memory_output_stream_steal_data (buffer);
+ out:
+  g_clear_object (&buffer);
+  return ret;
+}
+
+static guint64
+get_size_from_du (const gchar *path)
+{
+  GSubprocess *du;
+  gchar *result;
+  gchar *endptr;
+  guint64 size;
+  GError *error = NULL;
+
+  du = g_subprocess_new (G_SUBPROCESS_FLAGS_STDOUT_PIPE,
+                         &error,
+                         "du", "--bytes", "-s", path, NULL);
+  g_assert_no_error (error);
+
+  result = splice_to_string (g_subprocess_get_stdout_pipe (du), &error);
+  g_assert_no_error (error);
+
+  size = g_ascii_strtoll (result, &endptr, 10);
+
+  g_object_unref (du);
+  g_free (result);
+
+  return size;
+}
+
 static void
 test_measure (void)
 {
   GFile *file;
   guint64 size;
+  guint64 num_bytes;
   guint64 num_dirs;
   guint64 num_files;
   GError *error = NULL;
@@ -819,25 +868,133 @@ test_measure (void)
 
   path = g_test_build_filename (G_TEST_DIST, "desktop-files", NULL);
   file = g_file_new_for_path (path);
-  g_free (path);
+
+  if (g_find_program_in_path ("du"))
+    {
+      size = get_size_from_du (path);
+    }
+  else
+    {
+      g_test_message ("du not found, skipping byte measurement");
+      size = 0;
+    }
 
   ok = g_file_measure_disk_usage (file,
-                                  G_FILE_MEASURE_NONE,
+                                  G_FILE_MEASURE_APPARENT_SIZE,
                                   NULL,
                                   NULL,
                                   NULL,
-                                  &size,
+                                  &num_bytes,
                                   &num_dirs,
                                   &num_files,
                                   &error);
   g_assert (ok);
   g_assert_no_error (error);
 
-  g_assert_cmpuint (size, ==, 155648);
+  if (size > 0)
+    g_assert_cmpuint (num_bytes, ==, size);
   g_assert_cmpuint (num_dirs, ==, 6);
   g_assert_cmpuint (num_files, ==, 30);
 
   g_object_unref (file);
+  g_free (path);
+}
+
+typedef struct {
+  guint64 expected_bytes;
+  guint64 expected_dirs;
+  guint64 expected_files;
+  gint progress_count;
+  guint64 progress_bytes;
+  guint64 progress_dirs;
+  guint64 progress_files;
+} MeasureData;
+
+static void
+measure_progress (gboolean reporting,
+                  guint64  current_size,
+                  guint64  num_dirs,
+                  guint64  num_files,
+                  gpointer user_data)
+{
+  MeasureData *data = user_data;
+
+  data->progress_count += 1;
+
+  g_assert_cmpuint (current_size, >=, data->progress_bytes);
+  g_assert_cmpuint (num_dirs, >=, data->progress_dirs);
+  g_assert_cmpuint (num_files, >=, data->progress_files);
+
+  data->progress_bytes = current_size;
+  data->progress_dirs = num_dirs;
+  data->progress_files = num_files;
+}
+
+static void
+measure_done (GObject      *source,
+              GAsyncResult *res,
+              gpointer      user_data)
+{
+  MeasureData *data = user_data;
+  guint64 num_bytes, num_dirs, num_files;
+  GError *error = NULL;
+  gboolean ok;
+
+  ok = g_file_measure_disk_usage_finish (G_FILE (source), res, &num_bytes, &num_dirs, &num_files, &error);
+  g_assert (ok);
+  g_assert_no_error (error);
+
+  if (data->expected_bytes > 0)
+    g_assert_cmpuint (data->expected_bytes, ==, num_bytes);
+  g_assert_cmpuint (data->expected_dirs, ==, num_dirs);
+  g_assert_cmpuint (data->expected_files, ==, num_files);
+
+  g_assert_cmpuint (data->progress_count, >, 0);
+  g_assert_cmpuint (num_bytes, >=, data->progress_bytes);
+  g_assert_cmpuint (num_dirs, >=, data->progress_dirs);
+  g_assert_cmpuint (num_files, >=, data->progress_files);
+
+  g_free (data);
+  g_object_unref (source);
+}
+
+static void
+test_measure_async (void)
+{
+  gchar *path;
+  GFile *file;
+  MeasureData *data;
+
+  data = g_new (MeasureData, 1);
+
+  data->progress_count = 0;
+  data->progress_bytes = 0;
+  data->progress_files = 0;
+  data->progress_dirs = 0;
+
+  path = g_test_build_filename (G_TEST_DIST, "desktop-files", NULL);
+  file = g_file_new_for_path (path);
+
+  if (g_find_program_in_path ("du"))
+    {
+      data->expected_bytes = get_size_from_du (path);
+    }
+  else
+    {
+      g_test_message ("du not found, skipping byte measurement");
+      data->expected_bytes = 0;
+    }
+
+  g_free (path);
+
+  data->expected_dirs = 6;
+  data->expected_files = 30;
+
+  g_file_measure_disk_usage_async (file,
+                                   G_FILE_MEASURE_APPARENT_SIZE,
+                                   0, NULL,
+                                   measure_progress, data,
+                                   measure_done, data);
 }
 
 int
@@ -864,6 +1021,7 @@ main (int argc, char *argv[])
   g_test_add_func ("/file/copy-preserve-mode", test_copy_preserve_mode);
 #endif
   g_test_add_func ("/file/measure", test_measure);
+  g_test_add_func ("/file/measure-async", test_measure_async);
 
   return g_test_run ();
 }