lib/ntdb/test/run-expand-in-transaction.c

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