/*

  * Copyright (C) 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 <gdk/gdk.h>

 #include <gtk/gtk.h>

 #include <razor/razor.h>

 #include <plover/package.h>

 #include <plover/packageset.h>

 #include <plover-gtk/packagestore.h>

 #include "treemodel.h"

 static void verify_package_store(PloverPackageStore *store,

   GSList *expected_packages)

 {

     int i;

     const char *s;

     gchar *nosummary;

     PloverPackage *package;

     PloverPackageSet *package_set;

     GtkTreeModel *model;

     GtkTreeIter iter;

     GValue value={0,};

     GSList *expected,*lnk;

     GdkPixbuf *icon;

     model=GTK_TREE_MODEL(store);

     test_tree_model(model);

     expected=g_slist_copy(expected_packages);

     g_assert_cmpint(gtk_tree_model_get_n_columns(model),==,PLOVER_PACKAGE_STORE_NO_COLUMNS);

     g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_OBJ_COLUMN),==,PLOVER_TYPE_PACKAGE);

     g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_INSTALLED_COLUMN),==,G_TYPE_BOOLEAN);

     g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_ICON_COLUMN),==,GDK_TYPE_PIXBUF);

     g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_NAME_COLUMN),==,G_TYPE_STRING);

     g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_VERSION_COLUMN),==,G_TYPE_STRING);

     g_assert_cmpint(gtk_tree_model_get_column_type(model,PLOVER_PACKAGE_STORE_SUMMARY_COLUMN),==,G_TYPE_STRING);

     for(i=0;;i++)

     {

 	if (!gtk_tree_model_iter_nth_child(model,&iter,NULL,i))

 	    break;

 	gtk_tree_model_get_value(model,&iter,PLOVER_PACKAGE_STORE_OBJ_COLUMN,

 	  &value);

 	package=g_value_dup_object(&value);

 	lnk=g_slist_find(expected,package);

 	if (!lnk)

 	    g_warning("Unexpected package in store: %s",

 	      plover_package_get_name(package));

 	else

 	{

 	    g_object_unref(lnk->data);

 	    expected=g_slist_delete_link(expected,lnk);

 	}

 	g_value_unset(&value);

 	gtk_tree_model_get_value(model,&iter,

 	  PLOVER_PACKAGE_STORE_INSTALLED_COLUMN,&value);

 	g_assert_cmpint(G_VALUE_TYPE(&value),==,G_TYPE_BOOLEAN);

 	/* Can't check value of installed yet, it

 	 * isn't set (or even properly defined).

 	 */

 	g_value_unset(&value);

 	gtk_tree_model_get_value(model,&iter,

 	  PLOVER_PACKAGE_STORE_ICON_COLUMN,&value);

 	icon=g_value_get_object(&value);

 	if (icon)

 	    g_assert(GDK_IS_PIXBUF(icon));

 	g_value_unset(&value);

 	gtk_tree_model_get_value(model,&iter,PLOVER_PACKAGE_STORE_NAME_COLUMN,

 	  &value);

 	s=g_value_get_string(&value);

 	g_assert_cmpstr(plover_package_get_name(package),==,s);

 	g_value_unset(&value);

 	gtk_tree_model_get_value(model,&iter,

 	  PLOVER_PACKAGE_STORE_VERSION_COLUMN,&value);

 	s=g_value_get_string(&value);

 	g_assert_cmpstr(plover_package_get_version(package),==,s);

 	g_value_unset(&value);

 	gtk_tree_model_get_value(model,&iter,

 	  PLOVER_PACKAGE_STORE_SUMMARY_COLUMN,&value);

 	s=g_value_get_string(&value);

 	if (*plover_package_get_summary(package))

 	    g_assert_cmpstr(plover_package_get_summary(package),==,s);

 	else

 	{

 	    nosummary=g_strconcat("The ",plover_package_get_name(package),

 	      " package",NULL);

 	    g_assert_cmpstr(nosummary,==,s);

 	    g_free(nosummary);

 	}

 	g_value_unset(&value);

     }

     if (expected)

     {

 	package=PLOVER_PACKAGE(expected->data);

 	g_warning("%d missing package%s in store, including %s",

 	  g_slist_length(expected),g_slist_length(expected)==1?"":"s",

 	  plover_package_get_name(package));

     }

 }

 static void test_empty(void)

 {

     PloverPackageStore *store;

     store=plover_package_store_new();

     g_assert(!plover_package_store_get_sets(store));

     verify_package_store(store,NULL);

     g_object_unref(store);

 }

 static void test_basic(void)

 {

     PloverPackageSet *package_set;

     PloverPackageStore *store;

     GError *err=NULL;

     GSList *expected;

     package_set=plover_package_set_new_from_yum("../yum-repo-test-dir",NULL,

       &err);

     if (!package_set)

 	g_error("../yum-repo-test-dir: %s",err->message);

     store=plover_package_store_new();

     plover_package_store_add_set(store,package_set);

     expected=g_slist_copy(plover_package_set_get_packages(package_set));

     g_slist_foreach(expected,(GFunc)g_object_ref,NULL);

     g_object_unref(package_set);

     verify_package_store(store,expected);

     g_slist_foreach(expected,(GFunc)g_object_unref,NULL);

     g_slist_free(expected);

     g_object_unref(store);

 }

 static PloverPackageSet *no_details_new(void)

 {

     struct razor_set *set;

     struct razor_importer *importer;

     PloverPackageSet *package_set;

     importer=razor_importer_create();

     razor_importer_begin_package(importer,"no-details","1-1","noarch");

     razor_importer_add_details(importer,"","","","");

     razor_importer_add_property(importer,"no-details",

       RAZOR_PROPERTY_PROVIDES|RAZOR_PROPERTY_EQUAL,"1-1");

     razor_importer_finish_package(importer);

     set=razor_importer_finish(importer);

     package_set=plover_package_set_new_from_razor(set);

     razor_set_unref(set);

     return package_set;

 }

 static void test_no_details(void)

 {

     PloverPackageSet *package_set;

     PloverPackageStore *store;

     GSList *expected;

     package_set=no_details_new();

     store=plover_package_store_new();

     plover_package_store_add_set(store,package_set);

     expected=g_slist_copy(plover_package_set_get_packages(package_set));

     g_slist_foreach(expected,(GFunc)g_object_ref,NULL);

     g_object_unref(package_set);

     verify_package_store(store,expected);

     g_slist_foreach(expected,(GFunc)g_object_unref,NULL);

     g_slist_free(expected);

     g_object_unref(store);

 }

 static void test_remove(void)

 {

     PloverPackageSet *yum_set,*nodetails_set;

     PloverPackageStore *store;

     GSList *expected;

     GError *err=NULL;

     store=plover_package_store_new();

     yum_set=plover_package_set_new_from_yum("../yum-repo-test-dir",NULL,&err);

     if (!yum_set)

 	g_error("../yum-repo-test-dir: %s",err->message);

     plover_package_store_add_set(store,yum_set);

     nodetails_set=no_details_new();

     expected=g_slist_copy(plover_package_set_get_packages(nodetails_set));

     g_slist_foreach(expected,(GFunc)g_object_ref,NULL);

     plover_package_store_add_set(store,nodetails_set);

     plover_package_store_remove_set(store,yum_set);

     g_object_unref(nodetails_set);

     g_object_unref(yum_set);

     verify_package_store(store,expected);

     g_slist_foreach(expected,(GFunc)g_object_unref,NULL);

     g_slist_free(expected);

     g_object_unref(store);

 }

 int main(int argc,char **argv)

 {

     int retval;

     gtk_test_init(&argc,&argv,NULL);

     g_test_bug_base("mailto:ali@juiblex.co.uk");

     g_test_add_func("/packagestore/empty",test_empty);

     g_test_add_func("/packagestore/basic",test_basic);

     g_test_add_func("/packagestore/no-details",test_no_details);

     g_test_add_func("/packagestore/remove",test_remove);

     retval=g_test_run();

     return retval;

 }