+typedef struct {
+ GSubprocess *proc;
+ GCancellable *cancellable;
+ gboolean is_utf8;
+ gboolean running;
+ GError *error;
+} TestCancelledCommunicateData;
+
+static gboolean
+on_test_communicate_cancelled_idle (gpointer user_data)
+{
+ TestCancelledCommunicateData *data = user_data;
+ GBytes *input;
+ const gchar *hellostring;
+ GBytes *stdout_bytes = NULL, *stderr_bytes = NULL;
+ gchar *stdout_buf = NULL, *stderr_buf = NULL;
+
+ /* Include a leading hash and trailing newline so that if this gets onto the
+ * test’s stdout, it doesn’t mess up TAP output. */
+ hellostring = "# hello world\n";
+ input = g_bytes_new_static (hellostring, strlen (hellostring));
+
+ if (data->is_utf8)
+ g_subprocess_communicate_utf8 (data->proc, hellostring, data->cancellable,
+ &stdout_buf, &stderr_buf, &data->error);
+ else
+ g_subprocess_communicate (data->proc, input, data->cancellable, &stdout_bytes,
+ &stderr_bytes, &data->error);
+
+ data->running = FALSE;
+
+ if (data->is_utf8)
+ {
+ g_assert_null (stdout_buf);
+ g_assert_null (stderr_buf);
+ }
+ else
+ {
+ g_assert_null (stdout_bytes);
+ g_assert_null (stderr_bytes);
+ }
+
+ g_bytes_unref (input);
+
+ return G_SOURCE_REMOVE;
+}
+
+/* Test g_subprocess_communicate() can be cancelled correctly */
+static void
+test_communicate_cancelled (gconstpointer test_data)
+{
+ GSubprocessFlags flags = GPOINTER_TO_INT (test_data);
+ GPtrArray *args;
+ GSubprocess *proc;
+ GCancellable *cancellable = NULL;
+ GError *error = NULL;
+ TestCancelledCommunicateData data = { 0 };
+
+ args = get_test_subprocess_args ("cat", NULL);
+ proc = g_subprocess_newv ((const gchar* const*)args->pdata,
+ G_SUBPROCESS_FLAGS_STDIN_PIPE | flags,
+ &error);
+ g_assert_no_error (error);
+ g_ptr_array_free (args, TRUE);
+
+ cancellable = g_cancellable_new ();
+
+ data.proc = proc;
+ data.cancellable = cancellable;
+ data.error = error;
+
+ g_cancellable_cancel (cancellable);
+ g_idle_add (on_test_communicate_cancelled_idle, &data);
+
+ data.running = TRUE;
+ while (data.running)
+ g_main_context_iteration (NULL, TRUE);
+
+ g_assert_error (data.error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
+ g_clear_error (&data.error);
+
+ g_object_unref (cancellable);
+ g_object_unref (proc);
+}
+
+static void
+on_communicate_cancelled_complete (GObject *proc,
+ GAsyncResult *result,
+ gpointer user_data)
+{
+ TestAsyncCommunicateData *data = user_data;
+ GBytes *stdout_bytes = NULL, *stderr_bytes = NULL;
+ char *stdout_str = NULL, *stderr_str = NULL;
+
+ data->running = FALSE;
+ if (data->is_utf8)
+ (void) g_subprocess_communicate_utf8_finish ((GSubprocess*)proc, result,
+ &stdout_str, &stderr_str, &data->error);
+ else
+ (void) g_subprocess_communicate_finish ((GSubprocess*)proc, result,
+ &stdout_bytes, &stderr_bytes, &data->error);
+
+ if (data->is_utf8)
+ {
+ g_assert_null (stdout_str);
+ g_assert_null (stderr_str);
+ }
+ else
+ {
+ g_assert_null (stdout_bytes);
+ g_assert_null (stderr_bytes);
+ }
+}
+
+/* Test g_subprocess_communicate_async() can be cancelled correctly,
+ * as passed in via @test_data. */
+static void
+test_communicate_cancelled_async (gconstpointer test_data)
+{
+ GSubprocessFlags flags = GPOINTER_TO_INT (test_data);
+ GError *error = NULL;
+ GPtrArray *args;
+ TestAsyncCommunicateData data = { 0 };
+ GSubprocess *proc;
+ GCancellable *cancellable = NULL;
+ GBytes *input;
+ const char *hellostring;
+
+ args = get_test_subprocess_args ("cat", NULL);
+ proc = g_subprocess_newv ((const gchar* const*)args->pdata,
+ G_SUBPROCESS_FLAGS_STDIN_PIPE | flags,
+ &error);
+ g_assert_no_error (error);
+ g_ptr_array_free (args, TRUE);
+
+ /* Include a leading hash and trailing newline so that if this gets onto the
+ * test’s stdout, it doesn’t mess up TAP output. */
+ hellostring = "# hello world\n";
+ input = g_bytes_new_static (hellostring, strlen (hellostring));
+
+ cancellable = g_cancellable_new ();
+
+ g_subprocess_communicate_async (proc, input,
+ cancellable,
+ on_communicate_cancelled_complete,
+ &data);
+
+ g_cancellable_cancel (cancellable);
+
+ data.running = TRUE;
+ while (data.running)
+ g_main_context_iteration (NULL, TRUE);
+
+ g_assert_error (data.error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
+ g_clear_error (&data.error);
+
+ g_bytes_unref (input);
+ g_object_unref (cancellable);
+ g_object_unref (proc);
+}
+
projects / platform / upstream / glib.git / commitdiff