gstreamer/subprojects/gst-plugins-bad/tests/check/elements/kate.c

881 lines
28 KiB
C

/* 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., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, 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_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 GList *events = NULL;
static gboolean
collect_events_func (GstPad * pad, GstObject * parent, GstEvent * event)
{
GST_LOG ("event: %" GST_PTR_FORMAT, event);
events = g_list_append (events, gst_event_ref (event));
return gst_pad_event_default (pad, parent, event);
}
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);
mydecsinkpad = gst_check_setup_sink_pad (katedec, &sinktemplate);
gst_pad_set_event_function (mydecsinkpad, collect_events_func);
gst_pad_set_active (mydecsrcpad, TRUE);
gst_pad_set_active (mydecsinkpad, TRUE);
return katedec;
}
static void
cleanup_katedec (GstElement * katedec)
{
GList *l;
GST_DEBUG ("cleanup_katedec");
gst_element_set_state (katedec, GST_STATE_NULL);
for (l = events; l != NULL; l = l->next)
gst_event_unref (GST_EVENT (l->data));
g_list_free (events);
events = 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);
myencsinkpad = gst_check_setup_sink_pad (kateenc, &sinktemplate);
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);
myparsesinkpad = gst_check_setup_sink_pad (kateparse, &sinktemplate);
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);
mytagsinkpad = gst_check_setup_sink_pad (katetag, &sinktemplate);
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;
GstMapInfo info;
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_get_size (outbuffer) == 0);
assert_equals_int (gst_buffer_map (outbuffer, &info, GST_MAP_READ), TRUE);
if (i < num_headers) {
/* different headers packets */
packet_type = (0x80 | i);
fail_unless (info.data[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 (info.data[0] == packet_type);
} else {
/* data packet */
packet_type = 0;
fail_unless (info.data[0] >= 0 && info.data[0] < 0x7f);
}
gst_buffer_unmap (outbuffer, &info);
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_memdup (kate_header_0x80, 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;
GstCaps *caps;
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");
caps = gst_caps_new_empty_simple ("subtitle/x-kate");
gst_check_setup_events (mydecsrcpad, katedec, caps, GST_FORMAT_TIME);
gst_caps_unref (caps);
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;
GstCaps *caps;
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 ();
caps = gst_caps_new_empty_simple ("subtitle/x-kate");
gst_check_setup_events (mydecsrcpad, katedec, caps, GST_FORMAT_TIME);
gst_caps_unref (caps);
inbuffer =
gst_buffer_new_memdup (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_memdup (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 should've been a tag event */
{
gboolean found_tags = FALSE;
GList *l;
for (l = events; l != NULL; l = l->next) {
GstEvent *event = GST_EVENT (l->data);
if (GST_EVENT_TYPE (event) == GST_EVENT_TAG) {
GstTagList *tags = NULL;
gchar *language;
gchar *title;
found_tags = TRUE;
gst_event_parse_tag (event, &tags);
fail_unless (tags != NULL);
fail_unless (gst_tag_list_get_scope (tags) == GST_TAG_SCOPE_STREAM);
fail_unless_equals_int (gst_tag_list_get_tag_size (tags,
GST_TAG_LANGUAGE_CODE), 1);
fail_unless (gst_tag_list_get_string (tags, GST_TAG_LANGUAGE_CODE,
&language));
fail_unless_equals_string (language, "en");
g_free (language);
fail_unless_equals_int (gst_tag_list_get_tag_size (tags, "title"), 1);
fail_unless (gst_tag_list_get_string (tags, GST_TAG_TITLE, &title));
fail_unless_equals_string (title, "Tiger");
g_free (title);
}
}
fail_unless (found_tags);
}
/* 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);
caps = gst_caps_from_string ("text/x-raw, format=utf8");
fail_unless (caps != NULL);
gst_check_setup_events (myencsrcpad, kateenc, caps, GST_FORMAT_TIME);
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 ();
g_object_set (kateenc, "category", "subtitles", NULL);
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_memdup (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);
caps = gst_caps_from_string ("text/x-raw, format=utf8");
fail_unless (caps != NULL);
gst_check_setup_events (myencsrcpad, kateenc, caps, GST_FORMAT_TIME);
gst_caps_unref (caps);
gst_buffer_ref (inbuffer);
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 (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);
gst_check_drop_buffers ();
}
GST_END_TEST;
GST_START_TEST (test_kate_encode_spu)
{
GstElement *kateenc;
GstBuffer *inbuffer;
GstBus *bus;
GstCaps *caps;
kateenc = setup_kateenc ();
g_object_set (kateenc, "category", "spu-subtitles", NULL);
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_memdup (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);
caps = gst_caps_from_string ("subpicture/x-dvd");
fail_unless (caps != NULL);
gst_check_setup_events (myencsrcpad, kateenc, caps, GST_FORMAT_TIME);
gst_caps_unref (caps);
gst_buffer_ref (inbuffer);
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);
gst_check_drop_buffers ();
}
GST_END_TEST;
GST_START_TEST (test_kate_encode_keepalives)
{
GstElement *kateenc;
GstBus *bus;
guint i, round;
GstSegment segment;
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 ();
/* doesn't matter here, since we never send a packet */
g_object_set (kateenc, "category", "subtitles", NULL);
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);
gst_pad_push_event (myencsrcpad, gst_event_new_stream_start ("test"));
/* 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;
gst_segment_init (&segment, GST_FORMAT_TIME);
segment.start = t;
segment.position = 0;
fail_unless (gst_pad_push_event (myencsrcpad,
gst_event_new_segment (&segment)) == 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);
gst_check_drop_buffers ();
}
}
GST_END_TEST;
static void
test_kate_send_headers (GstElement * element, GstPad * pad)
{
GstBuffer *inbuffer;
GstCaps *caps;
GstMapInfo info;
int i;
caps = gst_caps_new_simple ("subtitle/x-kate", NULL, NULL);
gst_check_setup_events (pad, element, caps, GST_FORMAT_TIME);
gst_caps_unref (caps);
/* push headers */
inbuffer =
gst_buffer_new_memdup (kate_header_0x80, 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_memdup (kate_header_0x81, 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_memdup (kate_header_0x8x, sizeof (kate_header_0x8x));
fail_if (gst_buffer_map (inbuffer, &info, GST_MAP_WRITE) != TRUE);
info.data[0] = 0x80 | i;
gst_buffer_unmap (inbuffer, &info);
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_memdup (kate_header_0x88, 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 (kateparse, myparsesrcpad);
/* push a text packet */
inbuffer =
gst_buffer_new_memdup (kate_header_0x00, 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_memdup (kate_header_0x7f, 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_foreach (buffers, (GFunc) gst_buffer_unref, NULL);
gst_check_drop_buffers ();
}
GST_END_TEST;
GST_START_TEST (test_kate_tag_passthrough)
{
GstElement *katetag;
GstBus *bus;
GstBuffer *outbuffer;
GList *list;
GstMapInfo info;
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 (katetag, 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_if (gst_buffer_map (outbuffer, &info, GST_MAP_READ) != TRUE);
fail_unless_equals_int (info.size, sizeof (kate_header_0x80));
fail_unless_equals_int (memcmp (info.data, kate_header_0x80,
sizeof (kate_header_0x80)), 0);
gst_buffer_unmap (outbuffer, &info);
/* check comment header is unchanged */
list = g_list_nth (buffers, 1);
fail_unless (list != NULL);
outbuffer = list->data;
fail_if (gst_buffer_map (outbuffer, &info, GST_MAP_READ) != TRUE);
fail_unless_equals_int (info.size, sizeof (kate_header_0x81));
fail_unless_equals_int (memcmp (info.data, kate_header_0x81,
sizeof (kate_header_0x81)), 0);
gst_buffer_unmap (outbuffer, &info);
/* 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);
gst_check_drop_buffers ();
}
GST_END_TEST;
GST_START_TEST (test_kate_tag)
{
GstElement *katetag;
GstBus *bus;
GstBuffer *outbuffer;
GstMapInfo info;
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 (katetag, 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);
assert_equals_int (gst_buffer_map (outbuffer, &info, GST_MAP_READ), TRUE);
fail_if (info.size != 64);
fail_if (strcmp ((const char *) info.data + 32, "cy"));
fail_if (strcmp ((const char *) info.data + 48, "subtitles"));
gst_buffer_unmap (outbuffer, &info);
/* 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);
gst_check_drop_buffers ();
}
GST_END_TEST;
static Suite *
kate_suite (void)
{
Suite *s = suite_create ("kate");
TCase *tc_chain = tcase_create ("general");
suite_add_tcase (s, tc_chain);
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);
return s;
}
GST_CHECK_MAIN (kate);