#include <assert.h>
#include <stdio.h>
#include <dict.h>
#include <string.h>

#define ELEMS	16

/* Make starwars fans happy */
const char keys[][16] = {
	"It",
	"is",
	"a",
	"period",
	"of",
	"civil",
	"war.",
	"Rebel",
	"spaceships,",
	"striking",
	"from a",
	"hidden",
	"base,",
	"have",
	"won",
	"their..."
};

int values[16];

const int methods[3] = {
	AURA_DICT_HASH,
	AURA_DICT_LIST,
	AURA_DICT_SORTED_LIST
};

const char *
m2s(int method)
{
	switch(method) {
	case AURA_DICT_HASH:
		return "hash";
		break;	/* NOTREACHED */
	case AURA_DICT_LIST:
		return "list";
		break;	/* NOTREACHED */
	case AURA_DICT_SORTED_LIST:
		return "sorted_list";
		break;	/* NOTREACHED */
	default:
		return NULL;
		break;	/* NOTREACHED */
	}
}

static struct aura_dict *
test_store(int method)
{
	struct aura_dict *d;

	d = aura_dict_new(ELEMS, method);
	for (int i = 0; i < ELEMS; i++) {
		values[i] = i;
		aura_dict_store(d, keys[i], strlen(keys[i]),
		    &values[i], sizeof(int));
		printf("%s_store: %s=>%d\n",
		    m2s(method), keys[i], values[i]);
	}
	return d;
}

static int
test_fetch(struct aura_dict *d, int method)
{
	if (d == NULL)
		return -1;

	aura_dict_rewind(d);
	for (int i = 0; i < ELEMS; i++) {
		int *val = malloc(sizeof(int));
		size_t len = sizeof(int);
		aura_dict_fetch(d, keys[i], strlen(keys[i]),
		    (void *)&val, &len);
		if (*val != values[i]) {
			fprintf(stderr,
			    "mismatch *val=%d values[%d]=%d\n",
			    *val, i, values[i]);
			return -1;
		}
		printf("%s_fetch: %s=>%d\n", m2s(method), keys[i], *val);
		free(val);
	}
	aura_dict_free(d);

	return 0;
}

int
main(void)
{
	struct aura_dict *d;

	for (int i = 1; i <= 3; i++) {
		d = test_store(i);

		if ((test_fetch(d, i)) == -1) {
			return 1;
		}
	}

	return 0;
}