tests/check/gst/gsttagsetter.c

   1 /* GStreamer GstTagSetter interface unit tests
   2  * Copyright (C) 2007 Tim-Philipp Müller <tim centricular net>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19
  20 #include <gst/check/gstcheck.h>
  21 #include <gst/gst.h>
  22 #include <string.h>
  23
  24 /* some minimal GstTagSetter object */
  25 #define GST_TYPE_DUMMY_ENC gst_dummy_enc_get_type()
  26
  27 typedef GstElement GstDummyEnc;
  28 typedef GstElementClass GstDummyEncClass;
  29
  30 GType gst_dummy_enc_get_type (void);
  31 G_DEFINE_TYPE_WITH_CODE (GstDummyEnc, gst_dummy_enc,
  32     GST_TYPE_ELEMENT, G_IMPLEMENT_INTERFACE (GST_TYPE_TAG_SETTER, NULL));
  33
  34 static void
  35 gst_dummy_enc_class_init (GstDummyEncClass * klass)
  36 {
  37 }
  38
  39 static void
  40 gst_dummy_enc_init (GstDummyEnc * enc)
  41 {
  42 }
  43
  44 static void
  45 tag_list_foreach (const GstTagList * taglist, const gchar * tag, guint * p_num)
  46 {
  47   guint tag_size;
  48
  49   tag_size = gst_tag_list_get_tag_size (taglist, tag);
  50   GST_LOG ("%u+%u tag = %s", *p_num, tag_size, tag);
  51   *p_num += tag_size;
  52 }
  53
  54 static guint
  55 tag_setter_list_length (GstTagSetter * setter)
  56 {
  57   guint len = 0;
  58
  59   if (gst_tag_setter_get_tag_list (setter) == NULL)
  60     return 0;
  61
  62   gst_tag_list_foreach (gst_tag_setter_get_tag_list (setter),
  63       (GstTagForeachFunc) tag_list_foreach, &len);
  64   return len;
  65 }
  66
  67 static guint
  68 tag_list_length (const GstTagList * tag_list)
  69 {
  70   guint len = 0;
  71
  72   if (tag_list == NULL)
  73     return 0;
  74
  75   gst_tag_list_foreach (tag_list, (GstTagForeachFunc) tag_list_foreach, &len);
  76   return len;
  77 }
  78
  79 #define assert_tag_setter_list_length(setter,len) \
  80     fail_unless_equals_int (tag_setter_list_length(setter), len);
  81
  82 GST_START_TEST (test_merge)
  83 {
  84   GstTagSetter *setter;
  85   GstTagList *list1, *list2;
  86   GstElement *enc;
  87
  88   enc = g_object_new (GST_TYPE_DUMMY_ENC, NULL);
  89   fail_unless (enc != NULL);
  90
  91   setter = GST_TAG_SETTER (enc);
  92
  93   list1 = gst_tag_list_new_empty ();
  94   gst_tag_list_add (list1, GST_TAG_MERGE_APPEND, GST_TAG_ARTIST, "artist1",
  95       NULL);
  96   gst_tag_setter_merge_tags (setter, list1, GST_TAG_MERGE_APPEND);
  97   assert_tag_setter_list_length (setter, 1);
  98
  99   list2 = gst_tag_list_new_empty ();
 100   gst_tag_list_add (list2, GST_TAG_MERGE_APPEND, GST_TAG_ARTIST, "artist2",
 101       GST_TAG_TITLE, "title1", NULL);
 102   gst_tag_setter_merge_tags (setter, list2, GST_TAG_MERGE_APPEND);
 103   assert_tag_setter_list_length (setter, 3);
 104
 105   gst_tag_setter_merge_tags (setter, list2, GST_TAG_MERGE_REPLACE_ALL);
 106   assert_tag_setter_list_length (setter, 2);
 107
 108   gst_tag_setter_merge_tags (setter, list1, GST_TAG_MERGE_REPLACE_ALL);
 109   assert_tag_setter_list_length (setter, 1);
 110
 111   gst_tag_setter_add_tags (setter, GST_TAG_MERGE_APPEND, GST_TAG_ALBUM, "xyz",
 112       NULL);
 113   assert_tag_setter_list_length (setter, 2);
 114
 115   gst_tag_list_free (list2);
 116   gst_tag_list_free (list1);
 117
 118   g_object_unref (enc);
 119 }
 120
 121 GST_END_TEST
 122 GST_START_TEST (test_merge_modes)
 123 {
 124   GstTagMergeMode mode;
 125
 126   for (mode = GST_TAG_MERGE_REPLACE_ALL; mode < GST_TAG_MERGE_COUNT; mode++) {
 127     gint i;
 128
 129     for (i = 0; i < 4; i++) {
 130       GstElement *enc;
 131       GstTagSetter *setter;
 132       GstTagList *list1, *list2, *merged;
 133
 134       enc = g_object_new (GST_TYPE_DUMMY_ENC, NULL);
 135       fail_unless (enc != NULL);
 136
 137       setter = GST_TAG_SETTER (enc);
 138       list1 = gst_tag_list_new_empty ();
 139       list2 = gst_tag_list_new_empty ();
 140
 141       /* i = 0: -     -
 142        * i = 1: list1 -
 143        * i = 2: -     list2
 144        * i = 3: list1 list2 */
 145
 146       if (i % 2 == 1) {
 147         gst_tag_list_add (list1, GST_TAG_MERGE_APPEND, GST_TAG_ARTIST,
 148             "artist1", NULL);
 149       }
 150       if (i > 1) {
 151         gst_tag_list_add (list2, GST_TAG_MERGE_APPEND, GST_TAG_ARTIST,
 152             "artist2", NULL);
 153       }
 154
 155       gst_tag_setter_merge_tags (setter, list1, GST_TAG_MERGE_APPEND);
 156       gst_tag_setter_merge_tags (setter, list2, mode);
 157
 158       merged = gst_tag_list_merge (list1, list2, mode);
 159
 160       fail_unless_equals_int (tag_list_length (gst_tag_setter_get_tag_list
 161               (setter)), tag_list_length (merged));
 162
 163       gst_tag_list_free (list1);
 164       gst_tag_list_free (list2);
 165       gst_tag_list_free (merged);
 166       gst_object_unref (enc);
 167     }
 168   }
 169 }
 170
 171 GST_END_TEST
 172 GST_START_TEST (test_merge_modes_skip_empty)
 173 {
 174   GstTagMergeMode mode;
 175
 176   for (mode = GST_TAG_MERGE_REPLACE_ALL; mode < GST_TAG_MERGE_COUNT; mode++) {
 177     gint i;
 178
 179     for (i = 0; i < 2; i++) {
 180       GstElement *enc;
 181       GstTagSetter *setter;
 182       GstTagList *list1, *list2, *merged;
 183
 184       enc = g_object_new (GST_TYPE_DUMMY_ENC, NULL);
 185       fail_unless (enc != NULL);
 186
 187       setter = GST_TAG_SETTER (enc);
 188       list1 = gst_tag_list_new_empty ();
 189       list2 = gst_tag_list_new_empty ();
 190
 191       if (i == 1) {
 192         gst_tag_list_add (list2, GST_TAG_MERGE_APPEND, GST_TAG_ARTIST,
 193             "artist2", NULL);
 194       }
 195
 196       gst_tag_setter_merge_tags (setter, list2, mode);
 197
 198       merged = gst_tag_list_merge (list1, list2, mode);
 199
 200       fail_unless_equals_int (tag_list_length (gst_tag_setter_get_tag_list
 201               (setter)), tag_list_length (merged));
 202
 203       gst_tag_list_free (list1);
 204       gst_tag_list_free (list2);
 205       gst_tag_list_free (merged);
 206       gst_object_unref (enc);
 207     }
 208   }
 209 }
 210
 211 GST_END_TEST static int spin_and_wait = 1;
 212 static int threads_running = 0;
 213
 214 #define THREADS_TEST_SECONDS 1.5
 215
 216 static gpointer
 217 test_threads_thread_func1 (gpointer data)
 218 {
 219   GstTagSetter *setter = GST_TAG_SETTER (data);
 220   GTimer *timer;
 221
 222   timer = g_timer_new ();
 223
 224   g_atomic_int_inc (&threads_running);
 225   while (g_atomic_int_get (&spin_and_wait))
 226     g_usleep (0);
 227
 228   GST_INFO ("Go!");
 229   g_timer_start (timer);
 230
 231   while (g_timer_elapsed (timer, NULL) < THREADS_TEST_SECONDS) {
 232     gst_tag_setter_add_tags (setter, GST_TAG_MERGE_APPEND, GST_TAG_ARTIST,
 233         "some artist", GST_TAG_TITLE, "some title", GST_TAG_TRACK_NUMBER, 6,
 234         NULL);
 235   }
 236
 237   g_timer_destroy (timer);
 238   GST_INFO ("Done");
 239
 240   return NULL;
 241 }
 242
 243 static gpointer
 244 test_threads_thread_func2 (gpointer data)
 245 {
 246   GstTagSetter *setter = GST_TAG_SETTER (data);
 247   GTimer *timer;
 248
 249   timer = g_timer_new ();
 250
 251   g_atomic_int_inc (&threads_running);
 252   while (g_atomic_int_get (&spin_and_wait))
 253     g_usleep (0);
 254
 255   GST_INFO ("Go!");
 256   g_timer_start (timer);
 257
 258   while (g_timer_elapsed (timer, NULL) < THREADS_TEST_SECONDS) {
 259     gst_tag_setter_add_tags (setter, GST_TAG_MERGE_PREPEND, GST_TAG_CODEC,
 260         "MP42", GST_TAG_COMMENT, "deep insights go here", GST_TAG_TRACK_COUNT,
 261         10, NULL);
 262   }
 263
 264   g_timer_destroy (timer);
 265   GST_INFO ("Done");
 266
 267   return NULL;
 268 }
 269
 270 static gpointer
 271 test_threads_thread_func3 (gpointer data)
 272 {
 273   GstTagSetter *setter = GST_TAG_SETTER (data);
 274   GTimer *timer;
 275
 276   timer = g_timer_new ();
 277
 278   g_atomic_int_inc (&threads_running);
 279   while (g_atomic_int_get (&spin_and_wait))
 280     g_usleep (0);
 281
 282   GST_INFO ("Go!");
 283   g_timer_start (timer);
 284
 285   while (g_timer_elapsed (timer, NULL) < THREADS_TEST_SECONDS) {
 286     gst_tag_setter_reset_tags (setter);
 287   }
 288
 289   g_timer_destroy (timer);
 290   GST_INFO ("Done");
 291
 292   return NULL;
 293 }
 294
 295 GST_START_TEST (test_threads)
 296 {
 297   GstTagSetter *setter;
 298   GThread *threads[3];
 299
 300   setter = GST_TAG_SETTER (g_object_new (GST_TYPE_DUMMY_ENC, NULL));
 301
 302   spin_and_wait = TRUE;
 303   threads[0] = g_thread_create (test_threads_thread_func1, setter, TRUE, NULL);
 304   threads[1] = g_thread_create (test_threads_thread_func2, setter, TRUE, NULL);
 305   threads[2] = g_thread_create (test_threads_thread_func3, setter, TRUE, NULL);
 306
 307   while (g_atomic_int_get (&threads_running) < 3)
 308     g_usleep (10);
 309
 310   g_atomic_int_set (&spin_and_wait, FALSE);
 311
 312   g_thread_join (threads[0]);
 313   g_thread_join (threads[1]);
 314   g_thread_join (threads[2]);
 315
 316   g_object_unref (G_OBJECT (setter));
 317 }
 318
 319 GST_END_TEST static Suite *
 320 gst_tag_setter_suite (void)
 321 {
 322   Suite *s = suite_create ("GstTagSetter");
 323   TCase *tc_chain = tcase_create ("general");
 324
 325   suite_add_tcase (s, tc_chain);
 326   tcase_add_test (tc_chain, test_merge);
 327   tcase_add_test (tc_chain, test_merge_modes);
 328   tcase_add_test (tc_chain, test_merge_modes_skip_empty);
 329   tcase_add_test (tc_chain, test_threads);
 330
 331   return s;
 332 }
 333
 334 GST_CHECK_MAIN (gst_tag_setter);