gio/tests/gsocketclient-slow.c

   1 /* GIO - GLib Input, Output and Streaming Library
   2  *
   3  * Copyright (C) 2018 Igalia S.L.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General
  16  * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #include <gio/gio.h>
  20
  21 static void
  22 on_connected (GObject      *source_object,
  23               GAsyncResult *result,
  24               gpointer      user_data)
  25 {
  26   GSocketConnection *conn;
  27   GError *error = NULL;
  28
  29   conn = g_socket_client_connect_to_uri_finish (G_SOCKET_CLIENT (source_object), result, &error);
  30   g_assert_no_error (error);
  31
  32   g_object_unref (conn);
  33   g_main_loop_quit (user_data);
  34 }
  35
  36 static void
  37 test_happy_eyeballs (void)
  38 {
  39   GSocketClient *client;
  40   GSocketService *service;
  41   GError *error = NULL;
  42   guint16 port;
  43   GMainLoop *loop;
  44
  45   loop = g_main_loop_new (NULL, FALSE);
  46
  47   service = g_socket_service_new ();
  48   port = g_socket_listener_add_any_inet_port (G_SOCKET_LISTENER (service), NULL, &error);
  49   g_assert_no_error (error);
  50   g_socket_service_start (service);
  51
  52   /* All of the magic here actually happens in slow-connect-preload.c
  53    * which as you would guess is preloaded. So this is just making a
  54    * normal connection that happens to take 600ms each time. This will
  55    * trigger the logic to make multiple parallel connections.
  56    */
  57   client = g_socket_client_new ();
  58   g_socket_client_connect_to_host_async (client, "localhost", port, NULL, on_connected, loop);
  59   g_main_loop_run (loop);
  60
  61   g_main_loop_unref (loop);
  62   g_object_unref (service);
  63   g_object_unref (client);
  64 }
  65
  66 static void
  67 on_connected_cancelled (GObject      *source_object,
  68                         GAsyncResult *result,
  69                         gpointer      user_data)
  70 {
  71   GSocketConnection *conn;
  72   GError *error = NULL;
  73
  74   conn = g_socket_client_connect_to_uri_finish (G_SOCKET_CLIENT (source_object), result, &error);
  75   g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
  76   g_assert_null (conn);
  77
  78   g_error_free (error);
  79   g_main_loop_quit (user_data);
  80 }
  81
  82 static int
  83 on_timer (GCancellable *cancel)
  84 {
  85   g_cancellable_cancel (cancel);
  86   return G_SOURCE_REMOVE;
  87 }
  88
  89 static void
  90 on_event (GSocketClient      *client,
  91           GSocketClientEvent  event,
  92           GSocketConnectable *connectable,
  93           GIOStream          *connection,
  94           gboolean           *got_completed_event)
  95 {
  96   if (event == G_SOCKET_CLIENT_COMPLETE)
  97     {
  98       *got_completed_event = TRUE;
  99       g_assert_null (connection);
 100     }
 101 }
 102
 103 static void
 104 test_happy_eyeballs_cancel_delayed (void)
 105 {
 106   GSocketClient *client;
 107   GSocketService *service;
 108   GError *error = NULL;
 109   guint16 port;
 110   GMainLoop *loop;
 111   GCancellable *cancel;
 112   gboolean got_completed_event = FALSE;
 113
 114   /* This just tests that cancellation works as expected, still emits the completed signal,
 115    * and never returns a connection */
 116
 117   loop = g_main_loop_new (NULL, FALSE);
 118
 119   service = g_socket_service_new ();
 120   port = g_socket_listener_add_any_inet_port (G_SOCKET_LISTENER (service), NULL, &error);
 121   g_assert_no_error (error);
 122   g_socket_service_start (service);
 123
 124   client = g_socket_client_new ();
 125   cancel = g_cancellable_new ();
 126   g_socket_client_connect_to_host_async (client, "localhost", port, cancel, on_connected_cancelled, loop);
 127   g_timeout_add (1, (GSourceFunc) on_timer, cancel);
 128   g_signal_connect (client, "event", G_CALLBACK (on_event), &got_completed_event);
 129   g_main_loop_run (loop);
 130
 131   g_assert_true (got_completed_event);
 132   g_main_loop_unref (loop);
 133   g_object_unref (service);
 134   g_object_unref (client);
 135   g_object_unref (cancel);
 136 }
 137
 138 static void
 139 test_happy_eyeballs_cancel_instant (void)
 140 {
 141   GSocketClient *client;
 142   GSocketService *service;
 143   GError *error = NULL;
 144   guint16 port;
 145   GMainLoop *loop;
 146   GCancellable *cancel;
 147   gboolean got_completed_event = FALSE;
 148
 149   /* This tests the same things as above, test_happy_eyeballs_cancel_delayed(), but
 150    * with different timing since it sends an already cancelled cancellable */
 151
 152   loop = g_main_loop_new (NULL, FALSE);
 153
 154   service = g_socket_service_new ();
 155   port = g_socket_listener_add_any_inet_port (G_SOCKET_LISTENER (service), NULL, &error);
 156   g_assert_no_error (error);
 157   g_socket_service_start (service);
 158
 159   client = g_socket_client_new ();
 160   cancel = g_cancellable_new ();
 161   g_cancellable_cancel (cancel);
 162   g_socket_client_connect_to_host_async (client, "localhost", port, cancel, on_connected_cancelled, loop);
 163   g_signal_connect (client, "event", G_CALLBACK (on_event), &got_completed_event);
 164   g_main_loop_run (loop);
 165
 166   g_assert_true (got_completed_event);
 167   g_main_loop_unref (loop);
 168   g_object_unref (service);
 169   g_object_unref (client);
 170   g_object_unref (cancel);
 171 }
 172
 173 int
 174 main (int argc, char *argv[])
 175 {
 176   g_test_init (&argc, &argv, NULL);
 177
 178   g_test_add_func ("/socket-client/happy-eyeballs/slow", test_happy_eyeballs);
 179   g_test_add_func ("/socket-client/happy-eyeballs/cancellation/instant", test_happy_eyeballs_cancel_instant);
 180   g_test_add_func ("/socket-client/happy-eyeballs/cancellation/delayed", test_happy_eyeballs_cancel_delayed);
 181
 182
 183   return g_test_run ();
 184 }