Add new downloadbuffer element

See https://bugzilla.gnome.org/show_bug.cgi?id=680183

2 files changed, 228 insertions, 0 deletions

diff --git a/tests/check/Makefile.am b/tests/check/Makefile.am
index 6dde80d424..9424d0f7a9 100644
--- a/tests/check/Makefile.am
+++ b/tests/check/Makefile.am
@@ -91,6 +91,7 @@ REGISTRY_CHECKS =				\
 	libs/baseparse				\
 	libs/basesrc				\
 	libs/basesink				\
+	libs/sparsefile				\
 	libs/controller				\
 	libs/queuearray				\
 	libs/typefindhelper			\
@@ -142,6 +143,7 @@ check_PROGRAMS =				\
 	libs/bitreader-noinline		\
 	libs/bytereader-noinline	\
 	libs/bytewriter-noinline	\
+	libs/sparsefile				\
 	libs/collectpads			\
 	libs/gstnetclientclock			\
 	libs/gstnettimeprovider			\
diff --git a/tests/check/libs/sparsefile.c b/tests/check/libs/sparsefile.c
new file mode 100644
index 0000000000..c8cc78c6ad
--- /dev/null
+++ b/tests/check/libs/sparsefile.c
@@ -0,0 +1,226 @@
+/* GStreamer
+ *
+ * unit test for cachefile helper
+ *
+ * Copyright (C) 2014 Wim Taymans  <wtaymans@redhat.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.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <glib/gstdio.h>
+
+#include <gst/check/gstcheck.h>
+
+#include <gst/base/gstsparsefile.h>
+
+static void
+expect_range_before (GstSparseFile * file, gsize offset, gsize start,
+    gsize stop)
+{
+  gsize tstart, tstop;
+
+  fail_unless (gst_sparse_file_get_range_before (file, offset, &tstart,
+          &tstop) == TRUE);
+  fail_unless (tstart == start);
+  fail_unless (tstop == stop);
+}
+
+static void
+expect_range_after (GstSparseFile * file, gsize offset, gsize start, gsize stop)
+{
+  gsize tstart, tstop;
+
+  fail_unless (gst_sparse_file_get_range_after (file, offset, &tstart,
+          &tstop) == TRUE);
+  fail_unless (tstart == start);
+  fail_unless (tstop == stop);
+}
+
+static gboolean
+expect_write (GstSparseFile * file, gsize offset, gsize count, gsize result,
+    gsize avail)
+{
+  GError *error = NULL;
+  gchar buffer[200];
+  gsize res, a;
+
+  res = gst_sparse_file_write (file, offset, buffer, count, &a, &error);
+  if (res != result)
+    return FALSE;
+  if (res == 0) {
+    if (error == NULL)
+      return FALSE;
+    g_clear_error (&error);
+  } else if (a != avail)
+    return FALSE;
+  return TRUE;
+}
+
+static gboolean
+expect_read (GstSparseFile * file, gsize offset, gsize count, gsize result,
+    gsize avail)
+{
+  GError *error = NULL;
+  gchar buffer[200];
+  gsize res, a;
+
+  res = gst_sparse_file_read (file, offset, buffer, count, &a, &error);
+  if (res != result)
+    return FALSE;
+  if (res == 0) {
+    if (error == NULL)
+      return FALSE;
+    g_clear_error (&error);
+  } else if (a != avail)
+    return FALSE;
+  return TRUE;
+}
+
+GST_START_TEST (test_write_read)
+{
+  GstSparseFile *file;
+  gint fd;
+  gchar *name;
+  gsize start, stop;
+
+  name = g_strdup ("cachefile-testXXXXXX");
+  fd = g_mkstemp (name);
+  fail_if (fd == -1);
+
+  file = gst_sparse_file_new ();
+  fail_unless (file != NULL);
+  fail_unless (gst_sparse_file_set_fd (file, fd));
+  fail_unless (gst_sparse_file_n_ranges (file) == 0);
+
+  /* should fail, we didn't write anything yet */
+  fail_unless (expect_read (file, 0, 100, 0, 0));
+
+  /* no ranges, searching for a range should fail */
+  fail_unless (gst_sparse_file_n_ranges (file) == 0);
+  fail_unless (gst_sparse_file_get_range_before (file, 0, &start,
+          &stop) == FALSE);
+  fail_unless (gst_sparse_file_get_range_before (file, 10, &start,
+          &stop) == FALSE);
+  fail_unless (gst_sparse_file_get_range_after (file, 0, &start,
+          &stop) == FALSE);
+  fail_unless (gst_sparse_file_get_range_after (file, 10, &start,
+          &stop) == FALSE);
+
+  /* now write some data */
+  fail_unless (expect_write (file, 0, 100, 100, 0));
+
+  /* we have 1 range now */
+  fail_unless (gst_sparse_file_n_ranges (file) == 1);
+  expect_range_before (file, 0, 0, 100);
+  expect_range_after (file, 0, 0, 100);
+  expect_range_before (file, 100, 0, 100);
+  expect_range_before (file, 50, 0, 100);
+  expect_range_before (file, 200, 0, 100);
+  fail_unless (gst_sparse_file_get_range_after (file, 100, &start,
+          &stop) == FALSE);
+  expect_range_after (file, 50, 0, 100);
+
+  /* we can read all data now */
+  fail_unless (expect_read (file, 0, 100, 100, 0));
+  /* we can read less */
+  fail_unless (expect_read (file, 0, 50, 50, 50));
+  /* but we can't read more than what is written */
+  fail_unless (expect_read (file, 0, 101, 0, 0));
+
+  g_unlink (name);
+  gst_sparse_file_free (file);
+  g_free (name);
+}
+
+GST_END_TEST;
+
+GST_START_TEST (test_write_merge)
+{
+  GstSparseFile *file;
+  gint fd;
+  gchar *name;
+  gsize start, stop;
+
+  name = g_strdup ("cachefile-testXXXXXX");
+  fd = g_mkstemp (name);
+  fail_if (fd == -1);
+
+  file = gst_sparse_file_new ();
+  gst_sparse_file_set_fd (file, fd);
+
+  /* write something at offset 0 */
+  fail_unless (expect_write (file, 0, 100, 100, 0));
+  /* we have 1 range now */
+  fail_unless (gst_sparse_file_n_ranges (file) == 1);
+  expect_range_before (file, 110, 0, 100);
+  expect_range_after (file, 50, 0, 100);
+  fail_unless (gst_sparse_file_get_range_after (file, 100, &start,
+          &stop) == FALSE);
+
+  /* read should fail */
+  fail_unless (expect_read (file, 50, 150, 0, 0));
+
+  /* write something at offset 150 */
+  fail_unless (expect_write (file, 150, 100, 100, 0));
+  /* we have 2 ranges now */
+  fail_unless (gst_sparse_file_n_ranges (file) == 2);
+  expect_range_before (file, 110, 0, 100);
+  expect_range_after (file, 50, 0, 100);
+  expect_range_after (file, 100, 150, 250);
+  expect_range_before (file, 150, 150, 250);
+
+  /* read should still fail */
+  fail_unless (expect_read (file, 50, 150, 0, 0));
+
+  /* fill the hole */
+  fail_unless (expect_write (file, 100, 50, 50, 100));
+  /* we have 1 range now */
+  fail_unless (gst_sparse_file_n_ranges (file) == 1);
+  expect_range_before (file, 110, 0, 250);
+  expect_range_after (file, 50, 0, 250);
+  expect_range_after (file, 100, 0, 250);
+  expect_range_before (file, 150, 0, 250);
+  fail_unless (gst_sparse_file_get_range_after (file, 250, &start,
+          &stop) == FALSE);
+
+  /* read work */
+  fail_unless (expect_read (file, 50, 150, 150, 50));
+
+  g_unlink (name);
+  gst_sparse_file_free (file);
+  g_free (name);
+}
+
+GST_END_TEST;
+
+static Suite *
+gst_cachefile_suite (void)
+{
+  Suite *s = suite_create ("cachefile");
+  TCase *tc_chain = tcase_create ("general");
+
+  suite_add_tcase (s, tc_chain);
+  tcase_add_test (tc_chain, test_write_read);
+  tcase_add_test (tc_chain, test_write_merge);
+
+  return s;
+}
+
+GST_CHECK_MAIN (gst_cachefile);