Merge remote branch 'gvdb/master'
[platform/upstream/glib.git] / gio / tests / io-stream.c
1 /* GLib testing framework examples and tests
2  * Copyright (C) 2010 Collabora Ltd.
3  * Authors: Xavier Claessens <xclaesse@gmail.com>
4  *
5  * This work is provided "as is"; redistribution and modification
6  * in whole or in part, in any medium, physical or electronic is
7  * permitted without restriction.
8  *
9  * This work is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
12  *
13  * In no event shall the authors or contributors be liable for any
14  * direct, indirect, incidental, special, exemplary, or consequential
15  * damages (including, but not limited to, procurement of substitute
16  * goods or services; loss of use, data, or profits; or business
17  * interruption) however caused and on any theory of liability, whether
18  * in contract, strict liability, or tort (including negligence or
19  * otherwise) arising in any way out of the use of this software, even
20  * if advised of the possibility of such damage.
21  */
22
23 #include <glib/glib.h>
24 #include <gio/gio.h>
25 #include <stdlib.h>
26 #include <string.h>
27
28 typedef struct
29 {
30   GIOStream parent;
31   GInputStream *input_stream;
32   GOutputStream *output_stream;
33 } GTestIOStream;
34
35 typedef struct
36 {
37   GIOStreamClass parent_class;
38 } GTestIOStreamClass;
39
40 G_DEFINE_TYPE (GTestIOStream, g_test_io_stream, G_TYPE_IO_STREAM);
41
42
43 static GInputStream *
44 get_input_stream (GIOStream *io_stream)
45 {
46   GTestIOStream *self =  (GTestIOStream *) io_stream;
47
48   return self->input_stream;
49 }
50
51 static GOutputStream *
52 get_output_stream (GIOStream *io_stream)
53 {
54   GTestIOStream *self =  (GTestIOStream *) io_stream;
55
56   return self->output_stream;
57 }
58
59 static void
60 finalize (GObject *object)
61 {
62   GTestIOStream *self = (GTestIOStream *) object;
63
64   if (self->input_stream != NULL)
65     g_object_unref (self->input_stream);
66
67   if (self->output_stream != NULL)
68     g_object_unref (self->output_stream);
69
70   G_OBJECT_CLASS (g_test_io_stream_parent_class)->finalize (object);
71 }
72
73 static void
74 g_test_io_stream_class_init (GTestIOStreamClass *klass)
75 {
76   GObjectClass *object_class = G_OBJECT_CLASS (klass);
77   GIOStreamClass *io_class = G_IO_STREAM_CLASS (klass);
78
79   object_class->finalize = finalize;
80
81   io_class->get_input_stream = get_input_stream;
82   io_class->get_output_stream = get_output_stream;
83 }
84
85 static void
86 g_test_io_stream_init (GTestIOStream *self)
87 {
88 }
89
90 static GIOStream *
91 g_test_io_stream_new (GInputStream *input, GOutputStream *output)
92 {
93   GTestIOStream *self;
94
95   self = g_object_new (g_test_io_stream_get_type (), NULL);
96   self->input_stream = g_object_ref (input);
97   self->output_stream = g_object_ref (output);
98
99   return G_IO_STREAM (self);
100 }
101
102 typedef struct
103 {
104   GMainLoop *main_loop;
105   const gchar *data1;
106   const gchar *data2;
107   GIOStream *iostream1;
108   GIOStream *iostream2;
109 } TestCopyChunksData;
110
111 static void
112 test_copy_chunks_splice_cb (GObject *source_object,
113     GAsyncResult *res,
114     gpointer user_data)
115 {
116   TestCopyChunksData *data = user_data;
117   GMemoryOutputStream *ostream;
118   gchar *received_data;
119   GError *error = NULL;
120
121   g_io_stream_splice_finish (res, &error);
122   g_assert_no_error (error);
123
124   ostream = G_MEMORY_OUTPUT_STREAM (((GTestIOStream *) data->iostream1)->output_stream);
125   received_data = g_memory_output_stream_get_data (ostream);
126   g_assert_cmpstr (received_data, ==, data->data2);
127
128   ostream = G_MEMORY_OUTPUT_STREAM (((GTestIOStream *) data->iostream2)->output_stream);
129   received_data = g_memory_output_stream_get_data (ostream);
130   g_assert_cmpstr (received_data, ==, data->data1);
131
132   g_assert (g_io_stream_is_closed (data->iostream1));
133   g_assert (g_io_stream_is_closed (data->iostream2));
134
135   g_main_loop_quit (data->main_loop);
136 }
137
138 static void
139 test_copy_chunks (void)
140 {
141   TestCopyChunksData data;
142   GInputStream *istream;
143   GOutputStream *ostream;
144
145   data.main_loop = g_main_loop_new (NULL, FALSE);
146   data.data1 = "abcdefghijklmnopqrstuvwxyz";
147   data.data2 = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
148
149   istream = g_memory_input_stream_new_from_data (data.data1, -1, NULL);
150   ostream = g_memory_output_stream_new (NULL, 0, g_realloc, g_free);
151   data.iostream1 = g_test_io_stream_new (istream, ostream);
152   g_object_unref (istream);
153   g_object_unref (ostream);
154
155   istream = g_memory_input_stream_new_from_data (data.data2, -1, NULL);
156   ostream = g_memory_output_stream_new (NULL, 0, g_realloc, g_free);
157   data.iostream2 = g_test_io_stream_new (istream, ostream);
158   g_object_unref (istream);
159   g_object_unref (ostream);
160
161   g_io_stream_splice_async (data.iostream1, data.iostream2,
162       G_IO_STREAM_SPLICE_CLOSE_STREAM1 | G_IO_STREAM_SPLICE_CLOSE_STREAM2 |
163       G_IO_STREAM_SPLICE_WAIT_FOR_BOTH, G_PRIORITY_DEFAULT,
164       NULL, test_copy_chunks_splice_cb, &data);
165
166   /* We do not hold a ref in data struct, this is to make sure the operation
167    * keeps the iostream objects alive until it finishes */
168   g_object_unref (data.iostream1);
169   g_object_unref (data.iostream2);
170
171   g_main_loop_run (data.main_loop);
172   g_main_loop_unref (data.main_loop);
173 }
174
175 int
176 main (int   argc,
177       char *argv[])
178 {
179   g_type_init ();
180   g_test_init (&argc, &argv, NULL);
181
182   g_test_add_func ("/io-stream/copy-chunks", test_copy_chunks);
183
184   return g_test_run();
185 }