mirror of
https://github.com/samba-team/samba.git
synced 2025-01-14 19:24:43 +03:00
0265837ee8
This reduces test time from 31 seconds to 6, on my laptop. Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
91 lines
2.3 KiB
C
91 lines
2.3 KiB
C
#include "ntdb-source.h"
|
|
#include "tap-interface.h"
|
|
#include "logging.h"
|
|
|
|
static int drop_count(struct ntdb_context *ntdb, unsigned int *count)
|
|
{
|
|
if (--(*count) == 0)
|
|
return 1;
|
|
return 0;
|
|
}
|
|
|
|
static int set_found(struct ntdb_context *ntdb, bool found[3])
|
|
{
|
|
unsigned int idx;
|
|
|
|
if (strcmp(ntdb_name(ntdb), "run-ntdb_foreach0.ntdb") == 0)
|
|
idx = 0;
|
|
else if (strcmp(ntdb_name(ntdb), "run-ntdb_foreach1.ntdb") == 0)
|
|
idx = 1;
|
|
else if (strcmp(ntdb_name(ntdb), "run-ntdb_foreach2.ntdb") == 0)
|
|
idx = 2;
|
|
else
|
|
abort();
|
|
|
|
if (found[idx])
|
|
abort();
|
|
found[idx] = true;
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
unsigned int i, count;
|
|
bool found[3];
|
|
struct ntdb_context *ntdb0, *ntdb1, *ntdb;
|
|
int flags[] = { NTDB_DEFAULT, NTDB_NOMMAP,
|
|
NTDB_CONVERT, NTDB_NOMMAP|NTDB_CONVERT };
|
|
|
|
plan_tests(sizeof(flags) / sizeof(flags[0]) * 8);
|
|
for (i = 0; i < sizeof(flags) / sizeof(flags[0]); i++) {
|
|
ntdb0 = ntdb_open("run-ntdb_foreach0.ntdb",
|
|
flags[i]|MAYBE_NOSYNC,
|
|
O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr);
|
|
ntdb1 = ntdb_open("run-ntdb_foreach1.ntdb",
|
|
flags[i]|MAYBE_NOSYNC,
|
|
O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr);
|
|
ntdb = ntdb_open("run-ntdb_foreach2.ntdb",
|
|
flags[i]|MAYBE_NOSYNC,
|
|
O_RDWR|O_CREAT|O_TRUNC, 0600, &tap_log_attr);
|
|
|
|
memset(found, 0, sizeof(found));
|
|
ntdb_foreach(set_found, found);
|
|
ok1(found[0] && found[1] && found[2]);
|
|
|
|
/* Test premature iteration termination */
|
|
count = 1;
|
|
ntdb_foreach(drop_count, &count);
|
|
ok1(count == 0);
|
|
|
|
ntdb_close(ntdb1);
|
|
memset(found, 0, sizeof(found));
|
|
ntdb_foreach(set_found, found);
|
|
ok1(found[0] && !found[1] && found[2]);
|
|
|
|
ntdb_close(ntdb);
|
|
memset(found, 0, sizeof(found));
|
|
ntdb_foreach(set_found, found);
|
|
ok1(found[0] && !found[1] && !found[2]);
|
|
|
|
ntdb1 = ntdb_open("run-ntdb_foreach1.ntdb",
|
|
flags[i]|MAYBE_NOSYNC,
|
|
O_RDWR, 0600, &tap_log_attr);
|
|
memset(found, 0, sizeof(found));
|
|
ntdb_foreach(set_found, found);
|
|
ok1(found[0] && found[1] && !found[2]);
|
|
|
|
ntdb_close(ntdb0);
|
|
memset(found, 0, sizeof(found));
|
|
ntdb_foreach(set_found, found);
|
|
ok1(!found[0] && found[1] && !found[2]);
|
|
|
|
ntdb_close(ntdb1);
|
|
memset(found, 0, sizeof(found));
|
|
ntdb_foreach(set_found, found);
|
|
ok1(!found[0] && !found[1] && !found[2]);
|
|
ok1(tap_log_messages == 0);
|
|
}
|
|
|
|
return exit_status();
|
|
}
|