tests/dconf-mock-shm.c

   1 #include "../shm/dconf-shm.h"
   2
   3 #include "dconf-mock.h"
   4
   5 typedef struct
   6 {
   7   guint8 flagged;
   8   gint   ref_count;
   9 } DConfMockShm;
  10
  11 static GHashTable *dconf_mock_shm_table;
  12 static GMutex      dconf_mock_shm_lock;
  13 static GString    *dconf_mock_shm_log;
  14
  15 static void
  16 dconf_mock_shm_unref (gpointer data)
  17 {
  18   DConfMockShm *shm = data;
  19
  20   if (g_atomic_int_dec_and_test (&shm->ref_count))
  21     g_slice_free (DConfMockShm, shm);
  22 }
  23
  24 static DConfMockShm *
  25 dconf_mock_shm_ref (DConfMockShm *shm)
  26 {
  27   g_atomic_int_inc (&shm->ref_count);
  28
  29   return shm;
  30 }
  31
  32 guint8 *
  33 dconf_shm_open (const gchar *name)
  34 {
  35   DConfMockShm *shm;
  36
  37   g_mutex_lock (&dconf_mock_shm_lock);
  38
  39   if G_UNLIKELY (dconf_mock_shm_table == NULL)
  40     {
  41       dconf_mock_shm_table = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, dconf_mock_shm_unref);
  42       dconf_mock_shm_log = g_string_new (NULL);
  43     }
  44
  45   shm = g_hash_table_lookup (dconf_mock_shm_table, name);
  46   if (shm == NULL)
  47     {
  48       shm = g_slice_new0 (DConfMockShm);
  49       g_hash_table_insert (dconf_mock_shm_table, g_strdup (name), dconf_mock_shm_ref (shm));
  50     }
  51
  52   /* before unlocking... */
  53   dconf_mock_shm_ref (shm);
  54
  55   g_string_append_printf (dconf_mock_shm_log, "open %s;", name);
  56
  57   g_mutex_unlock (&dconf_mock_shm_lock);
  58
  59   return &shm->flagged;
  60 }
  61
  62 void
  63 dconf_shm_close (guint8 *shm)
  64 {
  65   if (shm)
  66     {
  67       g_mutex_lock (&dconf_mock_shm_lock);
  68       g_string_append (dconf_mock_shm_log, "close;");
  69       g_mutex_unlock (&dconf_mock_shm_lock);
  70
  71       dconf_mock_shm_unref (shm);
  72     }
  73 }
  74
  75 gint
  76 dconf_mock_shm_flag (const gchar *name)
  77 {
  78   DConfMockShm *shm;
  79   gint count = 0;
  80
  81   g_mutex_lock (&dconf_mock_shm_lock);
  82   shm = g_hash_table_lookup (dconf_mock_shm_table, name);
  83   if (shm)
  84     {
  85       shm->flagged = 1;
  86       count = shm->ref_count;
  87       g_hash_table_remove (dconf_mock_shm_table, name);
  88     }
  89   g_mutex_unlock (&dconf_mock_shm_lock);
  90
  91   return count;
  92 }
  93
  94 void
  95 dconf_mock_shm_reset (void)
  96 {
  97   g_mutex_lock (&dconf_mock_shm_lock);
  98   if (dconf_mock_shm_table != NULL)
  99     {
 100       GHashTableIter iter;
 101       gpointer value;
 102
 103       g_hash_table_iter_init (&iter, dconf_mock_shm_table);
 104       while (g_hash_table_iter_next (&iter, NULL, &value))
 105         {
 106           DConfMockShm *shm = value;
 107
 108           g_assert_cmpint (shm->ref_count, ==, 1);
 109           g_hash_table_iter_remove (&iter);
 110         }
 111
 112       g_string_truncate (dconf_mock_shm_log, 0);
 113     }
 114   g_mutex_unlock (&dconf_mock_shm_lock);
 115 }
 116
 117 void
 118 dconf_mock_shm_assert_log (const gchar *expected_log)
 119 {
 120   g_mutex_lock (&dconf_mock_shm_lock);
 121   g_assert_cmpstr (dconf_mock_shm_log->str, ==, expected_log);
 122   g_string_truncate (dconf_mock_shm_log, 0);
 123   g_mutex_unlock (&dconf_mock_shm_lock);
 124 }