/*

  * Copyright (C) 2011, 2016  J. Ali Harlow <ali@juiblex.co.uk>

  *

  * 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 2 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, write to the Free Software Foundation, Inc.,

  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

  */

 #include <stdlib.h>

 #include <string.h>

 #include <glib.h>

 #include <gtk/gtk.h>

 static void test_tree_model_iter(GtkTreeModel *model,GtkTreeIter *iter)

 {

     GtkTreePath *path,*path2;

     GtkTreeIter iter2,child;

     gchar *string,*string2,*s;

     int i,n;

     if (iter)   /* Not all tests make sense on the virtual root */

     {

 	/* Check we can convert to a path and back */

 	path=gtk_tree_model_get_path(model,iter);

 	g_assert(path!=NULL);

 	g_assert(gtk_tree_model_get_iter(model,&iter2,path));

 	path2=gtk_tree_model_get_path(model,&iter2);

 	g_assert(path2!=NULL);

 	g_assert(gtk_tree_path_compare(path,path2)==0);

 	gtk_tree_path_free(path2);

 	/* Check we can convert to a path string and back */

 	string=gtk_tree_model_get_string_from_iter(model,iter);

 	g_assert(string!=NULL);

 	g_assert(gtk_tree_model_get_iter_from_string(model,&iter2,string));

 	string2=gtk_tree_model_get_string_from_iter(model,&iter2);

 	g_assert(string2!=NULL);

 	g_assert(strcmp(string,string2)==0);

 	g_free(string2);

     }

     else

     {

 	path=NULL;

 	string=NULL;

     }

     if (string)

 	g_debug("Checking iter %s",string);

     else

 	g_debug("Checking virtual root iter");

     n=gtk_tree_model_iter_n_children(model,iter);

     g_assert(n>=0);

     if (n>0)

     {

 	/* Check that GTK_TREE_MODEL_LIST_ONLY is not set inappropriately */

 	if (iter)

 	{

 	    gboolean list_only_with_grandchildren;

 	    list_only_with_grandchildren=

 	      gtk_tree_model_get_flags(model)&GTK_TREE_MODEL_LIST_ONLY;

 	    g_assert(!list_only_with_grandchildren);

 	}

 	/* Check that gtk_tree_model_iter_has_child() agrees */

 	if (iter)

 	    g_assert(gtk_tree_model_iter_has_child(model,iter));

 	/* Check that gtk_tree_model_iter_children() returns the first child */

 	g_assert(gtk_tree_model_iter_children(model,&child,iter));

 	string2=gtk_tree_model_get_string_from_iter(model,&child);

 	g_assert(string2!=NULL);

 	if (string)

 	{

 	    g_assert(g_str_has_prefix(string2,string));

 	    g_assert(strlen(string2)==strlen(string)+2);

 	    g_assert(g_str_has_suffix(string2,":0"));

 	}

 	else

 	    g_assert(strcmp(string2,"0")==0);

 	g_free(string2);

 	for(i=0;i<n;i++)

 	{

 	    /*

 	     * Check that gtk_tree_model_iter_nth_child() returns the nth child

 	     */

 	    g_assert(gtk_tree_model_iter_nth_child(model,&child,iter,i));

 	    string2=gtk_tree_model_get_string_from_iter(model,&child);

 	    g_assert(string2!=NULL);

 	    if (string)

 		s=g_strdup_printf("%s:%d",string,i);

 	    else

 		s=g_strdup_printf("%d",i);

 	    g_assert(strcmp(string2,s)==0);

 	    g_free(s);

 	    g_free(string2);

 	    /* Check that my son's father is me */

 	    if (iter)

 	    {

 		g_assert(gtk_tree_model_iter_parent(model,&iter2,&child));

 		string2=gtk_tree_model_get_string_from_iter(model,&iter2);

 		g_assert(strcmp(string,string2)==0);

 		g_free(string2);

 	    }

 	    else

 		g_assert(!gtk_tree_model_iter_parent(model,&iter2,&child));

 	    /* Check next sibling */

 	    if (i+1<n)

 	    {

 		iter2=child;

 		g_assert(gtk_tree_model_iter_next(model,&iter2));

 		string2=gtk_tree_model_get_string_from_iter(model,&iter2);

 		g_assert(string2!=NULL);

 		if (string)

 		s=g_strdup_printf("%s:%d",string,i+1);

 		    else

 		s=g_strdup_printf("%d",i+1);

 		g_assert(strcmp(string2,s)==0);

 		g_free(s);

 		g_free(string2);

 	    }

 	    else

 	    {

 		iter2=child;

 		g_assert(!gtk_tree_model_iter_next(model,&iter2));

 	    }

 	    /* Check child */

 	    test_tree_model_iter(model,&child);

 	}

 	/*

 	 * Check that gtk_tree_model_iter_nth_child() returns no more children

 	 */

 	g_assert(!gtk_tree_model_iter_nth_child(model,&child,iter,n));

     }

     else

     {

 	/* Check that all APIs agree that iter has no children */

 	if (iter)

 	    g_assert(!gtk_tree_model_iter_has_child(model,iter));

 	g_assert(!gtk_tree_model_iter_children(model,&child,iter));

 	g_assert(!gtk_tree_model_iter_nth_child(model,&child,iter,0));

     }

     gtk_tree_path_free(path);

     g_free(string);

 }

 void test_tree_model(GtkTreeModel *model)

 {

     GtkTreeModelFlags undefined_flags;

     undefined_flags=gtk_tree_model_get_flags(model);

     undefined_flags&=~GTK_TREE_MODEL_ITERS_PERSIST;

     undefined_flags&=~GTK_TREE_MODEL_LIST_ONLY;

     g_assert(!undefined_flags);

     test_tree_model_iter(model,NULL);

 }