htable: fix type of htable_type _pick return.
[ccan] / junkcode / rusty@rustcorp.com.au-ntdb / test / api-21-parse_record.c
1 #include "config.h"
2 #include "../ntdb.h"
3 #include "../private.h"
4 #include "tap-interface.h"
5 #include "logging.h"
6 #include "helpapi-external-agent.h"
7
8 static enum NTDB_ERROR parse(NTDB_DATA key, NTDB_DATA data, NTDB_DATA *expected)
9 {
10         if (!ntdb_deq(data, *expected))
11                 return NTDB_ERR_EINVAL;
12         return NTDB_SUCCESS;
13 }
14
15 static enum NTDB_ERROR parse_err(NTDB_DATA key, NTDB_DATA data, void *unused)
16 {
17         return 100;
18 }
19
20 static bool test_records(struct ntdb_context *ntdb)
21 {
22         int i;
23         NTDB_DATA key = { (unsigned char *)&i, sizeof(i) };
24         NTDB_DATA data = { (unsigned char *)&i, sizeof(i) };
25
26         for (i = 0; i < 1000; i++) {
27                 if (ntdb_store(ntdb, key, data, NTDB_REPLACE) != 0)
28                         return false;
29         }
30
31         for (i = 0; i < 1000; i++) {
32                 if (ntdb_parse_record(ntdb, key, parse, &data) != NTDB_SUCCESS)
33                         return false;
34         }
35
36         if (ntdb_parse_record(ntdb, key, parse, &data) != NTDB_ERR_NOEXIST)
37                 return false;
38
39         /* Test error return from parse function. */
40         i = 0;
41         if (ntdb_parse_record(ntdb, key, parse_err, NULL) != 100)
42                 return false;
43
44         return true;
45 }
46
47 int main(int argc, char *argv[])
48 {
49         unsigned int i;
50         struct ntdb_context *ntdb;
51         int flags[] = { NTDB_INTERNAL, NTDB_DEFAULT, NTDB_NOMMAP,
52                         NTDB_INTERNAL|NTDB_CONVERT, NTDB_CONVERT,
53                         NTDB_NOMMAP|NTDB_CONVERT };
54
55         plan_tests(sizeof(flags) / sizeof(flags[0]) * 2 + 1);
56         for (i = 0; i < sizeof(flags) / sizeof(flags[0]); i++) {
57                 ntdb = ntdb_open("api-21-parse_record.ntdb",
58                                  flags[i]|MAYBE_NOSYNC,
59                                  O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr);
60                 if (ok1(ntdb))
61                         ok1(test_records(ntdb));
62                 ntdb_close(ntdb);
63         }
64
65         ok1(tap_log_messages == 0);
66         return exit_status();
67 }