/*
test_dict.c - simple test for the dict module
This file is part of the nss-pam-ldapd library.
Copyright (C) 2007, 2008, 2009, 2010, 2012 Arthur de Jong
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
02110-1301 USA
*/
#include "config.h"
#include <stdio.h>
#include <string.h>
#include <assert.h>
#include <stdlib.h>
#include "common/dict.h"
#include "compat/attrs.h"
/* Simple test that adds a few key/value pairs to the dict and the does
most operations. */
static void test_simple(void)
{
DICT *dict;
void *val;
static char *value1 = "value1";
static char *value2 = "value2";
static char *replace2 = "replace2";
const char **keys;
int i;
/* initialize */
dict = dict_new();
/* store some entries */
dict_put(dict, "key1", value1);
dict_put(dict, "key2", value2);
dict_put(dict, "key3", dict);
dict_put(dict, "key2", replace2);
/* check dictionary contents */
val = dict_get(dict, "key1");
assert(val == value1);
val = dict_get(dict, "key2");
assert(val == replace2);
val = dict_get(dict, "key3");
assert(val == dict);
val = dict_get(dict, "key4");
assert(val == NULL);
val = dict_get(dict, "KEY1");
assert(val == NULL);
/* remove a key */
dict_put(dict, "key3", NULL);
val = dict_get(dict, "key3");
assert(val == NULL);
/* loop over dictionary contents */
keys = dict_keys(dict);
for (i = 0; keys[i] != NULL; i++)
{
val = dict_get(dict, keys[i]);
assert(((val == value1) || (val == replace2)));
}
/* free stuff */
dict_free(dict);
free(keys);
}
/* Test to insert a large number of elements in the dict. */
static void test_lotsofelements(void)
{
DICT *dict;
char buf[80];
int i, r;
void *val;
const char **keys;
/* initialize */
dict = dict_new();
/* insert a number of entries */
for (i = 0; i < 1024; i++)
{
r = 1 + (int)(10000.0 * (rand() / (RAND_MAX + 1.0)));
sprintf(buf, "test%04d", r);
dict_put(dict, buf, &buf);
}
/* remove a number of entries */
for (i = 0; i < 100; i++)
{
r = 1 + (int)(10000.0 * (rand() / (RAND_MAX + 1.0)));
sprintf(buf, "test%04d", r);
dict_put(dict, buf, NULL);
}
/* add some more entries */
for (i = 0; i < 1024; i++)
{
r = 1 + (int)(10000.0 * (rand() / (RAND_MAX + 1.0)));
sprintf(buf, "test%04d", r);
dict_put(dict, buf, &buf);
}
/* loop over dictionary contents */
keys = dict_keys(dict);
for (i = 0; keys[i] != NULL; i++)
{
val = dict_get(dict, keys[i]);
assert(val == buf);
}
/* free stuff */
dict_free(dict);
free(keys);
}
/* Test to insert a large number of elements in the dict. */
static void test_readelements(const char *fname)
{
DICT *dict;
char buf[80];
FILE *fp;
void *val;
const char **keys;
int i;
/* initialize */
dict = dict_new();
/* read file and insert all entries */
fp = fopen(fname, "r");
assert(fp != NULL);
while (fgets(buf, sizeof(buf), fp) != NULL)
{
/* strip newline */
buf[strlen(buf) - 1] = '\0';
dict_put(dict, buf, &buf);
}
fclose(fp);
/* loop over dictionary contents */
keys = dict_keys(dict);
for (i = 0; keys[i] != NULL; i++)
{
val = dict_get(dict, keys[i]);
assert(val == buf);
}
/* free stuff */
dict_free(dict);
free(keys);
}
static void test_countelements(int num)
{
DICT *dict;
char buf[80];
int i, r;
const char **keys;
/* initialize */
dict = dict_new();
/* insert a number of entries */
for (i = 0; i < num; i++)
{
r = 1 + (int)(10000.0 * (rand() / (RAND_MAX + 1.0)));
sprintf(buf, "%04dx%04d", i, r);
dict_put(dict, buf, &buf);
}
/* loop over dictionary contents */
keys = dict_keys(dict);
for (i = 0; keys[i] != NULL; i++)
/* nothing */ ;
/* we should have num elements */
assert(i == num);
/* free stuff */
dict_free(dict);
free(keys);
}
/* the main program... */
int main(int UNUSED(argc), char UNUSED(*argv[]))
{
char *srcdir;
char fname[100];
/* build the name of the file */
srcdir = getenv("srcdir");
if (srcdir == NULL)
strcpy(fname, "usernames.txt");
else
snprintf(fname, sizeof(fname), "%s/usernames.txt", srcdir);
fname[sizeof(fname) - 1] = '\0';
/* run the tests */
test_simple();
test_lotsofelements();
test_readelements(fname);
test_countelements(0);
test_countelements(1);
test_countelements(2);
test_countelements(3);
test_countelements(4);
test_countelements(10);
test_countelements(20);
return 0;
}