bytewriter: Add unit test

2 files changed, 241 insertions, 0 deletions

diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am
index ccefe8f285..5e92d70198 100644
--- a/tests/check/Makefile.am
+++ b/tests/check/Makefile.am
@@ -106,6 +106,7 @@ check_PROGRAMS =				\
 	libs/adapter				\
 	libs/bitreader				\
 	libs/bytereader				\
+	libs/bytewriter				\
 	libs/gstnetclientclock			\
 	libs/gstnettimeprovider			\
 	libs/transform1				
@@ -170,6 +171,9 @@ libs_bitreader_LDADD = \
 libs_bytereader_LDADD = \
 	$(top_builddir)/libs/gst/base/libgstbase-@GST_MAJORMINOR@.la \
 	$(LDADD)
+libs_bytewriter_LDADD = \
+	$(top_builddir)/libs/gst/base/libgstbase-@GST_MAJORMINOR@.la \
+	$(LDADD)
 libs_adapter_LDADD = \
 	$(top_builddir)/libs/gst/base/libgstbase-@GST_MAJORMINOR@.la \
 	$(LDADD)
diff --git a/tests/check/libs/bytewriter.c b/tests/check/libs/bytewriter.c
new file mode 100644
index 0000000000..dccdc730dc
--- /dev/null
+++ b/tests/check/libs/bytewriter.c
@@ -0,0 +1,237 @@
+/* GStreamer
+ *
+ * unit test for GstByteWriter
+ *
+ * Copyright (C) <2009> Sebastian Dröge <sebastian.droege@collabora.co.uk>
+ *
+ * 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.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <gst/gst.h>
+#include <gst/check/gstcheck.h>
+#include <gst/base/gstbytewriter.h>
+
+GST_START_TEST (test_initialization)
+{
+  GstByteWriter writer, *writer2;
+  static guint8 sdata[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
+  guint8 *data, *tmp;
+
+  gst_byte_writer_init_with_size (&writer, 24, FALSE);
+  fail_unless_equals_int (gst_byte_writer_get_pos (&writer), 0);
+  fail_unless_equals_int (gst_byte_writer_get_size (&writer), 0);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer), -1);
+  fail_unless_equals_int (gst_byte_reader_get_remaining (GST_BYTE_READER
+          (&writer)), 0);
+  gst_byte_writer_reset (&writer);
+
+  data = g_memdup (sdata, sizeof (sdata));
+  gst_byte_writer_init_with_data (&writer, data, sizeof (sdata), FALSE);
+  fail_unless_equals_int (gst_byte_writer_get_pos (&writer), 0);
+  fail_unless_equals_int (gst_byte_writer_get_size (&writer), 0);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer),
+      sizeof (sdata));
+  fail_unless_equals_int (gst_byte_reader_get_remaining (GST_BYTE_READER
+          (&writer)), 0)
+      tmp = gst_byte_writer_reset_and_get_data (&writer);
+  fail_if (tmp == data);
+  g_free (tmp);
+  g_free (data);
+  data = tmp = NULL;
+
+  data = g_memdup (sdata, sizeof (sdata));
+  gst_byte_writer_init_with_data (&writer, data, sizeof (sdata), TRUE);
+  fail_unless_equals_int (gst_byte_writer_get_pos (&writer), 0);
+  fail_unless_equals_int (gst_byte_writer_get_size (&writer), sizeof (sdata));
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer),
+      sizeof (sdata));
+  fail_unless_equals_int (gst_byte_reader_get_remaining (GST_BYTE_READER
+          (&writer)), sizeof (sdata));
+  tmp = gst_byte_writer_reset_and_get_data (&writer);
+  fail_unless (tmp != data);
+  g_free (tmp);
+  g_free (data);
+  data = tmp = NULL;
+
+  writer2 = gst_byte_writer_new_with_size (24, FALSE);
+  data = gst_byte_writer_free_and_get_data (writer2);
+  fail_unless (data != NULL);
+  g_free (data);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_write_fixed)
+{
+  GstByteWriter writer;
+  guint64 end_data = 0xff34567890abcdef;
+  guint8 *data;
+  guint8 b = 0;
+  guint64 l = 0;
+
+  end_data = GUINT64_TO_BE (end_data);
+
+  gst_byte_writer_init_with_size (&writer, 8, TRUE);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer), 8);
+
+  fail_unless (gst_byte_writer_put_uint8 (&writer, 0x12));
+  fail_unless (gst_byte_writer_put_uint16_be (&writer, 0x3456));
+  fail_unless (gst_byte_writer_put_uint16_le (&writer, 0x9078));
+  fail_unless_equals_int (gst_byte_writer_get_pos (&writer), 5);
+  fail_unless_equals_int (gst_byte_writer_get_size (&writer), 5);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer), 8 - 5);
+  fail_unless_equals_int (gst_byte_reader_get_remaining (GST_BYTE_READER
+          (&writer)), 0);
+  fail_if (gst_byte_reader_get_uint8 (GST_BYTE_READER (&writer), &b));
+  fail_unless (gst_byte_writer_put_uint24_be (&writer, 0xabcdef));
+  fail_unless_equals_int (gst_byte_writer_get_pos (&writer), 8);
+  fail_unless_equals_int (gst_byte_writer_get_size (&writer), 8);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer), 8 - 8);
+  fail_unless_equals_int (gst_byte_reader_get_remaining (GST_BYTE_READER
+          (&writer)), 0);
+  fail_if (gst_byte_writer_put_uint8 (&writer, 0x12));
+  fail_unless (gst_byte_writer_set_pos (&writer, 0));
+  fail_unless (gst_byte_reader_peek_uint64_be (GST_BYTE_READER (&writer), &l));
+  fail_unless_equals_uint64 (l, 0x1234567890abcdef);
+  fail_unless (gst_byte_writer_put_uint8 (&writer, 0xff));
+  fail_unless (gst_byte_writer_set_pos (&writer, 0));
+  fail_unless (gst_byte_reader_get_uint64_be (GST_BYTE_READER (&writer), &l));
+  fail_unless_equals_uint64 (l, 0xff34567890abcdef);
+  fail_if (gst_byte_writer_put_uint64_be (&writer, 0x1234567890abcdef));
+
+  data = gst_byte_writer_reset_and_get_data (&writer);
+  fail_unless (data != NULL);
+  fail_unless (memcmp (&end_data, data, 8) == 0);
+  g_free (data);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_write_non_fixed)
+{
+  GstByteWriter writer;
+  guint64 end_data = 0xff34567890abcdef;
+  guint8 *data;
+  guint64 l = 0;
+
+  end_data = GUINT64_TO_BE (end_data);
+
+  gst_byte_writer_init_with_size (&writer, 6, FALSE);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer), -1);
+
+  fail_unless (gst_byte_writer_put_uint8 (&writer, 0x12));
+  fail_unless (gst_byte_writer_put_uint16_be (&writer, 0x3456));
+  fail_unless (gst_byte_writer_put_uint16_le (&writer, 0x9078));
+  fail_unless_equals_int (gst_byte_writer_get_pos (&writer), 5);
+  fail_unless_equals_int (gst_byte_writer_get_size (&writer), 5);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer), -1);
+  fail_unless_equals_int (gst_byte_reader_get_remaining (GST_BYTE_READER
+          (&writer)), 0);
+  fail_unless (gst_byte_writer_put_uint24_be (&writer, 0xabcdef));
+  fail_unless_equals_int (gst_byte_writer_get_pos (&writer), 8);
+  fail_unless_equals_int (gst_byte_writer_get_size (&writer), 8);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer), -1);
+  fail_unless_equals_int (gst_byte_reader_get_remaining (GST_BYTE_READER
+          (&writer)), 0);
+  fail_unless (gst_byte_writer_set_pos (&writer, 0));
+  fail_unless (gst_byte_reader_peek_uint64_be (GST_BYTE_READER (&writer), &l));
+  fail_unless_equals_uint64 (l, 0x1234567890abcdef);
+  fail_unless (gst_byte_writer_put_uint8 (&writer, 0xff));
+  fail_unless (gst_byte_writer_set_pos (&writer, 0));
+  fail_unless (gst_byte_reader_get_uint64_be (GST_BYTE_READER (&writer), &l));
+  fail_unless_equals_uint64 (l, 0xff34567890abcdef);
+  fail_unless (gst_byte_writer_set_pos (&writer, 8));
+  fail_unless (gst_byte_writer_put_uint64_be (&writer, 0x1234567890abcdef));
+
+  data = gst_byte_writer_reset_and_get_data (&writer);
+  fail_unless (data != NULL);
+  fail_unless (memcmp (&end_data, data, 8) == 0);
+  end_data = GUINT64_TO_BE (0x1234567890abcdef);
+  fail_unless (memcmp (&end_data, data + 8, 8) == 0);
+  g_free (data);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_from_data)
+{
+  GstByteWriter writer;
+  guint8 data[] = { 0x12, 0x34, 0x56, 0x78,
+    0x90, 0xab, 0xcd, 0xef
+  };
+  guint8 *data2;
+
+  gst_byte_writer_init_with_data (&writer, data, sizeof (data), TRUE);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer), 8);
+  fail_unless (gst_byte_writer_put_uint8 (&writer, 0xff));
+  fail_unless_equals_int (data[0], 0xff);
+  fail_unless_equals_int (gst_byte_writer_get_remaining (&writer), 7);
+  fail_unless_equals_int (gst_byte_writer_get_pos (&writer), 1);
+  fail_if (gst_byte_writer_put_uint64_be (&writer, 0x1234567890abcdef));
+  data2 = gst_byte_writer_reset_and_get_data (&writer);
+  fail_unless (data2 != NULL);
+  fail_if (data2 == data);
+  fail_unless (memcmp (data, data2, 8) == 0);
+  g_free (data2);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_put_data_strings)
+{
+  GstByteWriter writer;
+  guint8 data[] = { 0x12, 0x34, 0x56, 0x78,
+    0x90, 0xab, 0xcd, 0xef
+  };
+  guint8 *data2;
+
+  gst_byte_writer_init (&writer);
+  fail_unless (gst_byte_writer_put_data (&writer, data, 8));
+  fail_unless (gst_byte_writer_put_string (&writer, "somerandomteststring"));
+  fail_unless_equals_int (gst_byte_writer_get_pos (&writer),
+      8 + sizeof ("somerandomteststring"));
+
+  data2 = gst_byte_writer_reset_and_get_data (&writer);
+  fail_unless (data2 != NULL);
+  fail_unless (memcmp (data2, data, 8) == 0);
+  g_free (data2);
+}
+
+GST_END_TEST;
+
+static Suite *
+gst_byte_writer_suite (void)
+{
+  Suite *s = suite_create ("GstByteWriter");
+  TCase *tc_chain = tcase_create ("general");
+
+  suite_add_tcase (s, tc_chain);
+
+  tcase_add_test (tc_chain, test_initialization);
+  tcase_add_test (tc_chain, test_write_fixed);
+  tcase_add_test (tc_chain, test_write_non_fixed);
+  tcase_add_test (tc_chain, test_from_data);
+  tcase_add_test (tc_chain, test_put_data_strings);
+
+  return s;
+}
+
+
+GST_CHECK_MAIN (gst_byte_writer);