gst/gdp/gstgdpdepay.c: proxying get/set caps is the wrong thing to do, since we reall...
[platform/upstream/gstreamer.git] / tests / check / elements / gdpdepay.c
1 /* GStreamer
2  *
3  * Copyright (C) 2006 Thomas Vander Stichele <thomas at apestaart dot org>
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Library General Public
7  * License as published by the Free Software Foundation; either
8  * version 2 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  * Library General Public License for more details.
14  *
15  * You should have received a copy of the GNU Library General Public
16  * License along with this library; if not, write to the
17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18  * Boston, MA 02111-1307, USA.
19  */
20
21 #include <string.h>
22 #include <unistd.h>
23
24 #include <gst/check/gstcheck.h>
25 #include <gst/dataprotocol/dataprotocol.h>
26
27 gboolean have_eos = FALSE;
28
29 /* For ease of programming we use globals to keep refs for our floating
30  * src and sink pads we create; otherwise we always have to do get_pad,
31  * get_peer, and then remove references in every test function */
32 GstPad *mysrcpad, *mysinkpad, *myshsinkpad;
33
34 #define AUDIO_CAPS_TEMPLATE_STRING \
35   "audio/x-raw-int, " \
36     "rate = (int) [ 1, MAX ], " \
37     "channels = (int) [ 1, 8 ], " \
38     "endianness = (int) BYTE_ORDER, " \
39     "width = (int) {8, 16}, " \
40     "depth = (int) {8, 16}, " \
41     "signed = (boolean) true"
42
43 #define AUDIO_CAPS_STRING \
44   "audio/x-raw-int, " \
45     "rate = (int) 1000, " \
46     "channels = (int) 2, " \
47     "endianness = (int) BYTE_ORDER, " \
48     "width = (int) 16, " \
49     "depth = (int) 16, " \
50     "signed = (boolean) true"
51
52
53 static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
54     GST_PAD_SINK,
55     GST_PAD_ALWAYS,
56     GST_STATIC_CAPS (AUDIO_CAPS_TEMPLATE_STRING)
57     );
58 static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
59     GST_PAD_SRC,
60     GST_PAD_ALWAYS,
61     GST_STATIC_CAPS ("application/x-gdp")
62     );
63
64 /* takes over reference for outcaps */
65 GstElement *
66 setup_gdpdepay ()
67 {
68   GstElement *gdpdepay;
69
70   GST_DEBUG ("setup_gdpdepay");
71   gdpdepay = gst_check_setup_element ("gdpdepay");
72   mysrcpad = gst_check_setup_src_pad (gdpdepay, &srctemplate, NULL);
73   mysinkpad = gst_check_setup_sink_pad (gdpdepay, &sinktemplate, NULL);
74
75   return gdpdepay;
76 }
77
78 void
79 cleanup_gdpdepay (GstElement * gdpdepay)
80 {
81   GST_DEBUG ("cleanup_gdpdepay");
82
83   gst_check_teardown_src_pad (gdpdepay);
84   gst_check_teardown_sink_pad (gdpdepay);
85   gst_check_teardown_element (gdpdepay);
86 }
87
88 static void
89 gdpdepay_push_per_byte (gchar * reason, guint8 * bytes, guint length)
90 {
91   int i;
92   GstBuffer *inbuffer;
93
94   for (i = 0; i < length; ++i) {
95     inbuffer = gst_buffer_new_and_alloc (1);
96     GST_BUFFER_DATA (inbuffer)[0] = bytes[i];
97     fail_unless (gst_pad_push (mysrcpad, inbuffer) == GST_FLOW_OK,
98         "%s: failed pushing byte buffer", reason);
99   }
100 }
101
102 GST_START_TEST (test_audio_per_byte)
103 {
104   GstCaps *caps;
105   GstPad *srcpad;
106   GstElement *gdpdepay;
107   GstBuffer *buffer, *inbuffer, *outbuffer;
108   gint length;
109   guint8 *header, *payload;
110   guint len;
111   guint i;
112
113   gdpdepay = setup_gdpdepay ();
114   srcpad = gst_element_get_pad (gdpdepay, "src");
115
116   fail_unless (gst_element_set_state (gdpdepay,
117           GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
118       "could not set to playing");
119
120   caps = gst_pad_get_caps (srcpad);
121   fail_unless (gst_caps_is_any (caps));
122   gst_caps_unref (caps);
123   fail_if (gst_pad_get_negotiated_caps (srcpad));
124
125   /* create caps and buffer packets and push them */
126   caps = gst_caps_from_string (AUDIO_CAPS_STRING);
127   fail_unless (gst_dp_packet_from_caps (caps, 0, &len, &header, &payload));
128   gst_caps_unref (caps);
129   gdpdepay_push_per_byte ("caps header", header, len);
130   fail_unless_equals_int (g_list_length (buffers), 0);
131   gdpdepay_push_per_byte ("caps payload", payload,
132       gst_dp_header_payload_length (header));
133   fail_unless_equals_int (g_list_length (buffers), 0);
134   caps = gst_pad_get_caps (srcpad);
135   fail_if (gst_caps_is_any (caps));
136   gst_caps_unref (caps);
137
138   g_free (header);
139   g_free (payload);
140
141   buffer = gst_buffer_new_and_alloc (4);
142   memcpy (GST_BUFFER_DATA (buffer), "f00d", 4);
143   GST_BUFFER_TIMESTAMP (buffer) = GST_SECOND;
144   GST_BUFFER_DURATION (buffer) = GST_SECOND / 10;
145   fail_unless (gst_dp_header_from_buffer (buffer, 0, &len, &header));
146   gdpdepay_push_per_byte ("buffer header", header, len);
147   fail_unless_equals_int (g_list_length (buffers), 0);
148   gdpdepay_push_per_byte ("buffer payload", GST_BUFFER_DATA (buffer),
149       gst_dp_header_payload_length (header));
150   g_free (header);
151   gst_buffer_unref (buffer);
152
153   fail_unless_equals_int (g_list_length (buffers), 1);
154   fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);
155   fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (outbuffer), GST_SECOND);
156   fail_unless_equals_uint64 (GST_BUFFER_DURATION (outbuffer), GST_SECOND / 10);
157
158   buffers = g_list_remove (buffers, outbuffer);
159   gst_buffer_unref (outbuffer);
160
161   fail_unless (gst_element_set_state (gdpdepay,
162           GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
163
164   ASSERT_OBJECT_REFCOUNT (gdpdepay, "gdpdepay", 1);
165   g_list_foreach (buffers, (GFunc) gst_mini_object_unref, NULL);
166   g_list_free (buffers);
167   buffers = NULL;
168   gst_object_unref (srcpad);
169   cleanup_gdpdepay (gdpdepay);
170 }
171
172 GST_END_TEST;
173
174 GST_START_TEST (test_audio_in_one_buffer)
175 {
176   GstCaps *caps;
177   GstPad *srcpad;
178   GstElement *gdpdepay;
179   GstBuffer *buffer, *inbuffer, *outbuffer;
180   gint length;
181   guint8 *caps_header, *caps_payload, *buf_header, *buf_payload;
182   guint header_len, payload_len;
183   guint i;
184
185   gdpdepay = setup_gdpdepay ();
186   srcpad = gst_element_get_pad (gdpdepay, "src");
187
188   fail_unless (gst_element_set_state (gdpdepay,
189           GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
190       "could not set to playing");
191
192   /* make sure no caps are set yet */
193   caps = gst_pad_get_caps (srcpad);
194   fail_unless (gst_caps_is_any (caps));
195   gst_caps_unref (caps);
196   fail_if (gst_pad_get_negotiated_caps (srcpad));
197
198   /* create caps and buffer packets and push them as one buffer */
199   caps = gst_caps_from_string (AUDIO_CAPS_STRING);
200   fail_unless (gst_dp_packet_from_caps (caps, 0, &header_len, &caps_header,
201           &caps_payload));
202
203   buffer = gst_buffer_new_and_alloc (4);
204   memcpy (GST_BUFFER_DATA (buffer), "f00d", 4);
205   fail_unless (gst_dp_header_from_buffer (buffer, 0, &header_len, &buf_header));
206
207   payload_len = gst_dp_header_payload_length (caps_header);
208
209   inbuffer = gst_buffer_new_and_alloc (2 * GST_DP_HEADER_LENGTH +
210       payload_len + GST_BUFFER_SIZE (buffer));
211   memcpy (GST_BUFFER_DATA (inbuffer), caps_header, GST_DP_HEADER_LENGTH);
212   i = GST_DP_HEADER_LENGTH;
213   memcpy (GST_BUFFER_DATA (inbuffer) + i, caps_payload, payload_len);
214   i += payload_len;
215   memcpy (GST_BUFFER_DATA (inbuffer) + i, buf_header, GST_DP_HEADER_LENGTH);
216   i += GST_DP_HEADER_LENGTH;
217   memcpy (GST_BUFFER_DATA (inbuffer) + i, GST_BUFFER_DATA (buffer),
218       GST_BUFFER_SIZE (buffer));
219
220   gst_caps_unref (caps);
221   gst_buffer_unref (buffer);
222
223   g_free (caps_header);
224   g_free (caps_payload);
225   g_free (buf_header);
226
227   /* now push it */
228   gst_pad_push (mysrcpad, inbuffer);
229
230   /* the buffer is still queued */
231   fail_unless_equals_int (g_list_length (buffers), 1);
232
233   fail_unless (gst_element_set_state (gdpdepay,
234           GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
235
236   gst_object_unref (srcpad);
237   g_list_foreach (buffers, (GFunc) gst_mini_object_unref, NULL);
238   g_list_free (buffers);
239   buffers = NULL;
240   ASSERT_OBJECT_REFCOUNT (gdpdepay, "gdpdepay", 1);
241   cleanup_gdpdepay (gdpdepay);
242 }
243
244 GST_END_TEST;
245
246 static GstStaticPadTemplate shsinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
247     GST_PAD_SINK,
248     GST_PAD_ALWAYS,
249     GST_STATIC_CAPS ("application/x-gst-test-streamheader")
250     );
251
252 GstElement *
253 setup_gdpdepay_streamheader ()
254 {
255   GstElement *gdpdepay;
256
257   GST_DEBUG ("setup_gdpdepay");
258   gdpdepay = gst_check_setup_element ("gdpdepay");
259   mysrcpad = gst_check_setup_src_pad (gdpdepay, &srctemplate, NULL);
260   myshsinkpad = gst_check_setup_sink_pad (gdpdepay, &shsinktemplate, NULL);
261
262   return gdpdepay;
263 }
264
265 /* this tests deserialization of a GDP stream where the serialized caps
266  * have a streamheader set */
267 GST_START_TEST (test_streamheader)
268 {
269   GstCaps *caps, *padcaps;
270   GstPad *srcpad;
271   GstElement *gdpdepay;
272   GstBuffer *buffer, *inbuffer, *outbuffer, *shbuffer;
273   gint length;
274   guint8 *caps_header, *caps_payload, *buf_header, *buf_payload;
275   guint header_len, payload_len;
276   guint i;
277   GstStructure *structure;
278   GValue array = { 0 };
279   GValue value = { 0 };
280
281   gdpdepay = setup_gdpdepay_streamheader ();
282   srcpad = gst_element_get_pad (gdpdepay, "src");
283   ASSERT_OBJECT_REFCOUNT (gdpdepay, "gdpdepay", 1);
284
285   fail_unless (gst_element_set_state (gdpdepay,
286           GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
287       "could not set to playing");
288
289   /* make sure no caps are set yet */
290   caps = gst_pad_get_caps (srcpad);
291   fail_unless (gst_caps_is_any (caps));
292   gst_caps_unref (caps);
293   fail_if (gst_pad_get_negotiated_caps (srcpad));
294
295   /* create a streamheader buffer and the caps containing it */
296   caps = gst_caps_from_string ("application/x-gst-test-streamheader");
297   structure = gst_caps_get_structure (caps, 0);
298   buffer = gst_buffer_new_and_alloc (4);
299   memcpy (GST_BUFFER_DATA (buffer), "f00d", 4);
300   GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_IN_CAPS);
301   g_value_init (&array, GST_TYPE_ARRAY);
302   g_value_init (&value, GST_TYPE_BUFFER);
303   shbuffer = gst_buffer_copy (buffer);
304   gst_value_set_buffer (&value, shbuffer);
305   gst_buffer_unref (shbuffer);
306   gst_value_array_append_value (&array, &value);
307   g_value_unset (&value);
308   gst_structure_set_value (structure, "streamheader", &array);
309   g_value_unset (&array);
310
311   gst_buffer_set_caps (buffer, caps);
312
313   /* create GDP packets for the caps and the buffer, and put them in one
314    * GDP buffer */
315   fail_unless (gst_dp_packet_from_caps (caps, 0, &header_len, &caps_header,
316           &caps_payload));
317
318   fail_unless (gst_dp_header_from_buffer (buffer, 0, &header_len, &buf_header));
319
320   payload_len = gst_dp_header_payload_length (caps_header);
321
322   inbuffer = gst_buffer_new_and_alloc (2 * GST_DP_HEADER_LENGTH +
323       payload_len + GST_BUFFER_SIZE (buffer));
324   memcpy (GST_BUFFER_DATA (inbuffer), caps_header, GST_DP_HEADER_LENGTH);
325   i = GST_DP_HEADER_LENGTH;
326   memcpy (GST_BUFFER_DATA (inbuffer) + i, caps_payload, payload_len);
327   i += payload_len;
328   memcpy (GST_BUFFER_DATA (inbuffer) + i, buf_header, GST_DP_HEADER_LENGTH);
329   i += GST_DP_HEADER_LENGTH;
330   memcpy (GST_BUFFER_DATA (inbuffer) + i, GST_BUFFER_DATA (buffer),
331       GST_BUFFER_SIZE (buffer));
332
333   gst_caps_unref (caps);
334   gst_buffer_unref (buffer);
335
336   g_free (caps_header);
337   g_free (caps_payload);
338   g_free (buf_header);
339
340   /* now push it */
341   ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
342   gst_pad_push (mysrcpad, inbuffer);
343
344   /* our only output buffer is the streamheader buffer */
345   fail_unless_equals_int (g_list_length (buffers), 1);
346   fail_if ((outbuffer = (GstBuffer *) buffers->data) == NULL);
347   buffers = g_list_remove (buffers, outbuffer);
348   ASSERT_BUFFER_REFCOUNT (outbuffer, "outbuffer", 1);
349   fail_unless (GST_BUFFER_FLAG_IS_SET (outbuffer, GST_BUFFER_FLAG_IN_CAPS));
350
351   padcaps = gst_pad_get_negotiated_caps (myshsinkpad);
352   caps = gst_buffer_get_caps (outbuffer);
353   fail_if (caps == NULL);
354   fail_if (padcaps == NULL);
355   GST_DEBUG ("caps: %" GST_PTR_FORMAT ", padcaps: %" GST_PTR_FORMAT, caps,
356       padcaps);
357   fail_unless (gst_caps_is_equal (padcaps, caps));
358
359   /* FIXME: get streamheader, compare data with buffer */
360   gst_buffer_unref (outbuffer);
361   gst_caps_unref (padcaps);
362   gst_caps_unref (caps);
363
364   /* clean up */
365   fail_unless (gst_element_set_state (gdpdepay,
366           GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null");
367
368   gst_object_unref (srcpad);
369   g_list_foreach (buffers, (GFunc) gst_mini_object_unref, NULL);
370   g_list_free (buffers);
371   buffers = NULL;
372   ASSERT_OBJECT_REFCOUNT (gdpdepay, "gdpdepay", 1);
373   cleanup_gdpdepay (gdpdepay);
374 }
375
376 GST_END_TEST;
377
378
379 Suite *
380 gdpdepay_suite (void)
381 {
382   Suite *s = suite_create ("gdpdepay");
383   TCase *tc_chain = tcase_create ("general");
384
385   suite_add_tcase (s, tc_chain);
386   tcase_add_test (tc_chain, test_audio_per_byte);
387   tcase_add_test (tc_chain, test_audio_in_one_buffer);
388   tcase_add_test (tc_chain, test_streamheader);
389
390   return s;
391 }
392
393 int
394 main (int argc, char **argv)
395 {
396   int nf;
397
398   Suite *s = gdpdepay_suite ();
399   SRunner *sr = srunner_create (s);
400
401   gst_check_init (&argc, &argv);
402
403   srunner_run_all (sr, CK_NORMAL);
404   nf = srunner_ntests_failed (sr);
405   srunner_free (sr);
406
407   return nf;
408 }