git.ozlabs.org Git - ccan/blob - ccan/ntdb/test/run-expand-in-transaction.c

   1 #include "ntdb-source.h"
   2 #include "tap-interface.h"
   3 #include "logging.h"
   4 #include "helprun-external-agent.h"
   5
   6 int main(int argc, char *argv[])
   7 {
   8         unsigned int i;
   9         struct ntdb_context *ntdb;
  10         int flags[] = { NTDB_DEFAULT, NTDB_NOMMAP,
  11                         NTDB_CONVERT, NTDB_NOMMAP|NTDB_CONVERT };
  12         NTDB_DATA key = ntdb_mkdata("key", 3);
  13         NTDB_DATA data = ntdb_mkdata("data", 4);
  14
  15         plan_tests(sizeof(flags) / sizeof(flags[0]) * 9 + 1);
  16
  17         for (i = 0; i < sizeof(flags) / sizeof(flags[0]); i++) {
  18                 size_t size;
  19                 NTDB_DATA k, d;
  20                 ntdb = ntdb_open("run-expand-in-transaction.ntdb",
  21                                  flags[i]|MAYBE_NOSYNC,
  22                                  O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr);
  23                 ok1(ntdb);
  24                 if (!ntdb)
  25                         continue;
  26
  27                 size = ntdb->file->map_size;
  28                 /* Add a fake record to chew up the existing free space. */
  29                 k = ntdb_mkdata("fake", 4);
  30                 d.dsize = ntdb->file->map_size
  31                         - NEW_DATABASE_HDR_SIZE(ntdb->hash_bits) - 8;
  32                 d.dptr = malloc(d.dsize);
  33                 memset(d.dptr, 0, d.dsize);
  34                 ok1(ntdb_store(ntdb, k, d, NTDB_INSERT) == 0);
  35                 ok1(ntdb->file->map_size == size);
  36                 free(d.dptr);
  37                 ok1(ntdb_transaction_start(ntdb) == 0);
  38                 ok1(ntdb_store(ntdb, key, data, NTDB_INSERT) == 0);
  39                 ok1(ntdb->file->map_size > size);
  40                 ok1(ntdb_transaction_commit(ntdb) == 0);
  41                 ok1(ntdb->file->map_size > size);
  42                 ok1(ntdb_check(ntdb, NULL, NULL) == 0);
  43                 ntdb_close(ntdb);
  44         }
  45
  46         ok1(tap_log_messages == 0);
  47         return exit_status();
  48 }