/*
   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;
}