/*
   test_set.c - simple test for the set module
   This file is part of the nss-pam-ldapd library.

   Copyright (C) 2008-2014 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/set.h"
#include "compat/attrs.h"

static int isknownvalue(const char *value)
{
  return value != NULL && (
        (strcmp(value, "key1") == 0) || (strcmp(value, "key2") == 0) ||
        (strcmp(value, "key3") == 0));
}

/* the main program... */
int main(int UNUSED(argc), char UNUSED(*argv[]))
{
  SET *set;
  const char **list;
  int i;
  const char *v;

  /* initialize */
  set = set_new();

  /* store some entries */
  set_add(set, "key1");
  set_add(set, "key2");
  set_add(set, "key3");
  set_add(set, "key2");

  /* check set contents */
  assert(set_contains(set, "key1"));
  assert(set_contains(set, "key2"));
  assert(set_contains(set, "key3"));
  assert(!set_contains(set, "key4"));
  assert(!set_contains(set, "KEY1"));

  /* loop over set contents */
  list = set_tolist(set);
  for (i = 0; list[i] != NULL; i++)
  {
    assert(isknownvalue(list[i]));
  }

  /* remove keys from the set */
  assert(isknownvalue(v = set_pop(set)));
  free((void *)v);
  assert(isknownvalue(v = set_pop(set)));
  free((void *)v);
  assert(isknownvalue(v = set_pop(set)));
  free((void *)v);
  assert(set_pop(set) == NULL);

  /* free set */
  set_free(set);
  free(list);

  return 0;
}