#include "config.h"

#include <sys/types.h>
#include <sys/socket.h>
#include <munit.h>

#include "libei.h"
#include "eierpecken.h"

/* Macros intended just for readability to make it more obvious which part
   of a test handles server vs client */
#define with_server(peck_) for (struct eis *eis = peck_get_eis(peck_); eis; eis = NULL)
#define with_client(peck_) for (struct ei *ei = peck_get_ei(peck_); ei; ei = NULL)

static MunitResult
test_ei_ref_unref(const MunitParameter params[], void *user_data)
{
	struct ei *ei = ei_new(NULL);

	struct ei *refd = ei_ref(ei);
	munit_assert_ptr_equal(ei, refd);

	struct ei *unrefd = ei_unref(ei);
	munit_assert_ptr_null(unrefd);
	unrefd = ei_unref(ei);
	munit_assert_ptr_null(unrefd);

	/* memleak only shows up in valgrind */

	return MUNIT_OK;
}

static MunitResult
test_ei_reject(const MunitParameter params[], void *user_data)
{
	_cleanup_peck_ struct peck *peck = peck_new();

	/* Client is immediately rejected */
	peck_enable_eis_behavior(peck, PECK_EIS_BEHAVIOR_REJECT_CLIENT);
	peck_dispatch_ei(peck);
	peck_dispatch_eis(peck);

	/* Expect the client to get a disconnect event */
	with_client(peck) {
		ei_dispatch(ei);
		_cleanup_ei_event_ struct ei_event *e = ei_get_event(ei);
		munit_assert_ptr_not_null(e);
		munit_assert_int(ei_event_get_type(e), ==, EI_EVENT_DISCONNECT);
	}

	return MUNIT_OK;
}

static MunitResult
test_ei_reject_after_connect(const MunitParameter params[], void *user_data)
{
	_cleanup_peck_ struct peck *peck = peck_new();
	_cleanup_eis_client_ struct eis_client *client = NULL;

	peck_dispatch_ei(peck);

	with_server(peck) {
		eis_dispatch(eis);
		_cleanup_eis_event_ struct eis_event *e = eis_get_event(eis);
		munit_assert_ptr_not_null(e);
		munit_assert_int(eis_event_get_type(e), ==, EIS_EVENT_CLIENT_CONNECT);
		client = eis_client_ref(eis_event_get_client(e));
		eis_client_connect(client);
	}

	with_client(peck) {
		ei_dispatch(ei);
		_cleanup_ei_event_ struct ei_event *e = ei_get_event(ei);
		munit_assert_ptr_not_null(e);
		munit_assert_int(ei_event_get_type(e), ==, EI_EVENT_CONNECT);
	}

	with_server(peck) {
		eis_client_disconnect(client);
	}

	with_client(peck) {
		ei_dispatch(ei);
		_cleanup_ei_event_ struct ei_event *e = ei_get_event(ei);
		munit_assert_ptr_not_null(e);
		munit_assert_int(ei_event_get_type(e), ==, EI_EVENT_DISCONNECT);
	}

	return MUNIT_OK;
}

static MunitResult
test_ei_device_basics(const MunitParameter params[], void *user_data)
{
	_cleanup_peck_ struct peck *peck = peck_new();
	_cleanup_ei_device_ struct ei_device *device = NULL;

	peck_enable_eis_behavior(peck, PECK_EIS_BEHAVIOR_ACCEPT_ALL);
	peck_dispatch_eis(peck);
	peck_dispatch_ei(peck);

	/* device creation and getters/setters test */
	with_client(peck) {
		device = ei_device_new(ei);
		munit_assert_not_null(device);
		ei_device_configure_name(device, __func__);
		munit_assert_string_equal(ei_device_get_name(device), __func__);

		bool success = ei_device_configure_capability(device, EI_DEVICE_CAP_POINTER);
		munit_assert(success);
		munit_assert(ei_device_has_capability(device, EI_DEVICE_CAP_POINTER));

		/* Add it, but we don't care about whether it worked correctly in this test */
		ei_device_add(device);

		/* Device is immutable after ei_device_add() */
		bool failed_caps = ei_device_configure_capability(device, EI_DEVICE_CAP_POINTER);
		munit_assert(failed_caps == false);
		munit_assert(ei_device_has_capability(device, EI_DEVICE_CAP_POINTER));
	}

	/* Drain both sides, we don't care about the events themselves */
	with_server(peck) {
		peck_drain_eis(eis);
	}
	with_client(peck) {
		peck_drain_ei(ei);
	}

	/* device is still immutable */
	with_client(peck) {
		bool failed_caps = ei_device_configure_capability(device, EI_DEVICE_CAP_POINTER);
		munit_assert(failed_caps == false);
		munit_assert(ei_device_has_capability(device, EI_DEVICE_CAP_POINTER));
	}

	return MUNIT_OK;
}

static MunitResult
test_ei_device_add(const MunitParameter params[], void *user_data)
{
	_cleanup_peck_ struct peck *peck = peck_new();
	struct ei *ei = peck_get_ei(peck);

	peck_enable_eis_behavior(peck, PECK_EIS_BEHAVIOR_ACCEPT_ALL);
	peck_dispatch_eis(peck);

	peck_drain_ei(ei);

	_cleanup_ei_device_ struct ei_device *device = ei_device_new(ei);
	ei_device_configure_name(device, __func__);
	ei_device_configure_capability(device, EI_DEVICE_CAP_POINTER);
	ei_device_add(device);

	return MUNIT_OK;
}

static MunitTest ei_tests[] = {
	{ .name = "/ei/ref",			.test = test_ei_ref_unref		},
	{ .name = "/ei/reject",			.test = test_ei_reject			},
	{ .name = "/ei/reject_after_connect",	.test = test_ei_reject_after_connect	},
	{ .name = "/device/basics",		.test = test_ei_device_basics		},
	{ .name = "/device/add",		.test = test_ei_device_add		},
};

static const MunitSuite iotest_suite = {
	"/ei",
	ei_tests,
	NULL,
	1,
	MUNIT_SUITE_OPTION_NONE,
};

int
main(int argc, char* argv[MUNIT_ARRAY_PARAM(argc + 1)])
{
	return munit_suite_main(&iotest_suite, NULL, argc, argv);
}