glib/tests/array-test.c

   1 /* GLIB - Library of useful routines for C programming
   2  * Copyright (C) 1995-1997  Peter Mattis, Spencer Kimball and Josh MacDonald
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser 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  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser 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 /*
  21  * Modified by the GLib Team and others 1997-2000.  See the AUTHORS
  22  * file for a list of people on the GLib Team.  See the ChangeLog
  23  * files for a list of changes.  These files are distributed with
  24  * GLib at ftp://ftp.gtk.org/pub/gtk/.
  25  */
  26
  27 #undef G_DISABLE_ASSERT
  28 #undef G_LOG_DOMAIN
  29
  30 #include <stdio.h>
  31 #include <stdlib.h>
  32 #include <string.h>
  33 #include "glib.h"
  34
  35 static void
  36 sum_up (gpointer data,
  37         gpointer user_data)
  38 {
  39   gint *sum = (gint *)user_data;
  40
  41   *sum += GPOINTER_TO_INT (data);
  42 }
  43
  44 static void
  45 array_append (void)
  46 {
  47   GArray *garray;
  48   gint i;
  49
  50   garray = g_array_new (FALSE, FALSE, sizeof (gint));
  51   for (i = 0; i < 10000; i++)
  52     g_array_append_val (garray, i);
  53
  54   for (i = 0; i < 10000; i++)
  55     g_assert_cmpint (g_array_index (garray, gint, i), ==, i);
  56
  57   g_array_free (garray, TRUE);
  58 }
  59
  60 static void
  61 array_prepend (void)
  62 {
  63   GArray *garray;
  64   gint i;
  65
  66   garray = g_array_new (FALSE, FALSE, sizeof (gint));
  67   for (i = 0; i < 100; i++)
  68     g_array_prepend_val (garray, i);
  69
  70   for (i = 0; i < 100; i++)
  71     g_assert_cmpint (g_array_index (garray, gint, i), ==, (100 - i - 1));
  72
  73   g_array_free (garray, TRUE);
  74 }
  75
  76 static void
  77 array_ref_count (void)
  78 {
  79   GArray *garray;
  80   GArray *garray2;
  81   gint i;
  82
  83   garray = g_array_new (FALSE, FALSE, sizeof (gint));
  84   g_assert_cmpint (g_array_get_element_size (garray), ==, sizeof (gint));
  85   for (i = 0; i < 100; i++)
  86     g_array_prepend_val (garray, i);
  87
  88   /* check we can ref, unref and still access the array */
  89   garray2 = g_array_ref (garray);
  90   g_assert (garray == garray2);
  91   g_array_unref (garray2);
  92   for (i = 0; i < 100; i++)
  93     g_assert_cmpint (g_array_index (garray, gint, i), ==, (100 - i - 1));
  94
  95   /* garray2 should be an empty valid GArray wrapper */
  96   garray2 = g_array_ref (garray);
  97   g_array_free (garray, TRUE);
  98
  99   g_assert_cmpint (garray2->len, ==, 0);
 100   g_array_unref (garray2);
 101 }
 102
 103 static void
 104 array_large_size (void)
 105 {
 106   GArray* array;
 107
 108   g_test_bug ("568760");
 109
 110   array = g_array_new (TRUE, TRUE, sizeof (char));
 111
 112   /* it might take really long until the allocation happens */
 113   if (g_test_trap_fork (10 /* s */ * 1000 /* ms */ * 1000 /* µs */, 0))
 114     {
 115       g_array_set_size (array, 1073750016);
 116       exit (0); /* success */
 117     }
 118
 119   if (!g_test_trap_has_passed ())
 120     {
 121       g_test_trap_assert_stderr ("*failed to allocate 2147483648 bytes*");
 122     }
 123
 124   g_array_free (array, TRUE);
 125 }
 126
 127 static void
 128 pointer_array_add (void)
 129 {
 130   GPtrArray *gparray;
 131   gint i;
 132   gint sum = 0;
 133
 134   gparray = g_ptr_array_new ();
 135   for (i = 0; i < 10000; i++)
 136     g_ptr_array_add (gparray, GINT_TO_POINTER (i));
 137
 138   for (i = 0; i < 10000; i++)
 139     g_assert (g_ptr_array_index (gparray, i) == GINT_TO_POINTER (i));
 140
 141   g_ptr_array_foreach (gparray, sum_up, &sum);
 142   g_assert (sum == 49995000);
 143
 144   g_ptr_array_free (gparray, TRUE);
 145 }
 146
 147 static void
 148 pointer_array_ref_count (void)
 149 {
 150   GPtrArray *gparray;
 151   GPtrArray *gparray2;
 152   gint i;
 153   gint sum = 0;
 154
 155   gparray = g_ptr_array_new ();
 156   for (i = 0; i < 10000; i++)
 157     g_ptr_array_add (gparray, GINT_TO_POINTER (i));
 158
 159   /* check we can ref, unref and still access the array */
 160   gparray2 = g_ptr_array_ref (gparray);
 161   g_assert (gparray == gparray2);
 162   g_ptr_array_unref (gparray2);
 163   for (i = 0; i < 10000; i++)
 164     g_assert (g_ptr_array_index (gparray, i) == GINT_TO_POINTER (i));
 165
 166   g_ptr_array_foreach (gparray, sum_up, &sum);
 167   g_assert (sum == 49995000);
 168
 169   /* gparray2 should be an empty valid GPtrArray wrapper */
 170   gparray2 = g_ptr_array_ref (gparray);
 171   g_ptr_array_free (gparray, TRUE);
 172
 173   g_assert_cmpint (gparray2->len, ==, 0);
 174   g_ptr_array_unref (gparray2);
 175 }
 176
 177 static gint num_free_func_invocations = 0;
 178
 179 static void
 180 my_free_func (gpointer data)
 181 {
 182   num_free_func_invocations++;
 183   g_free (data);
 184 }
 185
 186 static void
 187 pointer_array_free_func (void)
 188 {
 189   GPtrArray *gparray;
 190   GPtrArray *gparray2;
 191   gchar **strv;
 192   gchar *s;
 193
 194   num_free_func_invocations = 0;
 195   gparray = g_ptr_array_new_with_free_func (my_free_func);
 196   g_ptr_array_unref (gparray);
 197   g_assert_cmpint (num_free_func_invocations, ==, 0);
 198
 199   gparray = g_ptr_array_new_with_free_func (my_free_func);
 200   g_ptr_array_free (gparray, TRUE);
 201   g_assert_cmpint (num_free_func_invocations, ==, 0);
 202
 203   num_free_func_invocations = 0;
 204   gparray = g_ptr_array_new_with_free_func (my_free_func);
 205   g_ptr_array_add (gparray, g_strdup ("foo"));
 206   g_ptr_array_add (gparray, g_strdup ("bar"));
 207   g_ptr_array_add (gparray, g_strdup ("baz"));
 208   g_ptr_array_remove_index (gparray, 0);
 209   g_assert_cmpint (num_free_func_invocations, ==, 1);
 210   s = g_strdup ("frob");
 211   g_ptr_array_add (gparray, s);
 212   g_assert (g_ptr_array_remove (gparray, s));
 213   g_assert_cmpint (num_free_func_invocations, ==, 2);
 214   g_ptr_array_set_size (gparray, 1);
 215   g_assert_cmpint (num_free_func_invocations, ==, 3);
 216   g_ptr_array_ref (gparray);
 217   g_ptr_array_unref (gparray);
 218   g_assert_cmpint (num_free_func_invocations, ==, 3);
 219   g_ptr_array_unref (gparray);
 220   g_assert_cmpint (num_free_func_invocations, ==, 4);
 221
 222   num_free_func_invocations = 0;
 223   gparray = g_ptr_array_new_with_free_func (my_free_func);
 224   g_ptr_array_add (gparray, g_strdup ("foo"));
 225   g_ptr_array_add (gparray, g_strdup ("bar"));
 226   g_ptr_array_add (gparray, g_strdup ("baz"));
 227   g_ptr_array_add (gparray, NULL);
 228   gparray2 = g_ptr_array_ref (gparray);
 229   strv = (gchar **) g_ptr_array_free (gparray, FALSE);
 230   g_assert_cmpint (num_free_func_invocations, ==, 0);
 231   g_strfreev (strv);
 232   g_ptr_array_unref (gparray2);
 233   g_assert_cmpint (num_free_func_invocations, ==, 0);
 234
 235   num_free_func_invocations = 0;
 236   gparray = g_ptr_array_new_with_free_func (my_free_func);
 237   g_ptr_array_add (gparray, g_strdup ("foo"));
 238   g_ptr_array_add (gparray, g_strdup ("bar"));
 239   g_ptr_array_add (gparray, g_strdup ("baz"));
 240   g_ptr_array_unref (gparray);
 241   g_assert_cmpint (num_free_func_invocations, ==, 3);
 242
 243   num_free_func_invocations = 0;
 244   gparray = g_ptr_array_new_with_free_func (my_free_func);
 245   g_ptr_array_add (gparray, g_strdup ("foo"));
 246   g_ptr_array_add (gparray, g_strdup ("bar"));
 247   g_ptr_array_add (gparray, g_strdup ("baz"));
 248   g_ptr_array_free (gparray, TRUE);
 249   g_assert_cmpint (num_free_func_invocations, ==, 3);
 250
 251   num_free_func_invocations = 0;
 252   gparray = g_ptr_array_new_with_free_func (my_free_func);
 253   g_ptr_array_add (gparray, "foo");
 254   g_ptr_array_add (gparray, "bar");
 255   g_ptr_array_add (gparray, "baz");
 256   g_ptr_array_set_free_func (gparray, NULL);
 257   g_ptr_array_free (gparray, TRUE);
 258   g_assert_cmpint (num_free_func_invocations, ==, 0);
 259 }
 260
 261 static void
 262 byte_array_append (void)
 263 {
 264   GByteArray *gbarray;
 265   gint i;
 266
 267   gbarray = g_byte_array_new ();
 268   for (i = 0; i < 10000; i++)
 269     g_byte_array_append (gbarray, (guint8*) "abcd", 4);
 270
 271   for (i = 0; i < 10000; i++)
 272     {
 273       g_assert (gbarray->data[4*i] == 'a');
 274       g_assert (gbarray->data[4*i+1] == 'b');
 275       g_assert (gbarray->data[4*i+2] == 'c');
 276       g_assert (gbarray->data[4*i+3] == 'd');
 277     }
 278
 279   g_byte_array_free (gbarray, TRUE);
 280 }
 281
 282 static void
 283 byte_array_ref_count (void)
 284 {
 285   GByteArray *gbarray;
 286   GByteArray *gbarray2;
 287   gint i;
 288
 289   gbarray = g_byte_array_new ();
 290   for (i = 0; i < 10000; i++)
 291     g_byte_array_append (gbarray, (guint8*) "abcd", 4);
 292
 293   gbarray2 = g_byte_array_ref (gbarray);
 294   g_assert (gbarray2 == gbarray);
 295   g_byte_array_unref (gbarray2);
 296   for (i = 0; i < 10000; i++)
 297     {
 298       g_assert (gbarray->data[4*i] == 'a');
 299       g_assert (gbarray->data[4*i+1] == 'b');
 300       g_assert (gbarray->data[4*i+2] == 'c');
 301       g_assert (gbarray->data[4*i+3] == 'd');
 302     }
 303
 304   gbarray2 = g_byte_array_ref (gbarray);
 305   g_assert (gbarray2 == gbarray);
 306   g_byte_array_free (gbarray, TRUE);
 307   g_assert_cmpint (gbarray2->len, ==, 0);
 308   g_byte_array_unref (gbarray2);
 309 }
 310
 311 int
 312 main (int argc, char *argv[])
 313 {
 314   g_test_init (&argc, &argv, NULL);
 315
 316   g_test_bug_base ("http://bugs.gnome.org/%s");
 317
 318   /* array tests */
 319   g_test_add_func ("/array/append", array_append);
 320   g_test_add_func ("/array/prepend", array_prepend);
 321   g_test_add_func ("/array/ref-count", array_ref_count);
 322   g_test_add_func ("/array/large-size", array_large_size);
 323
 324   /* pointer arrays */
 325   g_test_add_func ("/pointerarray/add", pointer_array_add);
 326   g_test_add_func ("/pointerarray/ref-count", pointer_array_ref_count);
 327   g_test_add_func ("/pointerarray/free-func", pointer_array_free_func);
 328
 329   /* byte arrays */
 330   g_test_add_func ("/bytearray/append", byte_array_append);
 331   g_test_add_func ("/bytearray/ref-count", byte_array_ref_count);
 332
 333   return g_test_run ();
 334 }
 335