| /* GStreamer |
| * |
| * unit test for avisubtitle |
| * |
| * Copyright (C) <2007> Thijs Vermeir <thijsvermeir@gmail.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. |
| */ |
| /* Element-Checklist-Version: 5 */ |
| |
| #include <unistd.h> |
| |
| #include <gst/gst.h> |
| #include <gst/check/gstcheck.h> |
| |
| GstPad *mysinkpad; |
| GstPad *mysrcpad; |
| |
| guint8 avisub_utf_8_with_bom[] = { |
| 0x47, 0x41, 0x42, 0x32, 0x00, 0x02, 0x00, 0x10, |
| 0x00, 0x00, 0x00, 0x45, 0x00, 0x6e, 0x00, 0x67, |
| 0x00, 0x6c, 0x00, 0x69, 0x00, 0x73, 0x00, 0x68, |
| 0x00, 0x00, 0x00, 0x04, 0x00, 0x8e, 0x00, 0x00, |
| 0x00, 0xef, 0xbb, 0xbf, 0x31, 0x0d, 0x0a, 0x30, |
| 0x30, 0x3a, 0x30, 0x30, 0x3a, 0x30, 0x30, 0x2c, |
| 0x31, 0x30, 0x30, 0x20, 0x2d, 0x2d, 0x3e, 0x20, |
| 0x30, 0x30, 0x3a, 0x30, 0x30, 0x3a, 0x30, 0x32, |
| 0x2c, 0x30, 0x30, 0x30, 0x0d, 0x0a, 0x3c, 0x62, |
| 0x3e, 0x41, 0x6e, 0x20, 0x55, 0x54, 0x46, 0x38, |
| 0x20, 0x53, 0x75, 0x62, 0x74, 0x69, 0x74, 0x6c, |
| 0x65, 0x20, 0x77, 0x69, 0x74, 0x68, 0x20, 0x42, |
| 0x4f, 0x4d, 0x3c, 0x2f, 0x62, 0x3e, 0x0d, 0x0a, |
| 0x0d, 0x0a, 0x32, 0x0d, 0x0a, 0x30, 0x30, 0x3a, |
| 0x30, 0x30, 0x3a, 0x30, 0x32, 0x2c, 0x31, 0x30, |
| 0x30, 0x20, 0x2d, 0x2d, 0x3e, 0x20, 0x30, 0x30, |
| 0x3a, 0x30, 0x30, 0x3a, 0x30, 0x34, 0x2c, 0x30, |
| 0x30, 0x30, 0x0d, 0x0a, 0x53, 0x6f, 0x6d, 0x65, |
| 0x74, 0x68, 0x69, 0x6e, 0x67, 0x20, 0x6e, 0x6f, |
| 0x6e, 0x41, 0x53, 0x43, 0x49, 0x49, 0x20, 0x2d, |
| 0x20, 0xc2, 0xb5, 0xc3, 0xb6, 0xc3, 0xa4, 0xc3, |
| 0xbc, 0xc3, 0x9f, 0x0d, 0x0a, 0x0d, 0x0a |
| }; |
| |
| static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink", |
| GST_PAD_SINK, |
| GST_PAD_ALWAYS, |
| GST_STATIC_CAPS ("application/x-subtitle") |
| ); |
| |
| static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src", |
| GST_PAD_SRC, |
| GST_PAD_ALWAYS, |
| GST_STATIC_CAPS ("application/x-subtitle-avi") |
| ); |
| |
| static GstElement * |
| setup_avisubtitle (void) |
| { |
| GstElement *avisubtitle; |
| GstCaps *srccaps; |
| |
| GST_DEBUG ("setup_avisubtitle"); |
| avisubtitle = gst_check_setup_element ("avisubtitle"); |
| mysinkpad = gst_check_setup_sink_pad (avisubtitle, &sink_template); |
| mysrcpad = gst_check_setup_src_pad (avisubtitle, &src_template); |
| gst_pad_set_active (mysinkpad, TRUE); |
| gst_pad_set_active (mysrcpad, TRUE); |
| srccaps = gst_caps_new_empty_simple ("application/x-subtitle-avi"); |
| gst_check_setup_events (mysrcpad, avisubtitle, srccaps, GST_FORMAT_TIME); |
| gst_caps_unref (srccaps); |
| return avisubtitle; |
| } |
| |
| static void |
| cleanup_avisubtitle (GstElement * avisubtitle) |
| { |
| gst_pad_set_active (mysinkpad, FALSE); |
| gst_pad_set_active (mysrcpad, FALSE); |
| gst_check_teardown_sink_pad (avisubtitle); |
| gst_check_teardown_src_pad (avisubtitle); |
| gst_check_teardown_element (avisubtitle); |
| } |
| |
| static void |
| check_wrong_buffer (guint8 * data, guint length) |
| { |
| GstBuffer *buffer = gst_buffer_new_allocate (NULL, length, 0); |
| GstElement *avisubtitle = setup_avisubtitle (); |
| |
| gst_buffer_fill (buffer, 0, data, length); |
| fail_unless (gst_element_set_state (avisubtitle, |
| GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS, |
| "could not set to playing"); |
| gst_buffer_ref (buffer); |
| ASSERT_BUFFER_REFCOUNT (buffer, "inbuffer", 2); |
| /* push the broken buffer */ |
| fail_unless (gst_pad_push (mysrcpad, buffer) == GST_FLOW_ERROR, |
| "accepted a broken buffer"); |
| /* check if we have unreffed this buffer on failure */ |
| ASSERT_BUFFER_REFCOUNT (buffer, "inbuffer", 1); |
| gst_buffer_unref (buffer); |
| fail_unless (gst_element_set_state (avisubtitle, |
| GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null"); |
| cleanup_avisubtitle (avisubtitle); |
| } |
| |
| static void |
| check_correct_buffer (guint8 * src_data, guint src_size, guint8 * dst_data, |
| guint dst_size) |
| { |
| GstBuffer *buffer = gst_buffer_new_allocate (NULL, src_size, 0); |
| GstBuffer *newBuffer; |
| GstElement *avisubtitle = setup_avisubtitle (); |
| GstEvent *event; |
| |
| fail_unless (g_list_length (buffers) == 0, "Buffers list needs to be empty"); |
| gst_buffer_fill (buffer, 0, src_data, src_size); |
| fail_unless (gst_element_set_state (avisubtitle, |
| GST_STATE_PLAYING) == GST_STATE_CHANGE_SUCCESS, |
| "could not set to playing"); |
| ASSERT_BUFFER_REFCOUNT (buffer, "inbuffer", 1); |
| event = gst_event_new_seek (1.0, GST_FORMAT_TIME, GST_SEEK_FLAG_FLUSH, |
| GST_SEEK_TYPE_SET, 2 * GST_SECOND, GST_SEEK_TYPE_SET, 5 * GST_SECOND); |
| fail_unless (gst_element_send_event (avisubtitle, event) == FALSE, |
| "Seeking is not possible when there is no buffer yet"); |
| fail_unless (gst_pad_push (mysrcpad, buffer) == GST_FLOW_OK, |
| "not accepted a correct buffer"); |
| /* we gave away our reference to the buffer, don't assume anything */ |
| buffer = NULL; |
| /* a new buffer is created in the list */ |
| fail_unless (g_list_length (buffers) == 1, |
| "No new buffer in the buffers list"); |
| event = gst_event_new_seek (1.0, GST_FORMAT_TIME, GST_SEEK_FLAG_FLUSH, |
| GST_SEEK_TYPE_SET, 2 * GST_SECOND, GST_SEEK_TYPE_SET, 5 * GST_SECOND); |
| fail_unless (gst_element_send_event (avisubtitle, event) == TRUE, |
| "seeking should be working now"); |
| fail_unless (g_list_length (buffers) == 2, |
| "After seeking we need another buffer in the buffers"); |
| newBuffer = GST_BUFFER (buffers->data); |
| buffers = g_list_remove (buffers, newBuffer); |
| fail_unless (g_list_length (buffers) == 1, "Buffers list needs to be empty"); |
| fail_unless (gst_buffer_get_size (newBuffer) == dst_size, |
| "size of the new buffer is wrong ( %d != %d)", |
| gst_buffer_get_size (newBuffer), dst_size); |
| fail_unless (gst_buffer_memcmp (newBuffer, 0, dst_data, dst_size) == 0, |
| "data of the buffer is not correct"); |
| gst_buffer_unref (newBuffer); |
| /* free the buffer from seeking */ |
| gst_buffer_unref (GST_BUFFER (buffers->data)); |
| buffers = g_list_remove (buffers, buffers->data); |
| fail_unless (gst_element_set_state (avisubtitle, |
| GST_STATE_NULL) == GST_STATE_CHANGE_SUCCESS, "could not set to null"); |
| cleanup_avisubtitle (avisubtitle); |
| } |
| |
| |
| GST_START_TEST (test_avisubtitle_negative) |
| { |
| guint8 wrong_magic[] = |
| { 0x47, 0x41, 0x41, 0x32, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x00 |
| }; |
| guint8 wrong_fixed_word_2[] = { |
| 0x47, 0x41, 0x42, 0x32, 0x00, 0x02, 0x01, 0x10, |
| 0x00, 0x00, 0x00, 0x45, 0x00, 0x6e, 0x00, 0x67, |
| 0x00, 0x6c, 0x00, 0x69, 0x00, 0x73, 0x00, 0x68 |
| }; |
| guint8 wrong_length_after_name[] = { |
| 0x47, 0x41, 0x42, 0x32, 0x00, 0x02, 0x00, 0x10, |
| 0x00, 0x00, 0x00, 0x45, 0x00, 0x6e, 0x00, 0x67, |
| 0x00, 0x6c, 0x00, 0x69, 0x00, 0x73, 0x00, 0x68 |
| }; |
| guint8 wrong_fixed_word_4[] = { |
| 0x47, 0x41, 0x42, 0x32, 0x00, 0x02, 0x00, 0x10, |
| 0x00, 0x00, 0x00, 0x45, 0x00, 0x6e, 0x00, 0x67, |
| 0x00, 0x6c, 0x00, 0x69, 0x00, 0x73, 0x00, 0x68, |
| 0x00, 0x00, 0x00, 0x04, 0x01, 0x8e, 0x00, 0x00, |
| 0x00, 0xef, 0xbb, 0xbf, 0x31, 0x0d, 0x0a, 0x30 |
| }; |
| guint8 wrong_total_length[] = { |
| 0x47, 0x41, 0x42, 0x32, 0x00, 0x02, 0x00, 0x10, |
| 0x00, 0x00, 0x00, 0x45, 0x00, 0x6e, 0x00, 0x67, |
| 0x00, 0x6c, 0x00, 0x69, 0x00, 0x73, 0x00, 0x68, |
| 0x00, 0x00, 0x00, 0x04, 0x00, 0x8e, 0x00, 0x00, |
| 0x00, 0xef, 0xbb, 0xbf, 0x31, 0x0d, 0x0a, 0x30 |
| }; |
| /* size of the buffer must be larger than 11 */ |
| check_wrong_buffer (avisub_utf_8_with_bom, 11); |
| /* buffer must start with 'GAB2\0' */ |
| check_wrong_buffer (wrong_magic, 14); |
| /* next word must be 2 */ |
| check_wrong_buffer (wrong_fixed_word_2, 24); |
| /* length must be larger than the length of the name + 17 */ |
| check_wrong_buffer (wrong_length_after_name, 24); |
| /* next word must be 4 */ |
| check_wrong_buffer (wrong_fixed_word_4, 36); |
| /* check wrong total length */ |
| check_wrong_buffer (wrong_total_length, 36); |
| } |
| |
| GST_END_TEST; |
| |
| GST_START_TEST (test_avisubtitle_positive) |
| { |
| guint8 avisub_utf_8_without_bom[] = { |
| 0x47, 0x41, 0x42, 0x32, 0x00, 0x02, 0x00, 0x10, |
| 0x00, 0x00, 0x00, 0x45, 0x00, 0x6e, 0x00, 0x67, |
| 0x00, 0x6c, 0x00, 0x69, 0x00, 0x73, 0x00, 0x68, |
| 0x00, 0x00, 0x00, 0x04, 0x00, 0x8b, 0x00, 0x00, |
| 0x00, 0x31, 0x0d, 0x0a, 0x30, |
| 0x30, 0x3a, 0x30, 0x30, 0x3a, 0x30, 0x30, 0x2c, |
| 0x31, 0x30, 0x30, 0x20, 0x2d, 0x2d, 0x3e, 0x20, |
| 0x30, 0x30, 0x3a, 0x30, 0x30, 0x3a, 0x30, 0x32, |
| 0x2c, 0x30, 0x30, 0x30, 0x0d, 0x0a, 0x3c, 0x62, |
| 0x3e, 0x41, 0x6e, 0x20, 0x55, 0x54, 0x46, 0x38, |
| 0x20, 0x53, 0x75, 0x62, 0x74, 0x69, 0x74, 0x6c, |
| 0x65, 0x20, 0x77, 0x69, 0x74, 0x68, 0x20, 0x42, |
| 0x4f, 0x4d, 0x3c, 0x2f, 0x62, 0x3e, 0x0d, 0x0a, |
| 0x0d, 0x0a, 0x32, 0x0d, 0x0a, 0x30, 0x30, 0x3a, |
| 0x30, 0x30, 0x3a, 0x30, 0x32, 0x2c, 0x31, 0x30, |
| 0x30, 0x20, 0x2d, 0x2d, 0x3e, 0x20, 0x30, 0x30, |
| 0x3a, 0x30, 0x30, 0x3a, 0x30, 0x34, 0x2c, 0x30, |
| 0x30, 0x30, 0x0d, 0x0a, 0x53, 0x6f, 0x6d, 0x65, |
| 0x74, 0x68, 0x69, 0x6e, 0x67, 0x20, 0x6e, 0x6f, |
| 0x6e, 0x41, 0x53, 0x43, 0x49, 0x49, 0x20, 0x2d, |
| 0x20, 0xc2, 0xb5, 0xc3, 0xb6, 0xc3, 0xa4, 0xc3, |
| 0xbc, 0xc3, 0x9f, 0x0d, 0x0a, 0x0d, 0x0a |
| }; |
| check_correct_buffer (avisub_utf_8_with_bom, 175, avisub_utf_8_with_bom + 36, |
| 139); |
| check_correct_buffer (avisub_utf_8_without_bom, 172, |
| avisub_utf_8_without_bom + 33, 139); |
| } |
| |
| GST_END_TEST; |
| |
| static Suite * |
| avisubtitle_suite (void) |
| { |
| Suite *s = suite_create ("avisubtitle"); |
| TCase *tc_chain = tcase_create ("general"); |
| |
| suite_add_tcase (s, tc_chain); |
| tcase_add_test (tc_chain, test_avisubtitle_negative); |
| tcase_add_test (tc_chain, test_avisubtitle_positive); |
| |
| return s; |
| } |
| |
| GST_CHECK_MAIN (avisubtitle); |