#include "hs_tests.h"

static int testfunc1(lua_State *L) { return 0; }
static int testfunc2(lua_State *L) { return 0; }

/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 *
 * tests for hs_create_table
 *
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 */

TEST(table_correct_index)
{
   int top_old = lua_gettop(L);
   int index = hs_create_table(L);
   mu_assert("the stack is unchanged!", lua_gettop(L) != top_old);
   mu_assert("returned incorrect index!", index == lua_gettop(L));
   return 0;
}

#define load_key(index, value, pushf) do {		\
      pushf(L, value); lua_gettable(L, index);		\
      mu_assert("value at key '" #value "' is nil!",	\
		!lua_isnil(L, -1));			\
   } while(0)

#define loadkey_bool(index, value)		\
   load_key(index, value, lua_pushboolean)
#define loadkey_int(index, value)		\
   load_key(index, value, lua_pushinteger)
#define loadkey_num(index, value)		\
   load_key(index, value, lua_pushnumber)
#define loadkey_str(index, value)		\
   load_key(index, value, lua_pushstring)
#define loadkey_tbl(index, value) do {				\
      lua_pushvalue(L, value); lua_gettable(L, index);		\
      mu_assert("value at table (index '" #value "') is nil!",	\
		!lua_isnil(L, -1));				\
   } while(0)
#define loadkey_func(index, value) do {					\
      lua_pushvalue(L, value); lua_gettable(L, index);			\
      mu_assert("value at function (index '" #value "') is nil!",	\
		!lua_isnil(L, -1));					\
   } while(0)
#define loadkey_cfunc(index, value)		\
   load_key(index, value, lua_pushcfunction)
#define loadkey_user(index, value) do {					\
      lua_pushvalue(L, value); lua_gettable(L, index);			\
      mu_assert("value at userdata (index '" #value "') is nil!",	\
		!lua_isnil(L, -1));					\
   } while(0)
#define loadkey_light(index, value)			\
   load_key(index, value, lua_pushlightuserdata)

#define check_value(is_type, typestring, type, conversion, comparison)	\
   do {									\
      mu_assert("value is not of type " typestring, is_type(L, -1));	\
      type value = conversion(L, -1); lua_pop(L, 1);			\
      mu_assert("test " #comparison " failed!", (comparison));	\
      lua_pop(L, 1); /* remove key */					\
   } while(0)

#define checkval_bool(expected)			\
   check_value(lua_isboolean, "boolean", bool, lua_toboolean, value==expected)
#define checkval_int(expected)			\
   check_value(lua_isnumber, "integer", lua_Integer, lua_tointeger, value==expected)
#define checkval_num(expected)			\
   check_value(lua_isnumber, "number", lua_Number, lua_tonumber, value==expected)
#define checkval_str(expected)			\
   check_value(lua_isstring, "string", const char *, lua_tostring, strcmp(value, expected))
#define checkval_tbl(expected)			\
   check_value(lua_istable, "table", const void *, lua_topointer, value==lua_topointer(L, expected))
#define checkval_func(expected)			\
   check_value(lua_isfunction, "function", const void *, lua_topointer, value==lua_topointer(L, expected))
#define checkval_cfunc(expected)			\
   check_value(lua_iscfunction, "C function", lua_CFunction, lua_tocfunction, value==expected)
#define checkval_user(expected)			\
   check_value(lua_isuserdata, "userdata", void *, lua_touserdata, value==lua_topointer(L, expected))
#define checkval_light(expected)			\
   check_value(lua_islightuserdata, "light userdata", void *, lua_touserdata, value==expected)

#define store(name)					\
   lua_pushvalue(L, -1);				\
   int name ## _ref = luaL_ref(L, LUA_REGISTRYINDEX);	\
   int name = lua_gettop(L);

#define load(name)					\
   lua_rawgeti(L, LUA_REGISTRYINDEX, name ## _ref);	\
   name = lua_gettop(L);


/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 *
 * boolean keys
 *
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 */

TEST(create_table_bool_bool)
{
   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_bool_bool(true, true), hs_bool_bool(false, true));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_bool(index, true);
   checkval_bool(true);
   loadkey_bool(index, false);
   checkval_bool(true);
   return 0;
}

TEST(create_table_bool_int)
{
   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_bool_int(true, 15), hs_bool_int(false, 25));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_bool(index, true);
   checkval_int(15);
   loadkey_bool(index, false);
   checkval_int(25);
   return 0;
}

TEST(create_table_bool_num)
{
   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_bool_num(true, 2.718), hs_bool_num(false, 1.618));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_bool(index, true);
   checkval_num(2.718);
   loadkey_bool(index, false);
   checkval_num(1.618);
   return 0;
}

TEST(create_table_bool_str)
{
   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_bool_str(true, "hello"), hs_bool_str(false, "world"));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_bool(index, true);
   checkval_str("hello");
   loadkey_bool(index, false);
   checkval_str("world");
   return 0;
}

TEST(create_table_bool_tbl)
{
   int oldtop = lua_gettop(L);

   lua_newtable(L);
   store(value1);
   lua_newtable(L);
   store(value2);

   int index = hs_create_table
      (L, hs_bool_tbl(true, value1), hs_bool_tbl(false, value2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));

   load(value1);
   load(value2);
   
   loadkey_bool(index, true);
   checkval_tbl(value1);
   loadkey_bool(index, false);
   checkval_tbl(value2);
   return 0;
}

TEST(create_table_bool_func)
{
   int oldtop = lua_gettop(L);

   luaL_loadstring(L, "print('hello')");
   store(value1);
   luaL_loadstring(L, "print('hello')");
   store(value2);
   
   int index = hs_create_table
      (L, hs_bool_func(true, value1), hs_bool_func(false, value2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));

   load(value1);
   load(value2);
   
   loadkey_bool(index, true);
   checkval_func(value1);
   loadkey_bool(index, false);
   checkval_func(value2);
   return 0;
}

TEST(create_table_bool_cfunc)
{

   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_bool_cfunc(true, testfunc1), hs_bool_cfunc(false, testfunc2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_bool(index, true);
   checkval_cfunc(testfunc1);
   loadkey_bool(index, false);
   checkval_cfunc(testfunc2);
   return 0;
}


TEST(create_table_bool_user)
{
   int oldtop = lua_gettop(L);

   lua_newuserdata(L, sizeof(char));
   store(user1);
   lua_newuserdata(L, sizeof(char));
   store(user2);

   int index = hs_create_table
      (L, hs_bool_user(true, user1), hs_bool_user(false, user2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));

   load(user1);
   load(user2);
   
   loadkey_bool(index, true);
   checkval_user(user1);
   loadkey_bool(index, false);
   checkval_user(user2);
   return 0;
}

TEST(create_table_bool_light)
{
   int a1 = 5;
   void *light1 = (void *) &a1;
   int a2 = 6;
   void *light2 = (void *) &a2;

   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_bool_light(true, light1), hs_bool_light(false, light2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_bool(index, true);
   checkval_light(light1);
   loadkey_bool(index, false);
   checkval_light(light2);
   return 0;
}


/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 *
 * integer keys
 *
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 */

TEST(create_table_int_bool)
{
   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_int_bool(15, true), hs_int_bool(25, true));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_int(index, 15);
   checkval_bool(true);
   loadkey_int(index, 25);
   checkval_bool(true);
   return 0;
}

TEST(create_table_int_int)
{
   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_int_int(15, 15), hs_int_int(25, 25));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_int(index, 15);
   checkval_int(15);
   loadkey_int(index, 25);
   checkval_int(25);
   return 0;
}

TEST(create_table_int_num)
{
   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_int_num(15, 2.718), hs_int_num(25, 1.618));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_int(index, 15);
   checkval_num(2.718);
   loadkey_int(index, 25);
   checkval_num(1.618);
   return 0;
}

TEST(create_table_int_str)
{
   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_int_str(15, "hello"), hs_int_str(25, "world"));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_int(index, 15);
   checkval_str("hello");
   loadkey_int(index, 25);
   checkval_str("world");
   return 0;
}

TEST(create_table_int_tbl)
{
   int oldtop = lua_gettop(L);

   lua_newtable(L);
   store(value1);
   lua_newtable(L);
   store(value2);

   int index = hs_create_table
      (L, hs_int_tbl(15, value1), hs_int_tbl(25, value2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));

   load(value1);
   load(value2);
   
   loadkey_int(index, 15);
   checkval_tbl(value1);
   loadkey_int(index, 25);
   checkval_tbl(value2);
   return 0;
}

TEST(create_table_int_func)
{
   int oldtop = lua_gettop(L);

   luaL_loadstring(L, "print('hello')");
   store(value1);
   luaL_loadstring(L, "print('hello')");
   store(value2);
   
   int index = hs_create_table
      (L, hs_int_func(15, value1), hs_int_func(25, value2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));

   load(value1);
   load(value2);
   
   loadkey_int(index, 15);
   checkval_func(value1);
   loadkey_int(index, 25);
   checkval_func(value2);
   return 0;
}

TEST(create_table_int_cfunc)
{

   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_int_cfunc(15, testfunc1), hs_int_cfunc(25, testfunc2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_int(index, 15);
   checkval_cfunc(testfunc1);
   loadkey_int(index, 25);
   checkval_cfunc(testfunc2);
   return 0;
}


TEST(create_table_int_user)
{
   int oldtop = lua_gettop(L);

   lua_newuserdata(L, sizeof(char));
   store(user1);
   lua_newuserdata(L, sizeof(char));
   store(user2);

   int index = hs_create_table
      (L, hs_int_user(15, user1), hs_int_user(25, user2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));

   load(user1);
   load(user2);
   
   loadkey_int(index, 15);
   checkval_user(user1);
   loadkey_int(index, 25);
   checkval_user(user2);
   return 0;
}

TEST(create_table_int_light)
{
   int a1 = 5;
   void *light1 = (void *) &a1;
   int a2 = 6;
   void *light2 = (void *) &a2;

   int oldtop = lua_gettop(L);
   int index = hs_create_table
      (L, hs_int_light(15, light1), hs_int_light(25, light2));
   mu_assert_equal(oldtop + 1, lua_gettop(L));
   
   loadkey_int(index, 15);
   checkval_light(light1);
   loadkey_int(index, 25);
   checkval_light(light2);
   return 0;
}


/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 *
 * test suite
 *
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 */

void hs_create_table_tests()
{
   printf("running hs_create_table() tests...\n");
   
   mu_run_test("return correct stack index", table_correct_index);

   /* boolean keys */
   mu_run_test("create table with boolean keys and boolean values",
	       create_table_bool_bool);
   mu_run_test("create table with boolean keys and integer values", \
	       create_table_bool_int);
   mu_run_test("create table with boolean keys and number values", \
	       create_table_bool_num);
   mu_run_test("create table with boolean keys and string values", \
	       create_table_bool_str);
   mu_run_test("create table with boolean keys and table values",	\
	       create_table_bool_tbl);
   mu_run_test("create table with boolean keys and function values", \
	       create_table_bool_func);
   mu_run_test("create table with boolean keys and C function values", \
	       create_table_bool_cfunc);
   mu_run_test("create table with boolean keys and userdata values", \
	       create_table_bool_user);
   mu_run_test("create table with boolean keys and light userdata values",		\
	       create_table_bool_light);

   /* integer keys */
   mu_run_test("create table with integer keys and boolean values",
	       create_table_int_bool);
   mu_run_test("create table with integer keys and integer values", \
	       create_table_int_int);
   mu_run_test("create table with integer keys and number values", \
	       create_table_int_num);
   mu_run_test("create table with integer keys and string values", \
	       create_table_int_str);
   mu_run_test("create table with integer keys and table values",	\
	       create_table_int_tbl);
   mu_run_test("create table with integer keys and function values", \
	       create_table_int_func);
   mu_run_test("create table with integer keys and C function values", \
	       create_table_int_cfunc);
   mu_run_test("create table with integer keys and userdata values", \
	       create_table_int_user);
   mu_run_test("create table with integer keys and light userdata values",		\
	       create_table_int_light);

   /* number keys /
   mu_run_test("create table with number keys and boolean values",
	       create_table_num_bool);
   mu_run_test("create table with number keys and integer values", \
	       create_table_num_int);
   mu_run_test("create table with number keys and number values", \
	       create_table_num_num);
   mu_run_test("create table with number keys and string values", \
	       create_table_num_str);
   mu_run_test("create table with number keys and table values",	\
	       create_table_num_tbl);
   mu_run_test("create table with number keys and function values", \
	       create_table_num_func);
   mu_run_test("create table with number keys and C function values", \
	       create_table_num_cfunc);
   mu_run_test("create table with number keys and userdata values", \
	       create_table_num_user);
   mu_run_test("create table with number keys and light userdata values",		\
	       create_table_num_light);

   /* string keys /
   mu_run_test("create table with string keys and boolean values",
	       create_table_str_bool);
   mu_run_test("create table with string keys and integer values", \
	       create_table_str_int);
   mu_run_test("create table with string keys and number values", \
	       create_table_str_num);
   mu_run_test("create table with string keys and string values", \
	       create_table_str_str);
   mu_run_test("create table with string keys and table values",	\
	       create_table_str_tbl);
   mu_run_test("create table with string keys and function values", \
	       create_table_str_func);
   mu_run_test("create table with string keys and C function values", \
	       create_table_str_cfunc);
   mu_run_test("create table with string keys and userdata values", \
	       create_table_str_user);
   mu_run_test("create table with string keys and light userdata values",		\
	       create_table_str_light);

   /* table keys /
   mu_run_test("create table with table keys and boolean values",
	       create_table_tbl_bool);
   mu_run_test("create table with table keys and integer values", \
	       create_table_tbl_int);
   mu_run_test("create table with table keys and number values", \
	       create_table_tbl_num);
   mu_run_test("create table with table keys and string values", \
	       create_table_tbl_str);
   mu_run_test("create table with table keys and table values",	\
	       create_table_tbl_tbl);
   mu_run_test("create table with table keys and function values", \
	       create_table_tbl_func);
   mu_run_test("create table with table keys and C function values", \
	       create_table_tbl_cfunc);
   mu_run_test("create table with table keys and userdata values", \
	       create_table_tbl_user);
   mu_run_test("create table with table keys and light userdata values",		\
	       create_table_tbl_light);
	       /**/

}