summaryrefslogtreecommitdiff
path: root/tests/check/elements/kate.c
diff options
context:
space:
mode:
Diffstat (limited to 'tests/check/elements/kate.c')
-rw-r--r--tests/check/elements/kate.c839
1 files changed, 839 insertions, 0 deletions
diff --git a/tests/check/elements/kate.c b/tests/check/elements/kate.c
new file mode 100644
index 000000000..f7ac3ed4a
--- /dev/null
+++ b/tests/check/elements/kate.c
@@ -0,0 +1,839 @@
+/* GStreamer
+ *
+ * unit test for kate
+ *
+ * Copyright (C) <2007> Stefan Kost <ensonic@users.sf.net>
+ * Copyright (C) <2008> ogg.k.ogg.k <ogg.k.ogg.k@googlemail.com>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Library General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Library General Public License for more details.
+ *
+ * You should have received a copy of the GNU Library General Public
+ * License along with this library; if not, write to the
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#include <unistd.h>
+
+#include <gst/check/gstcheck.h>
+#include <gst/base/gsttypefindhelper.h>
+
+
+static const guint8 kate_header_0x80[64] = {
+ 0x80, 0x6b, 0x61, 0x74, 0x65, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x09, 0x00, 0x00, 0x00, 0x20, /* .kate...... ... */
+ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xe8, 0x03, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, /* ................ */
+ 0x65, 0x6e, 0x5f, 0x47, 0x42, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* en_GB........... */
+ 0x6e, 0x6f, 0x6e, 0x65, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* none............ */
+};
+
+static const guint8 kate_header_0x81[53] = {
+ 0x81, 0x6b, 0x61, 0x74, 0x65, 0x00, 0x00, 0x00, 0x00, 0x15, 0x00, 0x00, 0x00, 0x6c, 0x69, 0x62, /* .kate........lib */
+ 0x6b, 0x61, 0x74, 0x65, 0x20, 0x30, 0x2e, 0x31, 0x2e, 0x30, 0x20, 0x28, 0x54, 0x69, 0x67, 0x65, /* kate 0.1.0 (Tige */
+ 0x72, 0x29, 0x01, 0x00, 0x00, 0x00, 0x0b, 0x00, 0x00, 0x00, 0x54, 0x49, 0x54, 0x4c, 0x45, 0x3d, /* r)........TITLE= */
+ 0x54, 0x69, 0x67, 0x65, 0x72, /* Tiger */
+};
+static const guint8 kate_header_0x8x[10] = {
+ 0x80, 0x6b, 0x61, 0x74, 0x65, 0x00, 0x00, 0x00, 0x00
+};
+static const guint8 kate_header_0x88[11] = {
+ 0x88, 0x6b, 0x61, 0x74, 0x65, 0x00, 0x00, 0x00, 0x00, 0x00
+};
+static const guint8 kate_header_0x00[45] = {
+ 0x00, 0xe8, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xd0, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, /* ................ */
+ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x70, 0x6c, 0x61, /* .............pla */
+ 0x69, 0x6e, 0x20, 0x6f, 0x6c, 0x64, 0x20, 0x74, 0x65, 0x78, 0x74, 0x08, 0x00 /* in old text.. */
+};
+static const guint8 kate_header_0x01[1] = {
+ 0x01
+};
+static const guint8 kate_header_0x7f[1] = {
+ 0x7f
+};
+
+static const unsigned char kate_spu[] = {
+ 0x00, 0x1b, /* size */
+ 0x00, 0x06, /* commands at offset 6 */
+ 0x45, /* first line data - 2 pixels of colors 0 and 1 */
+ 0x76, /* first line data - 2 pixels of colors 3 and 2 */
+ 0x00, 0x00, /* timestamp */
+ 0x00, 0x06, /* link to next command sequence - points back to the current one to mark no more */
+ 0x06, 0x00, 0x04, 0x00, 0x05, /* pointers to data */
+ 0x05, 0x00, 0x30, 0x04, 0x00, 0x10, 0x02, /* area: 3x1 -> 4x2 */
+ 0x04, 0x0f, 0xff, /* alpha: color 0 transparent, all others opaque */
+ 0x01, /* show */
+ 0xff /* end */
+};
+
+/* A lot of these taken from the vorbisdec test */
+
+/* For ease of programming we use globals to keep refs for our floating
+ * src and sink pads we create; otherwise we always have to do get_pad,
+ * get_peer, and then remove references in every test function */
+static GstPad *mydecsrcpad, *mydecsinkpad;
+static GstPad *myencsrcpad, *myencsinkpad;
+static GstPad *myparsesrcpad, *myparsesinkpad;
+static GstPad *mytagsrcpad, *mytagsinkpad;
+
+static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",
+ GST_PAD_SINK,
+ GST_PAD_ALWAYS,
+ GST_STATIC_CAPS_ANY);
+static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
+ GST_PAD_SRC,
+ GST_PAD_ALWAYS,
+ GST_STATIC_CAPS_ANY);
+
+static GstElement *
+setup_katedec (void)
+{
+ GstElement *katedec;
+
+ GST_DEBUG ("setup_katedec");
+ katedec = gst_check_setup_element ("katedec");
+ mydecsrcpad = gst_check_setup_src_pad (katedec, &srctemplate, NULL);
+ mydecsinkpad = gst_check_setup_sink_pad (katedec, &sinktemplate, NULL);
+ gst_pad_set_active (mydecsrcpad, TRUE);
+ gst_pad_set_active (mydecsinkpad, TRUE);
+
+ return katedec;
+}
+
+static void
+cleanup_katedec (GstElement * katedec)
+{
+ GST_DEBUG ("cleanup_katedec");
+ gst_element_set_state (katedec, GST_STATE_NULL);
+
+ gst_pad_set_active (mydecsrcpad, FALSE);
+ gst_pad_set_active (mydecsinkpad, FALSE);
+ gst_check_teardown_src_pad (katedec);
+ gst_check_teardown_sink_pad (katedec);
+ gst_check_teardown_element (katedec);
+}
+
+static GstElement *
+setup_kateenc (void)
+{
+ GstElement *kateenc;
+
+ GST_DEBUG ("setup_kateenc");
+ kateenc = gst_check_setup_element ("kateenc");
+ myencsrcpad = gst_check_setup_src_pad (kateenc, &srctemplate, NULL);
+ myencsinkpad = gst_check_setup_sink_pad (kateenc, &sinktemplate, NULL);
+ gst_pad_set_active (myencsrcpad, TRUE);
+ gst_pad_set_active (myencsinkpad, TRUE);
+
+ return kateenc;
+}
+
+static void
+cleanup_kateenc (GstElement * kateenc)
+{
+ GST_DEBUG ("cleanup_kateenc");
+ gst_element_set_state (kateenc, GST_STATE_NULL);
+
+ gst_pad_set_active (myencsrcpad, FALSE);
+ gst_pad_set_active (myencsinkpad, FALSE);
+ gst_check_teardown_src_pad (kateenc);
+ gst_check_teardown_sink_pad (kateenc);
+ gst_check_teardown_element (kateenc);
+}
+
+static GstElement *
+setup_kateparse (void)
+{
+ GstElement *kateparse;
+
+ GST_DEBUG ("setup_kateparse");
+ kateparse = gst_check_setup_element ("kateparse");
+ myparsesrcpad = gst_check_setup_src_pad (kateparse, &srctemplate, NULL);
+ myparsesinkpad = gst_check_setup_sink_pad (kateparse, &sinktemplate, NULL);
+ gst_pad_set_active (myparsesrcpad, TRUE);
+ gst_pad_set_active (myparsesinkpad, TRUE);
+
+ return kateparse;
+}
+
+static void
+cleanup_kateparse (GstElement * kateparse)
+{
+ GST_DEBUG ("cleanup_kateparse");
+ gst_element_set_state (kateparse, GST_STATE_NULL);
+
+ gst_pad_set_active (myparsesrcpad, FALSE);
+ gst_pad_set_active (myparsesinkpad, FALSE);
+ gst_check_teardown_src_pad (kateparse);
+ gst_check_teardown_sink_pad (kateparse);
+ gst_check_teardown_element (kateparse);
+}
+
+static GstElement *
+setup_katetag (void)
+{
+ GstElement *katetag;
+
+ GST_DEBUG ("setup_katetag");
+ katetag = gst_check_setup_element ("katetag");
+ mytagsrcpad = gst_check_setup_src_pad (katetag, &srctemplate, NULL);
+ mytagsinkpad = gst_check_setup_sink_pad (katetag, &sinktemplate, NULL);
+ gst_pad_set_active (mytagsrcpad, TRUE);
+ gst_pad_set_active (mytagsinkpad, TRUE);
+
+ return katetag;
+}
+
+static void
+cleanup_katetag (GstElement * katetag)
+{
+ GST_DEBUG ("cleanup_katetag");
+ gst_element_set_state (katetag, GST_STATE_NULL);
+
+ gst_pad_set_active (mytagsrcpad, FALSE);
+ gst_pad_set_active (mytagsinkpad, FALSE);
+ gst_check_teardown_src_pad (katetag);
+ gst_check_teardown_sink_pad (katetag);
+ gst_check_teardown_element (katetag);
+}
+
+static void
+check_buffers (guint expected, gboolean headers_in_caps)
+{
+ GstBuffer *outbuffer;
+ guint i, num_buffers;
+ const int num_headers = 9;
+ unsigned char packet_type;
+
+ /* check buffers are the type we expect */
+ num_buffers = g_list_length (buffers);
+ fail_unless (num_buffers >= num_headers + expected); /* at least 9 headers, plus a variable number of data packets */
+ for (i = 0; i < num_buffers; ++i) {
+ outbuffer = GST_BUFFER (buffers->data);
+ fail_if (outbuffer == NULL);
+ fail_if (GST_BUFFER_SIZE (outbuffer) == 0);
+
+ if (i < num_headers) {
+ /* different headers packets */
+ packet_type = (0x80 | i);
+ fail_unless (GST_BUFFER_DATA (outbuffer)[0] == packet_type);
+ /* headers could be in caps, so would have an extra ref */
+ } else if (i == num_buffers - 1) {
+ /* eos data packet */
+ packet_type = 0x7f;
+ fail_unless (GST_BUFFER_DATA (outbuffer)[0] == packet_type);
+ } else {
+ /* data packet */
+ packet_type = 0;
+ fail_unless (GST_BUFFER_DATA (outbuffer)[0] >= 0
+ && GST_BUFFER_DATA (outbuffer)[0] < 0x7f);
+ }
+
+ buffers = g_list_remove (buffers, outbuffer);
+
+ ASSERT_BUFFER_REFCOUNT (outbuffer, "outbuffer", 1);
+ gst_buffer_unref (outbuffer);
+ outbuffer = NULL;
+ }
+}
+
+GST_START_TEST (test_kate_typefind)
+{
+ GstTypeFindProbability prob;
+ const gchar *type;
+ GstBuffer *buf;
+ GstCaps *caps = NULL;
+
+ buf = gst_buffer_new ();
+ GST_BUFFER_DATA (buf) = (guint8 *) kate_header_0x80;
+ GST_BUFFER_SIZE (buf) = sizeof (kate_header_0x80);
+ GST_BUFFER_OFFSET (buf) = 0;
+
+ caps = gst_type_find_helper_for_buffer (NULL, buf, &prob);
+ fail_unless (caps != NULL);
+ GST_LOG ("Found type: %" GST_PTR_FORMAT, caps);
+
+ type = gst_structure_get_name (gst_caps_get_structure (caps, 0));
+ fail_unless_equals_string (type, "application/x-kate");
+ fail_unless (prob > GST_TYPE_FIND_MINIMUM && prob <= GST_TYPE_FIND_MAXIMUM);
+
+ gst_buffer_unref (buf);
+ gst_caps_unref (caps);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_kate_empty_identification_header)
+{
+ GstElement *katedec;
+ GstBuffer *inbuffer;
+ GstBus *bus;
+
+ katedec = setup_katedec ();
+ bus = gst_bus_new ();
+
+ fail_unless (gst_element_set_state (katedec,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+
+ inbuffer = gst_buffer_new_and_alloc (0);
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+
+ /* set a bus here so we avoid getting state change messages */
+ gst_element_set_bus (katedec, bus);
+
+ fail_unless_equals_int (gst_pad_push (mydecsrcpad, inbuffer), GST_FLOW_ERROR);
+ /* ... but it ends up being collected on the global buffer list */
+ fail_unless_equals_int (g_list_length (buffers), 0);
+
+ gst_element_set_bus (katedec, NULL);
+
+ /* cleanup */
+ gst_object_unref (GST_OBJECT (bus));
+ cleanup_katedec (katedec);
+}
+
+GST_END_TEST;
+
+/* FIXME: also tests comment header */
+GST_START_TEST (test_kate_identification_header)
+{
+ GstElement *katedec;
+ GstBuffer *inbuffer;
+ GstBus *bus;
+ GstMessage *message;
+ GstTagList *tag_list;
+ gchar *language;
+ gchar *title;
+
+ katedec = setup_katedec ();
+ fail_unless (gst_element_set_state (katedec,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+ bus = gst_bus_new ();
+
+ inbuffer = gst_buffer_new_and_alloc (sizeof (kate_header_0x80));
+ memcpy (GST_BUFFER_DATA (inbuffer), kate_header_0x80,
+ sizeof (kate_header_0x80));
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+ gst_buffer_ref (inbuffer);
+
+ gst_element_set_bus (katedec, bus);
+ /* pushing gives away my reference ... */
+ fail_unless (gst_pad_push (mydecsrcpad, inbuffer) == GST_FLOW_OK);
+ /* ... and nothing ends up on the global buffer list */
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+ gst_buffer_unref (inbuffer);
+ fail_unless (g_list_length (buffers) == 0);
+
+ inbuffer = gst_buffer_new_and_alloc (sizeof (kate_header_0x81));
+ memcpy (GST_BUFFER_DATA (inbuffer), kate_header_0x81,
+ sizeof (kate_header_0x81));
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+ gst_buffer_ref (inbuffer);
+
+ /* pushing gives away my reference ... */
+ fail_unless (gst_pad_push (mydecsrcpad, inbuffer) == GST_FLOW_OK);
+ /* ... and nothing ends up on the global buffer list */
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+ gst_buffer_unref (inbuffer);
+ fail_unless (g_list_length (buffers) == 0);
+ /* there's a tag message waiting */
+ fail_if ((message = gst_bus_pop (bus)) == NULL);
+ gst_message_parse_tag (message, &tag_list);
+ fail_unless_equals_int (gst_tag_list_get_tag_size (tag_list,
+ GST_TAG_LANGUAGE_CODE), 1);
+ fail_unless (gst_tag_list_get_string (tag_list, GST_TAG_LANGUAGE_CODE,
+ &language));
+ fail_unless_equals_string (language, "en_GB");
+ g_free (language);
+ fail_unless_equals_int (gst_tag_list_get_tag_size (tag_list, "title"), 1);
+ fail_unless (gst_tag_list_get_string (tag_list, GST_TAG_TITLE, &title));
+ fail_unless_equals_string (title, "Tiger");
+ g_free (title);
+ gst_tag_list_free (tag_list);
+ gst_message_unref (message);
+
+ /* cleanup */
+ gst_bus_set_flushing (bus, TRUE);
+ gst_element_set_bus (katedec, NULL);
+ gst_object_unref (GST_OBJECT (bus));
+ cleanup_katedec (katedec);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_kate_encode_nothing)
+{
+ GstElement *kateenc;
+
+ kateenc = setup_kateenc ();
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+
+ fail_unless (gst_pad_push_event (myencsrcpad, gst_event_new_eos ()) == TRUE);
+
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to ready");
+
+ /* cleanup */
+ cleanup_kateenc (kateenc);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_kate_encode_empty)
+{
+ GstElement *kateenc;
+ GstBuffer *inbuffer;
+ GstBus *bus;
+ GstCaps *caps;
+
+ kateenc = setup_kateenc ();
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+ bus = gst_bus_new ();
+
+ inbuffer = gst_buffer_new_and_alloc (0);
+ GST_BUFFER_TIMESTAMP (inbuffer) = GST_BUFFER_OFFSET (inbuffer) =
+ 1 * GST_SECOND;
+ GST_BUFFER_DURATION (inbuffer) = 5 * GST_SECOND;
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+ gst_buffer_ref (inbuffer);
+
+ caps = gst_caps_from_string ("text/plain");
+ fail_unless (caps != NULL);
+ gst_buffer_set_caps (inbuffer, caps);
+ gst_caps_unref (caps);
+
+ gst_element_set_bus (kateenc, bus);
+ /* pushing gives away my reference ... */
+ fail_unless (gst_pad_push (myencsrcpad, inbuffer) == GST_FLOW_ERROR);
+
+ fail_unless (gst_pad_push_event (myencsrcpad, gst_event_new_eos ()) == TRUE);
+
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to ready");
+
+ gst_element_set_bus (kateenc, NULL);
+
+ /* cleanup */
+ gst_object_unref (GST_OBJECT (bus));
+ cleanup_kateenc (kateenc);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_kate_encode_simple)
+{
+ GstElement *kateenc;
+ GstBuffer *inbuffer;
+ GstBus *bus;
+ const gchar *test_string = "";
+ GstCaps *caps;
+
+ kateenc = setup_kateenc ();
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+ bus = gst_bus_new ();
+
+ inbuffer = gst_buffer_new_and_alloc (strlen (test_string) + 1);
+ memcpy (GST_BUFFER_DATA (inbuffer), test_string, strlen (test_string) + 1);
+ GST_BUFFER_TIMESTAMP (inbuffer) = GST_BUFFER_OFFSET (inbuffer) =
+ 1 * GST_SECOND;
+ GST_BUFFER_DURATION (inbuffer) = 5 * GST_SECOND;
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+ gst_buffer_ref (inbuffer);
+
+ caps = gst_caps_from_string ("text/plain");
+ fail_unless (caps != NULL);
+ gst_buffer_set_caps (inbuffer, caps);
+ gst_caps_unref (caps);
+
+ gst_element_set_bus (kateenc, bus);
+ /* pushing gives away my reference ... */
+ fail_unless (gst_pad_push (myencsrcpad, inbuffer) == GST_FLOW_OK);
+ /* ... and nothing ends up on the global buffer list */
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+ gst_buffer_unref (inbuffer);
+ fail_unless (gst_pad_push_event (myencsrcpad, gst_event_new_eos ()) == TRUE);
+
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to ready");
+
+ /* at least one data packet and one EOS packet should have been emitted */
+ check_buffers (1 + 1, FALSE);
+
+ /* cleanup */
+ gst_bus_set_flushing (bus, TRUE);
+ gst_element_set_bus (kateenc, NULL);
+ gst_object_unref (GST_OBJECT (bus));
+ cleanup_kateenc (kateenc);
+ g_list_free (buffers);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_kate_encode_spu)
+{
+ GstElement *kateenc;
+ GstBuffer *inbuffer;
+ GstBus *bus;
+ GstCaps *caps;
+
+ kateenc = setup_kateenc ();
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+ bus = gst_bus_new ();
+
+ inbuffer = gst_buffer_new_and_alloc (sizeof (kate_spu));
+ memcpy (GST_BUFFER_DATA (inbuffer), kate_spu, sizeof (kate_spu));
+ GST_BUFFER_TIMESTAMP (inbuffer) = GST_BUFFER_OFFSET (inbuffer) =
+ 1 * GST_SECOND;
+ GST_BUFFER_DURATION (inbuffer) = 5 * GST_SECOND;
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+ gst_buffer_ref (inbuffer);
+
+ caps = gst_caps_from_string ("video/x-dvd-subpicture");
+ fail_unless (caps != NULL);
+ gst_buffer_set_caps (inbuffer, caps);
+ gst_caps_unref (caps);
+
+ gst_element_set_bus (kateenc, bus);
+ /* pushing gives away my reference ... */
+ fail_unless_equals_int (gst_pad_push (myencsrcpad, inbuffer), GST_FLOW_OK);
+ /* ... and nothing ends up on the global buffer list */
+ ASSERT_BUFFER_REFCOUNT (inbuffer, "inbuffer", 1);
+ gst_buffer_unref (inbuffer);
+ fail_unless (gst_pad_push_event (myencsrcpad, gst_event_new_eos ()) == TRUE);
+
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to ready");
+
+ /* at least one data packet and one EOS packet should have been emitted */
+ check_buffers (2, FALSE);
+
+ /* cleanup */
+ gst_bus_set_flushing (bus, TRUE);
+ gst_element_set_bus (kateenc, NULL);
+ gst_object_unref (GST_OBJECT (bus));
+ cleanup_kateenc (kateenc);
+ g_list_free (buffers);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_kate_encode_keepalives)
+{
+ GstElement *kateenc;
+ GstBus *bus;
+ guint i, round;
+ enum
+ { n_keepalives = 1000 };
+ static const struct
+ {
+ gdouble keepalive_min_time;
+ gint packets;
+ } cfg[3] = {
+ {
+ 0.5, n_keepalives}, {
+ 2.0, n_keepalives / 2}, {
+ 5.0, n_keepalives / 5},};
+
+ for (round = 0; round < 3; ++round) {
+ kateenc = setup_kateenc ();
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+ bus = gst_bus_new ();
+
+ gst_element_set_bus (kateenc, bus);
+
+ g_object_set (kateenc, "keepalive-min-time", cfg[round].keepalive_min_time,
+ NULL);
+
+ /* the second one here should not emit a keepalive since the time since last packet
+ is less than the keepalive delay */
+ for (i = 1; i <= n_keepalives; ++i) {
+ gint64 t = i * GST_SECOND;
+ fail_unless (gst_pad_push_event (myencsrcpad,
+ gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, t, -1,
+ 0)) == TRUE);
+ }
+
+ fail_unless (gst_pad_push_event (myencsrcpad,
+ gst_event_new_eos ()) == TRUE);
+
+ fail_unless (gst_element_set_state (kateenc,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to ready");
+
+ /* at least a number data packet and an EOS packet should have been emitted */
+ check_buffers (cfg[round].packets + 1, FALSE);
+
+ /* cleanup */
+ gst_bus_set_flushing (bus, TRUE);
+ gst_element_set_bus (kateenc, NULL);
+ gst_object_unref (GST_OBJECT (bus));
+ cleanup_kateenc (kateenc);
+ g_list_free (buffers);
+ }
+}
+
+GST_END_TEST;
+
+static void
+test_kate_send_headers (GstPad * pad)
+{
+ GstBuffer *inbuffer;
+ int i;
+
+ /* push headers */
+ inbuffer = gst_buffer_new ();
+ GST_BUFFER_DATA (inbuffer) = (guint8 *) kate_header_0x80;
+ GST_BUFFER_SIZE (inbuffer) = sizeof (kate_header_0x80);
+ GST_BUFFER_OFFSET (inbuffer) = GST_BUFFER_OFFSET_END (inbuffer) = 0;
+ fail_unless_equals_int (gst_pad_push (pad, inbuffer), GST_FLOW_OK);
+
+ inbuffer = gst_buffer_new ();
+ GST_BUFFER_DATA (inbuffer) = (guint8 *) kate_header_0x81;
+ GST_BUFFER_SIZE (inbuffer) = sizeof (kate_header_0x81);
+ GST_BUFFER_OFFSET (inbuffer) = GST_BUFFER_OFFSET_END (inbuffer) = 0;
+ fail_unless_equals_int (gst_pad_push (pad, inbuffer), GST_FLOW_OK);
+
+ for (i = 2; i < 8; ++i) {
+ inbuffer = gst_buffer_new_and_alloc (sizeof (kate_header_0x8x));
+ memcpy (GST_BUFFER_DATA (inbuffer), (guint8 *) kate_header_0x8x,
+ sizeof (kate_header_0x8x));
+ GST_BUFFER_DATA (inbuffer)[0] = 0x80 | i;
+ GST_BUFFER_OFFSET (inbuffer) = GST_BUFFER_OFFSET_END (inbuffer) = 0;
+ fail_unless_equals_int (gst_pad_push (pad, inbuffer), GST_FLOW_OK);
+ }
+
+ inbuffer = gst_buffer_new ();
+ GST_BUFFER_DATA (inbuffer) = (guint8 *) kate_header_0x88;
+ GST_BUFFER_SIZE (inbuffer) = sizeof (kate_header_0x88);
+ GST_BUFFER_OFFSET (inbuffer) = GST_BUFFER_OFFSET_END (inbuffer) = 0;
+ fail_unless_equals_int (gst_pad_push (pad, inbuffer), GST_FLOW_OK);
+}
+
+GST_START_TEST (test_kate_parse)
+{
+ GstElement *kateparse;
+ GstBuffer *inbuffer;
+ GstBus *bus;
+
+ kateparse = setup_kateparse ();
+ fail_unless (gst_element_set_state (kateparse,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+ bus = gst_bus_new ();
+
+ gst_element_set_bus (kateparse, bus);
+
+ test_kate_send_headers (myparsesrcpad);
+
+ /* push a text packet */
+ inbuffer = gst_buffer_new ();
+ GST_BUFFER_DATA (inbuffer) = (guint8 *) kate_header_0x00;
+ GST_BUFFER_SIZE (inbuffer) = sizeof (kate_header_0x00);
+ GST_BUFFER_TIMESTAMP (inbuffer) = GST_BUFFER_OFFSET (inbuffer) =
+ 1 * GST_SECOND;
+ GST_BUFFER_DURATION (inbuffer) = 5 * GST_SECOND;
+ GST_BUFFER_OFFSET_END (inbuffer) = (GST_BUFFER_TIMESTAMP (inbuffer) << 32); /* granpos */
+ fail_unless_equals_int (gst_pad_push (myparsesrcpad, inbuffer), GST_FLOW_OK);
+
+ /* push a eos packet */
+ inbuffer = gst_buffer_new ();
+ GST_BUFFER_DATA (inbuffer) = (guint8 *) kate_header_0x7f;
+ GST_BUFFER_SIZE (inbuffer) = sizeof (kate_header_0x7f);
+ GST_BUFFER_TIMESTAMP (inbuffer) = GST_BUFFER_OFFSET (inbuffer) =
+ 6 * GST_SECOND;
+ GST_BUFFER_DURATION (inbuffer) = 0;
+ GST_BUFFER_OFFSET_END (inbuffer) = (GST_BUFFER_TIMESTAMP (inbuffer) << 32); /* granpos */
+ fail_unless_equals_int (gst_pad_push (myparsesrcpad, inbuffer), GST_FLOW_OK);
+
+ /* signal eos */
+ fail_unless (gst_pad_push_event (myparsesrcpad,
+ gst_event_new_eos ()) == TRUE);
+
+ fail_unless (gst_element_set_state (kateparse,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to ready");
+
+ /* at least one data packet and one EOS packet should have been emitted */
+ check_buffers (2, TRUE);
+
+ /* cleanup */
+ gst_bus_set_flushing (bus, TRUE);
+ gst_element_set_bus (kateparse, NULL);
+ gst_object_unref (GST_OBJECT (bus));
+ cleanup_kateparse (kateparse);
+ g_list_free (buffers);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_kate_tag_passthrough)
+{
+ GstElement *katetag;
+ GstBus *bus;
+ GstBuffer *outbuffer;
+ GList *list;
+
+ katetag = setup_katetag ();
+ fail_unless (gst_element_set_state (katetag,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+ bus = gst_bus_new ();
+
+ gst_element_set_bus (katetag, bus);
+
+ test_kate_send_headers (mytagsrcpad);
+
+ /* signal eos */
+ fail_unless (gst_pad_push_event (mytagsrcpad, gst_event_new_eos ()) == TRUE);
+
+ fail_unless (gst_element_set_state (katetag,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to ready");
+
+ /* get the first buffer and check language/category */
+ fail_unless (g_list_length (buffers) >= 2); /* ID header, Vorbis comments header */
+ outbuffer = GST_BUFFER (buffers->data);
+ fail_if (outbuffer == NULL);
+
+ /* check identification header is unchanged */
+ list = g_list_nth (buffers, 0);
+ fail_unless (list != NULL);
+ outbuffer = list->data;
+ fail_unless_equals_int (GST_BUFFER_SIZE (outbuffer),
+ sizeof (kate_header_0x80));
+ fail_unless_equals_int (memcmp (GST_BUFFER_DATA (outbuffer), kate_header_0x80,
+ sizeof (kate_header_0x80)), 0);
+
+ /* check comment header is unchanged */
+ list = g_list_nth (buffers, 1);
+ fail_unless (list != NULL);
+ outbuffer = list->data;
+ fail_unless_equals_int (GST_BUFFER_SIZE (outbuffer),
+ sizeof (kate_header_0x81));
+ fail_unless_equals_int (memcmp (GST_BUFFER_DATA (outbuffer), kate_header_0x81,
+ sizeof (kate_header_0x81)), 0);
+
+ /* all headers should have been emitted, but no particular packets */
+ check_buffers (0, TRUE);
+
+ /* cleanup */
+ gst_bus_set_flushing (bus, TRUE);
+ gst_element_set_bus (katetag, NULL);
+ gst_object_unref (GST_OBJECT (bus));
+ cleanup_katetag (katetag);
+ g_list_free (buffers);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_kate_tag)
+{
+ GstElement *katetag;
+ GstBus *bus;
+ GstBuffer *outbuffer;
+
+ katetag = setup_katetag ();
+ fail_unless (gst_element_set_state (katetag,
+ GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to playing");
+ bus = gst_bus_new ();
+
+ gst_element_set_bus (katetag, bus);
+
+ g_object_set (katetag, "language", "cy", NULL);
+ g_object_set (katetag, "category", "subtitles", NULL);
+
+ test_kate_send_headers (mytagsrcpad);
+
+ /* signal eos */
+ fail_unless (gst_pad_push_event (mytagsrcpad, gst_event_new_eos ()) == TRUE);
+
+ fail_unless (gst_element_set_state (katetag,
+ GST_STATE_READY) == GST_STATE_CHANGE_SUCCESS,
+ "could not set to ready");
+
+ /* get the first buffer and check language/category */
+ fail_unless (g_list_length (buffers) >= 1);
+ outbuffer = GST_BUFFER (buffers->data);
+ fail_if (outbuffer == NULL);
+ fail_if (GST_BUFFER_SIZE (outbuffer) != 64);
+ fail_if (strcmp ((const char *) GST_BUFFER_DATA (outbuffer) + 32, "cy"));
+ fail_if (strcmp ((const char *) GST_BUFFER_DATA (outbuffer) + 48,
+ "subtitles"));
+
+ /* all headers should have been emitted, but no particular packets */
+ check_buffers (0, TRUE);
+
+ /* cleanup */
+ gst_bus_set_flushing (bus, TRUE);
+ gst_element_set_bus (katetag, NULL);
+ gst_object_unref (GST_OBJECT (bus));
+ cleanup_katetag (katetag);
+ g_list_free (buffers);
+}
+
+GST_END_TEST;
+
+Suite *
+kate_suite (void)
+{
+ Suite *s = suite_create ("kate");
+ TCase *tc_chain = tcase_create ("general");
+
+ suite_add_tcase (s, tc_chain);
+
+#define X if (0)
+ tcase_add_test (tc_chain, test_kate_typefind);
+ tcase_add_test (tc_chain, test_kate_empty_identification_header);
+ tcase_add_test (tc_chain, test_kate_identification_header);
+ tcase_add_test (tc_chain, test_kate_encode_nothing);
+ tcase_add_test (tc_chain, test_kate_encode_empty);
+ tcase_add_test (tc_chain, test_kate_encode_simple);
+ tcase_add_test (tc_chain, test_kate_encode_spu);
+ tcase_add_test (tc_chain, test_kate_encode_keepalives);
+ tcase_add_test (tc_chain, test_kate_parse);
+ tcase_add_test (tc_chain, test_kate_tag_passthrough);
+ tcase_add_test (tc_chain, test_kate_tag);
+#undef X
+
+ return s;
+}
+
+int
+main (int argc, char **argv)
+{
+ int nf;
+
+ Suite *s = kate_suite ();
+ SRunner *sr = srunner_create (s);
+
+ gst_check_init (&argc, &argv);
+
+ srunner_run_all (sr, CK_NORMAL);
+ nf = srunner_ntests_failed (sr);
+ srunner_free (sr);
+
+ return nf;
+}