lib/tdb2/test/run-tdb1-traverse-in-transaction.c

   1 #include "config.h"
   2 #include "tdb1-lock-tracking.h"
   3 #define fcntl fcntl_with_lockcheck1
   4 #include "tdb2-source.h"
   5 #include "tap-interface.h"
   6 #undef fcntl_with_lockcheck
   7 #include <stdlib.h>
   8 #include <stdbool.h>
   9 #include "tdb1-external-agent.h"
  10 #include "logging.h"
  11
  12 static struct agent *agent;
  13
  14 static bool correct_key(TDB_DATA key)
  15 {
  16         return key.dsize == strlen("hi")
  17                 && memcmp(key.dptr, "hi", key.dsize) == 0;
  18 }
  19
  20 static bool correct_data(TDB_DATA data)
  21 {
  22         return data.dsize == strlen("world")
  23                 && memcmp(data.dptr, "world", data.dsize) == 0;
  24 }
  25
  26 static int traverse(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data,
  27                      void *p)
  28 {
  29         ok1(correct_key(key));
  30         ok1(correct_data(data));
  31         return 0;
  32 }
  33
  34 int main(int argc, char *argv[])
  35 {
  36         struct tdb_context *tdb;
  37         TDB_DATA key, data;
  38         union tdb_attribute hsize;
  39
  40         hsize.base.attr = TDB_ATTRIBUTE_TDB1_HASHSIZE;
  41         hsize.base.next = &tap_log_attr;
  42         hsize.tdb1_hashsize.hsize = 1024;
  43
  44         plan_tests(13);
  45         agent = prepare_external_agent1();
  46         if (!agent)
  47                 err(1, "preparing agent");
  48
  49         tdb = tdb_open("run-traverse-in-transaction.tdb1",
  50                        TDB_VERSION1, O_CREAT|O_TRUNC|O_RDWR,
  51                        0600, &hsize);
  52         ok1(tdb);
  53
  54         key = tdb_mkdata("hi", strlen("hi"));
  55         data = tdb_mkdata("world", strlen("world"));
  56
  57         ok1(tdb_store(tdb, key, data, TDB_INSERT) == TDB_SUCCESS);
  58
  59         ok1(external_agent_operation1(agent, OPEN, tdb->name) == SUCCESS);
  60
  61         ok1(tdb_transaction_start(tdb) == TDB_SUCCESS);
  62         ok1(external_agent_operation1(agent, TRANSACTION_START, tdb->name)
  63             == WOULD_HAVE_BLOCKED);
  64         tdb_traverse(tdb, traverse, NULL);
  65
  66         /* That should *not* release the transaction lock! */
  67         ok1(external_agent_operation1(agent, TRANSACTION_START, tdb->name)
  68             == WOULD_HAVE_BLOCKED);
  69         tdb_traverse(tdb, traverse, NULL);
  70
  71         /* That should *not* release the transaction lock! */
  72         ok1(external_agent_operation1(agent, TRANSACTION_START, tdb->name)
  73             == WOULD_HAVE_BLOCKED);
  74         ok1(tdb_transaction_commit(tdb) == TDB_SUCCESS);
  75         /* Now we should be fine. */
  76         ok1(external_agent_operation1(agent, TRANSACTION_START, tdb->name)
  77             == SUCCESS);
  78
  79         tdb_close(tdb);
  80
  81         return exit_status();
  82 }