
#include <winpr/crt.h>
#include <winpr/tchar.h>
#include <winpr/collections.h>

static char* key1 = "key1";
static char* key2 = "key2";
static char* key3 = "key3";

static char* val1 = "val1";
static char* val2 = "val2";
static char* val3 = "val3";

int TestListDictionary(int argc, char* argv[])
{
	size_t count = 0;
	char* value = nullptr;
	wListDictionary* list = nullptr;

	WINPR_UNUSED(argc);
	WINPR_UNUSED(argv);

	list = ListDictionary_New(TRUE);
	if (!list)
		return -1;

	if (!ListDictionary_Add(list, key1, val1) || !ListDictionary_Add(list, key2, val2) ||
	    !ListDictionary_Add(list, key3, val3))
		return -1;

	count = ListDictionary_Count(list);

	if (count != 3)
	{
		printf("ListDictionary_Count: Expected : 3, Actual: %" PRIuz "\n", count);
		return -1;
	}

	ListDictionary_Remove(list, key2);

	count = ListDictionary_Count(list);

	if (count != 2)
	{
		printf("ListDictionary_Count: Expected : 2, Actual: %" PRIuz "\n", count);
		return -1;
	}

	ListDictionary_Remove(list, key3);

	count = ListDictionary_Count(list);

	if (count != 1)
	{
		printf("ListDictionary_Count: Expected : 1, Actual: %" PRIuz "\n", count);
		return -1;
	}

	ListDictionary_Remove(list, key1);

	count = ListDictionary_Count(list);

	if (count != 0)
	{
		printf("ListDictionary_Count: Expected : 0, Actual: %" PRIuz "\n", count);
		return -1;
	}

	if (!ListDictionary_Add(list, key1, val1) || !ListDictionary_Add(list, key2, val2) ||
	    !ListDictionary_Add(list, key3, val3))
		return -1;

	count = ListDictionary_Count(list);

	if (count != 3)
	{
		printf("ListDictionary_Count: Expected : 3, Actual: %" PRIuz "\n", count);
		return -1;
	}

	value = (char*)ListDictionary_GetItemValue(list, key1);

	if (strcmp(value, val1) != 0)
	{
		printf("ListDictionary_GetItemValue: Expected : %" PRIuz ", Actual: %" PRIuz "\n",
		       (size_t)val1, (size_t)value);
		return -1;
	}

	value = (char*)ListDictionary_GetItemValue(list, key2);

	if (strcmp(value, val2) != 0)
	{
		printf("ListDictionary_GetItemValue: Expected : %" PRIuz ", Actual: %" PRIuz "\n",
		       (size_t)val2, (size_t)value);
		return -1;
	}

	value = (char*)ListDictionary_GetItemValue(list, key3);

	if (strcmp(value, val3) != 0)
	{
		printf("ListDictionary_GetItemValue: Expected : %" PRIuz ", Actual: %" PRIuz "\n",
		       (size_t)val3, (size_t)value);
		return -1;
	}

	ListDictionary_SetItemValue(list, key2, "apple");

	value = (char*)ListDictionary_GetItemValue(list, key2);

	if (strcmp(value, "apple") != 0)
	{
		printf("ListDictionary_GetItemValue: Expected : %s, Actual: %s\n", "apple", value);
		return -1;
	}

	if (!ListDictionary_Contains(list, key2))
	{
		printf("ListDictionary_Contains: Expected : TRUE, Actual: FALSE\n");
		return -1;
	}

	if (!ListDictionary_Take(list, key2))
	{
		printf("ListDictionary_Remove: Expected : TRUE, Actual: FALSE\n");
		return -1;
	}

	if (ListDictionary_Take(list, key2))
	{
		printf("ListDictionary_Remove: Expected : FALSE, Actual: TRUE\n");
		return -1;
	}

	value = ListDictionary_Take_Head(list);
	count = ListDictionary_Count(list);
	if ((strncmp(value, val1, 4) != 0) || (count != 1))
	{
		printf("ListDictionary_Remove_Head: Expected : %s, Actual: %s Count: %" PRIuz "\n", val1,
		       value, count);
		return -1;
	}

	value = ListDictionary_Take_Head(list);
	count = ListDictionary_Count(list);
	if ((strncmp(value, val3, 4) != 0) || (count != 0))
	{
		printf("ListDictionary_Remove_Head: Expected : %s, Actual: %s Count: %" PRIuz "\n", val3,
		       value, count);
		return -1;
	}

	value = ListDictionary_Take_Head(list);
	if (value)
	{
		printf("ListDictionary_Remove_Head: Expected : (null), Actual: %s\n", value);
		return -1;
	}

	if (!ListDictionary_Add(list, key1, val1) || !ListDictionary_Add(list, key2, val2) ||
	    !ListDictionary_Add(list, key3, val3))
		return -1;

	ListDictionary_Clear(list);

	count = ListDictionary_Count(list);

	if (count != 0)
	{
		printf("ListDictionary_Count: Expected : 0, Actual: %" PRIuz "\n", count);
		return -1;
	}

	ListDictionary_Free(list);

	return 0;
}
