[Spice-devel] [spice-gtk v4 23/24] tests: file-transfer cancel on read async

[Victor Toso]

---
 tests/file-transfer.c | 64 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)

diff --git a/tests/file-transfer.c b/tests/file-transfer.c
index 61d0dcc..018d6ad 100644
--- a/tests/file-transfer.c
+++ b/tests/file-transfer.c
@@ -172,6 +172,62 @@ test_cancel_on_task_init(Fixture *f, gconstpointer user_data)
     g_main_loop_run (f->loop);
 }
 
+/*******************************************************************************
+ * TEST CANCEL ON READ ASYNC
+ ******************************************************************************/
+
+static void
+transfer_cancelled_read_async_cb(GObject *source_object,
+                                 GAsyncResult *res,
+                                 gpointer user_data)
+{
+    SpiceFileTransferTask *xfer_task;
+    gssize count;
+    char *buffer;
+    GError *error = NULL;
+
+    xfer_task = SPICE_FILE_TRANSFER_TASK(source_object);
+    count = spice_file_transfer_task_read_finish(xfer_task, res, &buffer, &error);
+    g_assert_error(error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
+    g_assert_cmpint(count, ==, -1);
+
+    transfer_xfer_task_on_finished(NULL, NULL, user_data);
+}
+
+static void
+transfer_on_init_async_cb_cancel(GObject *obj, GAsyncResult *res, gpointer data)
+{
+    GFileInfo *info;
+    SpiceFileTransferTask *xfer_task;
+    GError *error = NULL;
+    GCancellable *cancellable;
+
+    xfer_task = SPICE_FILE_TRANSFER_TASK(obj);
+    info = spice_file_transfer_task_init_task_finish(xfer_task, res, &error);
+    g_assert_no_error(error);
+    g_assert_nonnull(info);
+
+    cancellable = spice_file_transfer_task_get_cancellable(xfer_task);
+    g_cancellable_cancel(cancellable);
+    spice_file_transfer_task_read_async(xfer_task, transfer_cancelled_read_async_cb, data);
+}
+
+static void
+test_cancel_on_read_async(Fixture *f, gconstpointer user_data)
+{
+    GHashTableIter iter;
+    gpointer key, value;
+
+    f->xfer_task_ht = spice_file_transfer_task_create_tasks(f->files, NULL, G_FILE_COPY_NONE, NULL);
+    g_hash_table_iter_init(&iter, f->xfer_task_ht);
+    while (g_hash_table_iter_next(&iter, &key, &value)) {
+        SpiceFileTransferTask *xfer_task = SPICE_FILE_TRANSFER_TASK(value);
+        g_signal_connect(xfer_task, "finished", G_CALLBACK(transfer_xfer_task_on_finished), f);
+        spice_file_transfer_task_init_task_async(xfer_task, transfer_on_init_async_cb_cancel, f);
+    }
+    g_main_loop_run (f->loop);
+}
+
 /* Tests summary:
  *
  * This tests are specific to SpiceFileTransferTask and how it handles the
@@ -214,6 +270,10 @@ int main(int argc, char* argv[])
                Fixture, GUINT_TO_POINTER(SINGLE_FILE),
                f_setup, test_cancel_on_task_init, f_teardown);
 
+    g_test_add("/spice-file-transfer-task/single/cancel/on-read-async",
+               Fixture, GUINT_TO_POINTER(SINGLE_FILE),
+               f_setup, test_cancel_on_read_async, f_teardown);
+
     g_test_add("/spice-file-transfer-task/multiple/simple-transfer",
                Fixture, GUINT_TO_POINTER(MULTIPLE_FILES),
                f_setup, test_simple_transfer, f_teardown);
@@ -222,5 +282,9 @@ int main(int argc, char* argv[])
                Fixture, GUINT_TO_POINTER(MULTIPLE_FILES),
                f_setup, test_cancel_on_task_init, f_teardown);
 
+    g_test_add("/spice-file-transfer-task/multiple/cancel/on-read-async",
+               Fixture, GUINT_TO_POINTER(MULTIPLE_FILES),
+               f_setup, test_cancel_on_read_async, f_teardown);
+
     return g_test_run();
 }
-- 
2.7.4