/*

 * Copyright (C) 2007-2010 Gary Kramlich <grim@reaperworld.com>

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation, either version 3 of the License, or

 * (at your option) any later version.

 *

 * This program 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program.  If not, see <http://www.gnu.org/licenses/>.

 */

#include <gplate/gplate.h>

#include <gplate/variables/gplate-dictionary-variable.h>

#include <glib.h>

/******************************************************************************

 * Structs

 *****************************************************************************/

typedef struct {

	const gchar *name;

	const gchar *value;

} GPlateForFunctionTestData;

typedef struct {

	GPlateTemplate *template;

	GPlateVariable *dict;

	const gchar *template_string;

	gchar *actual;

	const gchar *expected;

	GError *error;

	GPlateForFunctionTestData *data;

} GPlateForFunctionTestFixture;

/******************************************************************************

 * Fixtures

 *****************************************************************************/

static void

gplate_for_function_test_setup(GPlateForFunctionTestFixture *fixture,

                               gconstpointer d)

{

	fixture->template = gplate_template_new();

	fixture->dict = gplate_dictionary_variable_new("list");

	gplate_collection_add_variable(GPLATE_COLLECTION(fixture->template),

	                               fixture->dict);

}

static void

gplate_for_function_test_teardown(GPlateForFunctionTestFixture *fixture,

                                  gconstpointer d)

{

	g_object_unref(fixture->template);

	fixture->template = NULL;

	g_object_unref(fixture->dict);

	fixture->dict = NULL;

	g_free(fixture->actual);

	fixture->actual = NULL;

	fixture->expected = NULL;

	if(fixture->error) {

		g_error_free(fixture->error);

		fixture->error = NULL;

	}

}

/******************************************************************************

 * Helpers

 *****************************************************************************/

static void

gplate_for_function_test(GPlateForFunctionTestFixture *fixture) {

	gint i = 0;

	/* add the variables */

	for(i = 0; fixture->data[i].name; i++) {

		gplate_collection_add_string(fixture->dict,

		                             fixture->data[i].name,

		                             fixture->data[i].value);

	}

	fixture->actual = gplate_template_render(fixture->template,

	                                         fixture->template_string,

	                                         &fixture->error);

	g_assert(fixture->error == NULL);

	g_assert_cmpstr(fixture->expected, ==, fixture->actual);

}

/******************************************************************************

 * Simple For's

 *****************************************************************************/

static void

gplate_for_function_test_zero_elements(GPlateForFunctionTestFixture *fixture,

                                       gconstpointer user_data)

{

	GPlateForFunctionTestData data[] = {

		{ NULL, NULL },

	};

	fixture->data = data;

	fixture->template_string = "{% for in in line %}{{ i }}{% endfor %}";

	fixture->expected = "";

	gplate_for_function_test(fixture);

}

static void

gplate_for_function_test_one_element(GPlateForFunctionTestFixture *fixture,

                                     gconstpointer user_data)

{

	GPlateForFunctionTestData data[] = {

		{ "one", "1" },

		{ NULL, NULL },

	};

	fixture->data = data;

	fixture->template_string = "{% for i in list %}{{ i }}{% endfor %}";

	fixture->expected = "1";

	gplate_for_function_test(fixture);

}

static void

gplate_for_function_test_two_elements(GPlateForFunctionTestFixture *fixture,

                                      gconstpointer user_data)

{

	GPlateForFunctionTestData data[] = {

		{ "a", "A" },

		{ "b", "B" },

		{ NULL, NULL },

	};

	fixture->data = data;

	fixture->template_string = "{% for i in list %}{{ i }}{% endfor %}";

	fixture->expected = "AB";

	gplate_for_function_test(fixture);

}

/******************************************************************************

 * Main!

 *****************************************************************************/

gint

main(gint argc, gchar **argv) {

	g_test_init(&argc, &argv, NULL);

	g_type_init();

	gplate_config_load_default();

	g_test_add("/functions/foo/zero_elements",

	           GPlateForFunctionTestFixture,

	           NULL,

	           gplate_for_function_test_setup,

	           gplate_for_function_test_zero_elements,

	           gplate_for_function_test_teardown);

	g_test_add("/functions/foo/one_element",

	           GPlateForFunctionTestFixture,

	           NULL,

	           gplate_for_function_test_setup,

	           gplate_for_function_test_one_element,

	           gplate_for_function_test_teardown);

	g_test_add("/functions/foo/two_elements",

	           GPlateForFunctionTestFixture,

	           NULL,

	           gplate_for_function_test_setup,

	           gplate_for_function_test_two_elements,

	           gplate_for_function_test_teardown);

	return g_test_run();

}