/*
 * Copyright (C) 2016 Robin Krahl <robin.krahl@ireas.org>
 *
 * dbfp_check.c
 */

#include <check.h>
#include <stdio.h>
#include <stdlib.h>

#include "dbfp.h"

static char *api_key;

static void assert_location_eq(size_t n, struct dbfp_location *locations,
		size_t i, const char *name, const char *id, float lon,
		float lat);

static void assert_departure_eq(size_t n, struct dbfp_departure *departures,
		size_t i, const char *name, const char *type,
		const char *stopid, const char *stop, time_t time,
		const char *direction, const char *track);

START_TEST(test_dbfp_create)
{
	struct dbfp dbfp;
	int err;

	err = dbfp_init(&dbfp, "test-key");
	ck_assert_int_eq(err, 0);
	ck_assert_str_eq(dbfp.key, "test-key");
	dbfp_close(&dbfp);
}
END_TEST

START_TEST(test_dbfp_access)
{
	int err;
	struct dbfp dbfp;
	struct dbfp_location *locations = NULL;
	size_t n;

	err = dbfp_init(&dbfp, api_key);
	ck_assert_int_eq(err, 0);

	err = dbfp_query_location(&dbfp, "", &n, &locations);
	ck_assert_int_eq(err, 0);

	free(locations);
	dbfp_close(&dbfp);
}
END_TEST

START_TEST(test_dbfp_location_simple)
{
	int err;
	struct dbfp dbfp;
	struct dbfp_location *locations = NULL;
	size_t n;
	size_t i;

	err = dbfp_init(&dbfp, api_key);
	ck_assert_int_eq(err, 0);

	err = dbfp_query_location(&dbfp, "freiburg", &n, &locations);
	ck_assert_int_eq(err, 0);
	assert_location_eq(n, locations, 0, "Freiburg(Breisgau) Hbf",
			"008000107", 7.841173f, 47.997696f);
	for (i = 0; i < n; i++)
		dbfp_location_close(&locations[i]);
	free(locations);

	err = dbfp_query_location(&dbfp, "freib", &n, &locations);
	ck_assert_int_eq(err, 0);
	assert_location_eq(n, locations, 0, "Freiburg(Breisgau) Hbf",
			"008000107", 7.841173f, 47.997696f);
	for (i = 0; i < n; i++)
		dbfp_location_close(&locations[i]);
	free(locations);

	err = dbfp_query_location(&dbfp, "freiburg im", &n, &locations);
	ck_assert_int_eq(err, 0);
	assert_location_eq(n, locations, 0, "Freiburg(Breisgau) Hbf",
			"008000107", 7.841173f, 47.997696f);
	for (i = 0; i < n; i++)
		dbfp_location_close(&locations[i]);
	free(locations);

	dbfp_close(&dbfp);
}
END_TEST

START_TEST(test_dbfp_departure_simple)
{
	int err;
	struct dbfp dbfp;
	struct dbfp_departure *departures = NULL;
	struct tm tm = { .tm_year = 116, .tm_mon = 5, .tm_mday = 1,
		.tm_hour = 17, .tm_min = 42 };
	time_t time = mktime(&tm);
	size_t n;
	size_t i;

	err = dbfp_init(&dbfp, api_key);
	ck_assert_int_eq(err, 0);

	err = dbfp_query_departure(&dbfp, NULL, "8000105", 0, &n, &departures);
	ck_assert_int_eq(err, 0);
	ck_assert_int_gt(n, 0);
	for (i = 0; i < n; i++)
		dbfp_departure_close(&departures[i]);
	free(departures);

	err = dbfp_query_departure(&dbfp, NULL, "8000105", time, &n,
			&departures);
	ck_assert_int_eq(err, 0);
	tm.tm_hour++;
	time = mktime(&tm);
	assert_departure_eq(n, departures, 0, "ICE 1556", "ICE", "8000105",
			"Frankfurt(Main)Hbf", time, "Wiesbaden Hbf", "6");
	tm.tm_min = 49;
	time = mktime(&tm);
	assert_departure_eq(n, departures, 1, "IC 2274", "IC", "8000105",
			"Frankfurt(Main)Hbf", time, "Kassel-Wilhelmshöhe",
			"11");
	for (i = 0; i < n; i++)
		dbfp_departure_close(&departures[i]);
	free(departures);

	dbfp_close(&dbfp);
}
END_TEST

int main(int argc, char **argv)
{
	int num_failed = 0;
	Suite *s;
	SRunner *sr;
	TCase *tc_basic;
	TCase *tc_location;
	TCase *tc_departure;

	api_key = getenv("DBFP_API_KEY");
	if (!api_key) {
		fprintf(stderr, "DBFP_API_KEY not set\n");
		return EXIT_FAILURE;
	}

	s = suite_create("dbfp");
	sr = srunner_create(s);

	tc_basic = tcase_create("basic");
	tcase_add_test(tc_basic, test_dbfp_create);
	tcase_add_test(tc_basic, test_dbfp_access);
	suite_add_tcase(s, tc_basic);

	tc_location = tcase_create("location");
	tcase_add_test(tc_location, test_dbfp_location_simple);
	suite_add_tcase(s, tc_location);

	tc_departure = tcase_create("departure");
	tcase_add_test(tc_departure, test_dbfp_departure_simple);
	suite_add_tcase(s, tc_departure);

	srunner_run_all(sr, CK_ENV);
	num_failed = srunner_ntests_failed(sr);
	srunner_free(sr);

	return (num_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
}

static void assert_location_eq(size_t n, struct dbfp_location *locations,
		size_t i, const char *name, const char *id, float lon,
		float lat)
{
	ck_assert_ptr_ne(locations, NULL);
	ck_assert_uint_lt(i, n);

	ck_assert_str_eq(locations[i].name, name);
	ck_assert_str_eq(locations[i].id, id);
	ck_assert(locations[i].lon == lon);
	ck_assert(locations[i].lat == lat);
}

static void assert_departure_eq(size_t n, struct dbfp_departure *departures,
		size_t i, const char *name, const char *type,
		const char *stopid, const char *stop, time_t time,
		const char *direction, const char *track)
{
	ck_assert_ptr_ne(departures, NULL);
	ck_assert_uint_lt(i, n);

	ck_assert_str_eq(departures[i].name, name);
	ck_assert_str_eq(departures[i].type, type);
	ck_assert_str_eq(departures[i].stopid, stopid);
	ck_assert_str_eq(departures[i].stop, stop);
	ck_assert_uint_eq(departures[i].time, time);
	ck_assert_str_eq(departures[i].direction, direction);
	ck_assert_str_eq(departures[i].track, track);
}