#ifdef HAVE_CONFIG_H
# include <config.h>
#endif /* HAVE_CONFIG_H */

#include <check.h>
#include "Eina_Graph.h"

static void
_feed_simple_graph(Eina_Graph *g)
{
   ck_assert(eina_graph_edge_add(g, 0, 5) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 4, 3) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 0, 1) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 9, 12) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 6, 4) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 5, 4) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 0, 2) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 11, 12) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 9, 10) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 0, 6) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 7, 8) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 9, 11) == EINA_TRUE);
   ck_assert(eina_graph_edge_add(g, 5, 3) == EINA_TRUE);
}

START_TEST (test_eina_graph_simple)
{
   ck_assert(eina_graph_init() == 1);
   ck_assert(eina_graph_init() == 2);

   Eina_Graph *g = eina_graph_new(13, 2);
   ck_assert(g != NULL);

   _feed_simple_graph(g);

   ck_assert(eina_graph_edge_add(g, -1, 3) == EINA_FALSE);
   ck_assert(eina_graph_edge_add(g, 3, -1) == EINA_FALSE);
   ck_assert(eina_graph_edge_add(g, 13, 3) == EINA_FALSE);
   ck_assert(eina_graph_edge_add(g, 3, 13) == EINA_FALSE);

   ck_assert(eina_graph_vertices_count(g) == 13);
   ck_assert(eina_graph_edges_count(g) == 13);
   ck_assert(eina_graph_degree(g, 4) == 3);
   ck_assert(eina_graph_degree(g, 0) == 4);
   ck_assert(eina_graph_degree_max(g) == 4);
   ck_assert(eina_graph_degree_avg(g) == 2.0);
   ck_assert(eina_graph_self_loops(g) == 0);

   ck_assert(eina_graph_edge_add(g, 0, 0) == EINA_TRUE);

   ck_assert(eina_graph_vertices_count(g) == 13);
   ck_assert(eina_graph_edges_count(g) == 14);
   ck_assert(eina_graph_degree(g, 4) == 3);
   ck_assert(eina_graph_degree(g, 0) == 5);
   ck_assert(eina_graph_degree_max(g) == 5);
   ck_assert(eina_graph_degree_avg(g) > 2.153);
   ck_assert(eina_graph_degree_avg(g) < 2.154);
   ck_assert(eina_graph_self_loops(g) == 1);

   eina_graph_free(g);

   ck_assert(eina_graph_shutdown() == 1);
   ck_assert(eina_graph_shutdown() == 0);
}
END_TEST

Suite *
eina_graph_suite (void)
{
   Suite *s = suite_create ("Eina Graph");

   TCase *tc_simple = tcase_create ("Simple Graph");
   tcase_add_test (tc_simple, test_eina_graph_simple);
   suite_add_tcase (s, tc_simple);

   return s;
}

int
main(int argc EINA_UNUSED, char *argv[] EINA_UNUSED)
{
   int number_failed;
   Suite *s = eina_graph_suite();
   SRunner *sr = srunner_create(s);
   srunner_run_all(sr, CK_ENV);
   number_failed = srunner_ntests_failed(sr);
   srunner_free(sr);
   return ((number_failed == 0) ? 0: 1);
}