END_TEST
START_TEST(test_mgr_check_no_keys)
{
bloom_config config;
int res = config_from_filename(NULL, &config);
fail_unless(res == 0);
bloom_filtmgr *mgr;
res = init_filter_manager(&config, &mgr);
fail_unless(res == 0);
res = filtmgr_create_filter(mgr, "zab2", NULL);
fail_unless(res == 0);
char *keys[] = {"hey","there","person"};
char result[] = {1, 1, 1};
res = filtmgr_check_keys(mgr, "zab2", (char**)&keys, 3, (char*)&result);
fail_unless(res == 0);
fail_unless(!result[0]);
fail_unless(!result[1]);
fail_unless(!result[2]);
res = filtmgr_drop_filter(mgr, "zab2");
fail_unless(res == 0);
res = destroy_filter_manager(mgr);
fail_unless(res == 0);
}
END_TEST
START_TEST(test_mgr_clear_reload)
{
bloom_config config;
int res = config_from_filename(NULL, &config);
fail_unless(res == 0);
bloom_filtmgr *mgr;
res = init_filter_manager(&config, &mgr);
fail_unless(res == 0);
res = filtmgr_create_filter(mgr, "zab9", NULL);
fail_unless(res == 0);
// Try to add keys now
char *keys[] = {"hey","there","person"};
char result[] = {0, 0, 0};
res = filtmgr_set_keys(mgr, "zab9", (char**)&keys, 3, (char*)&result);
fail_unless(res == 0);
fail_unless(result[0]);
fail_unless(result[1]);
fail_unless(result[2]);
res = filtmgr_unmap_filter(mgr, "zab9");
fail_unless(res == 0);
// F*****G annoying umask permissions bullshit
// Cused by the Check test framework
fail_unless(chmod("/tmp/bloomd/bloomd.zab9/config.ini", 0777) == 0);
fail_unless(chmod("/tmp/bloomd/bloomd.zab9/data.000.mmap", 0777) == 0);
res = filtmgr_clear_filter(mgr, "zab9");
fail_unless(res == 0);
// Force a vacuum
filtmgr_vacuum(mgr);
// This should rediscover
res = filtmgr_create_filter(mgr, "zab9", NULL);
fail_unless(res == 0);
// Try to check keys now
res = filtmgr_check_keys(mgr, "zab9", (char**)&keys, 3, (char*)&result);
fail_unless(res == 0);
fail_unless(result[0]);
fail_unless(result[1]);
fail_unless(result[2]);
res = filtmgr_drop_filter(mgr, "zab9");
fail_unless(res == 0);
res = destroy_filter_manager(mgr);
fail_unless(res == 0);
}
END_TEST
/* Close & Restore */
START_TEST(test_mgr_restore)
{
bloom_config config;
int res = config_from_filename(NULL, &config);
fail_unless(res == 0);
bloom_filtmgr *mgr;
res = init_filter_manager(&config, &mgr);
fail_unless(res == 0);
res = filtmgr_create_filter(mgr, "zab8", NULL);
fail_unless(res == 0);
char *keys[] = {"hey","there","person"};
char result[] = {0, 0, 0};
res = filtmgr_set_keys(mgr, "zab8", (char**)&keys, 3, (char*)&result);
fail_unless(res == 0);
fail_unless(result[0]);
fail_unless(result[1]);
fail_unless(result[2]);
// Shutdown
res = destroy_filter_manager(mgr);
fail_unless(res == 0);
// F*****G annoying umask permissions bullshit
// Cused by the Check test framework
fail_unless(chmod("/tmp/bloomd/bloomd.zab8/config.ini", 0777) == 0);
fail_unless(chmod("/tmp/bloomd/bloomd.zab8/data.000.mmap", 0777) == 0);
// Restrore
res = init_filter_manager(&config, &mgr);
fail_unless(res == 0);
for (int i=0;i<3;i++) result[i] = 0;
res = filtmgr_check_keys(mgr, "zab8", (char**)&keys, 3, (char*)&result);
fail_unless(res == 0);
fail_unless(result[0]);
fail_unless(result[1]);
fail_unless(result[2]);
res = filtmgr_drop_filter(mgr, "zab8");
fail_unless(res == 0);
res = destroy_filter_manager(mgr);
fail_unless(res == 0);
}
END_TEST
/* Unmap in memory */
START_TEST(test_mgr_unmap_in_mem)
{
bloom_config config;
int res = config_from_filename(NULL, &config);
fail_unless(res == 0);
config.in_memory = 1;
bloom_filtmgr *mgr;
res = init_filter_manager(&config, &mgr);
fail_unless(res == 0);
res = filtmgr_create_filter(mgr, "mem1", NULL);
fail_unless(res == 0);
// Try to add keys now
char *keys[] = {"hey","there","person"};
char result[] = {0, 0, 0};
res = filtmgr_set_keys(mgr, "mem1", (char**)&keys, 3, (char*)&result);
fail_unless(res == 0);
fail_unless(result[0]);
fail_unless(result[1]);
fail_unless(result[2]);
res = filtmgr_unmap_filter(mgr, "mem1");
fail_unless(res == 0);
// Try to add keys now
for (int i=0;i<3;i++) result[i] = 0;
res = filtmgr_check_keys(mgr, "mem1", (char**)&keys, 3, (char*)&result);
fail_unless(res == 0);
fail_unless(result[0]);
fail_unless(result[1]);
fail_unless(result[2]);
res = filtmgr_drop_filter(mgr, "mem1");
fail_unless(res == 0);
res = destroy_filter_manager(mgr);
fail_unless(res == 0);
}
END_TEST
START_TEST(test_mgr_add_check_keys)
{
bloom_config config;
int res = config_from_filename(NULL, &config);
fail_unless(res == 0);
bloom_filtmgr *mgr;
res = init_filter_manager(&config, 0, &mgr);
fail_unless(res == 0);
res = filtmgr_create_filter(mgr, "zab1", NULL);
fail_unless(res == 0);
char *keys[] = {"hey","there","person"};
char result[] = {0, 0, 0};
res = filtmgr_set_keys(mgr, "zab1", (char**)&keys, 3, (char*)&result);
fail_unless(res == 0);
fail_unless(result[0]);
fail_unless(result[1]);
fail_unless(result[2]);
for (int i=0;i<3;i++) result[i] = 0;
res = filtmgr_check_keys(mgr, "zab1", (char**)&keys, 3, (char*)&result);
fail_unless(res == 0);
fail_unless(result[0]);
fail_unless(result[1]);
fail_unless(result[2]);
res = filtmgr_drop_filter(mgr, "zab1");
fail_unless(res == 0);
res = destroy_filter_manager(mgr);
fail_unless(res == 0);
}
