/*

 * 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 <locale.h>

#include <sys/stat.h>

#include <glib.h>

#include <gdk/gdk.h>

#include <gtk/gtk.h>

#include <plover/plover.h>

#include <plover-gtk/transactionhelper.h>

GtkBuilder *ui;

gboolean manual_mode=FALSE;

PloverTransactionHelper *get_transaction_helper(void)

{

    const char *dir;

    gchar *s;

    GError *err=NULL;

    PloverTransactionHelper *helper;

    dir=g_getenv("gtk_srcdir");

    s=g_build_filename(dir,"software-installation.ui",NULL);

    ui=gtk_builder_new();

    if (!gtk_builder_add_from_file(ui,s,&err))

	g_error("%s: %s",s,err->message);

    g_free(s);

    helper=plover_transaction_helper_new(ui);

    g_object_unref(ui);

    return helper;

}

static void test_init(void)

{

    PloverTransactionHelper *helper;

    helper=get_transaction_helper();

    g_object_unref(helper);

}

static void test_basic_properties(void)

{

    const char *prefix;

    GError *err=NULL;

    struct comps *comps;

    PloverTransactionHelper *helper;

    PloverPackageSet *installed;

    PloverRepository *upstream;

    upstream=plover_repository_new_from_yum("../yum-repo-test-dir",&err);

    if (!upstream)

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

    installed=plover_package_set_new_from_installed("../razor-test-dir",&err);

    if (!installed)

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

    helper=get_transaction_helper();

    g_assert(!plover_transaction_helper_get_visible(helper));

    plover_transaction_helper_set_installed(helper,installed);

    g_assert(plover_transaction_helper_get_installed(helper)==installed);

    plover_transaction_helper_set_upstream(helper,upstream);

    g_assert(plover_transaction_helper_get_upstream(helper,&err)==upstream);

    g_assert(!err);

    plover_transaction_helper_set_base(helper,"../yum-repo-test-dir");

    g_assert_cmpstr(plover_transaction_helper_get_base(helper),==,"../yum-repo-test-dir");

    comps=plover_transaction_helper_get_comps(helper,&err);

    g_assert(!err);

    g_assert(plover_comps_lookup_group(comps,"base"));

    prefix=plover_transaction_helper_get_prefix(helper,&err);

    g_assert(!err);

    g_assert_cmpstr(prefix,==,plover_default_prefix_for_vendor("Acme Corporation"));

    g_assert(!plover_transaction_helper_get_visible(helper));

    g_assert(!plover_transaction_helper_get_error(helper,NULL));

    g_object_unref(upstream);

    g_object_unref(installed);

    g_object_unref(helper);

}

static void test_install_group(void)

{

    gchar *root,*root_uri;

    GError *err=NULL;

    GFile *file;

    PloverPackageSet *installed;

    PloverTransactionHelper *helper;

    root=g_strdup("razor-test-dir-XXXXXX");

    g_assert(mkdtemp(root));

    file=g_file_new_for_path(root);

    g_free(root);

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    helper=get_transaction_helper();

    installed=plover_package_set_new_from_installed("../razor-test-dir",&err);

    if (!installed)

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

    plover_transaction_helper_set_installed(helper,installed);

    g_object_unref(installed);

    plover_transaction_helper_set_base(helper,"../yum-repo-test-dir");

    if (!plover_transaction_helper_install_group(helper,"base",&err))

	g_error("base: %s",err->message);

    g_assert(!err);

    g_object_unref(helper);

    g_unsetenv("RAZOR_ROOT");

}

static void test_remove_group(void)

{

    gchar *root,*root_uri;

    GError *err=NULL;

    GFile *file;

    PloverPackageSet *installed;

    PloverTransactionHelper *helper;

    struct plover_vector *packages;

    char *pkgs[]={"zip",NULL};

    root=g_strdup("razor-test-dir-XXXXXX");

    g_assert(mkdtemp(root));

    file=g_file_new_for_path(root);

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    helper=get_transaction_helper();

    plover_transaction_helper_set_base(helper,"../yum-repo-test-dir");

    packages=plover_transaction_helper_group_get_default_packages(helper,

      "zappy",&err);

    if (!packages)

	g_error("zappy: %s",err->message);

    if (!plover_install("../yum-repo-test-dir",NULL,packages->strings,&err))

	g_error("plover_install: %s",err->message);

    plover_vector_free(packages);

    installed=plover_package_set_new_from_installed(root,&err);

    if (!installed)

	g_error("%s: %s",root,err->message);

    plover_transaction_helper_set_installed(helper,installed);

    g_object_unref(installed);

    if (!plover_transaction_helper_remove_group(helper,"zappy",&err))

	g_error("zappy: %s",err->message);

    g_assert(!err);

    g_object_unref(helper);

    g_unsetenv("RAZOR_ROOT");

    g_free(root);

}

static void test_update(void)

{

    gchar *root,*root_uri;

    GError *err=NULL;

    GFile *file;

    PloverPackageSet *installed;

    PloverTransactionHelper *helper;

    root=g_strdup("razor-test-dir-XXXXXX");

    g_assert(mkdtemp(root));

    file=g_file_new_for_path(root);

    g_free(root);

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    helper=get_transaction_helper();

    installed=plover_package_set_new_from_installed("../razor-test-dir",&err);

    if (!installed)

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

    plover_transaction_helper_set_installed(helper,installed);

    g_object_unref(installed);

    plover_transaction_helper_set_base(helper,"../yum-repo-test-dir");

    if (plover_transaction_helper_update(helper,&err))

	g_error("plover_transaction_helper_update reports work to be done");

    g_assert_error(err,PLOVER_GENERAL_ERROR,PLOVER_GENERAL_ERROR_NO_WORK);

    g_object_unref(helper);

    g_unsetenv("RAZOR_ROOT");

}

struct run_install_baton {

    enum {

	RI_STATE_INIT = 0,

	RI_STATE_SUMMARY,

	RI_STATE_PROGRESS,

	RI_STATE_PROGRESS_DELAY,

	RI_STATE_DONE,

	RI_STATE_FINISH

    } state;

    guint eid;		/* event ID (or 0) */

    PloverTransactionHelper *helper;

};

gboolean run_install_tick(gpointer data)

{

    gboolean retval=TRUE;

    struct run_install_baton *baton=data;

    GtkWidget *page;

    GtkAssistant *assistant=baton->helper->assistant;

    switch(baton->state)

    {

	case RI_STATE_INIT:

	    if (!assistant || !gtk_widget_get_visible(GTK_WIDGET(assistant)))

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->forward));

	    break;

	case RI_STATE_SUMMARY:

	    if (gtk_assistant_get_current_page(assistant)<1)

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->apply));

	    break;

	case RI_STATE_PROGRESS:

	    if (gtk_assistant_get_current_page(assistant)<2)

		return TRUE;

	    page=gtk_assistant_get_nth_page(assistant,2);

	    baton->eid=g_timeout_add_seconds(1,run_install_tick,baton);

	    if (!gtk_assistant_get_page_complete(assistant,page))

		return FALSE;

	    else

		retval=FALSE;

	    break;

	case RI_STATE_PROGRESS_DELAY:

	    retval=FALSE;

	    baton->eid=g_idle_add_full(G_PRIORITY_LOW,run_install_tick,baton,

	      NULL);

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->forward));

	    break;

	case RI_STATE_DONE:

	    if (gtk_assistant_get_current_page(assistant)<3)

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->close));

	    break;

	case RI_STATE_FINISH:

	    if (assistant && gtk_widget_get_visible(GTK_WIDGET(assistant)))

		return TRUE;

	    gtk_main_quit();

	    baton->eid=0;

	    return FALSE;

    }

    baton->state++;

    return retval;

}

static void test_run_install(void)

{

    gchar *root,*root_uri;

    GError *err=NULL;

    GFile *file;

    struct plover_vector *packages;

    PloverPackageSet *installed;

    PloverTransactionHelper *helper;

    struct run_install_baton baton={0,};

    root=g_strdup("razor-test-dir-XXXXXX");

    g_assert(mkdtemp(root));

    file=g_file_new_for_path(root);

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    helper=get_transaction_helper();

    installed=plover_package_set_new();

    if (!plover_package_set_open(installed,root,TRUE,&err))

	g_error("%s: %s",root,err->message);

    plover_transaction_helper_set_installed(helper,installed);

    g_object_unref(installed);

    plover_transaction_helper_set_base(helper,"../yum-repo-test-dir");

    packages=plover_vector_new();

    plover_vector_append(packages,"zappy-tools");

    if (!plover_transaction_helper_install_packages(helper,packages,&err))

	g_error("zappy-tools: %s",err->message);

    g_assert(!err);

    plover_vector_free(packages);

    plover_transaction_helper_present(helper);

    baton.helper=helper;

    baton.eid=g_idle_add_full(G_PRIORITY_LOW,run_install_tick,&baton,NULL);

    gtk_main();

    g_object_unref(helper);

    g_unsetenv("RAZOR_ROOT");

    g_free(root);

}

struct run_remove_baton {

    enum {

	RR_STATE_INIT = 0,

	RR_STATE_SUMMARY,

	RR_STATE_PROGRESS,

	RR_STATE_PROGRESS_DELAY,

	RR_STATE_DONE,

	RR_STATE_FINISH

    } state;

    guint eid;		/* event ID (or 0) */

    PloverTransactionHelper *helper;

};

gboolean run_remove_tick(gpointer data)

{

    gboolean retval=TRUE;

    struct run_remove_baton *baton=data;

    GtkWidget *page;

    GtkAssistant *assistant=baton->helper->assistant;

    switch(baton->state)

    {

	case RR_STATE_INIT:

	    if (!assistant || !gtk_widget_get_visible(GTK_WIDGET(assistant)))

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->forward));

	    break;

	case RR_STATE_SUMMARY:

	    if (gtk_assistant_get_current_page(assistant)<1)

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->apply));

	    break;

	case RR_STATE_PROGRESS:

	    if (gtk_assistant_get_current_page(assistant)<2)

		return TRUE;

	    page=gtk_assistant_get_nth_page(assistant,2);

	    baton->eid=g_timeout_add_seconds(1,run_remove_tick,baton);

	    if (!gtk_assistant_get_page_complete(assistant,page))

		return FALSE;

	    else

		retval=FALSE;

	    break;

	case RR_STATE_PROGRESS_DELAY:

	    retval=FALSE;

	    baton->eid=g_idle_add_full(G_PRIORITY_LOW,run_remove_tick,baton,

	      NULL);

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->forward));

	    break;

	case RR_STATE_DONE:

	    if (gtk_assistant_get_current_page(assistant)<3)

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->close));

	    break;

	case RR_STATE_FINISH:

	    if (assistant && gtk_widget_get_visible(GTK_WIDGET(assistant)))

		return TRUE;

	    gtk_main_quit();

	    baton->eid=0;

	    return FALSE;

    }

    baton->state++;

    return retval;

}

static void test_run_remove(void)

{

    gchar *root,*root_uri;

    GError *err=NULL;

    GFile *file;

    struct plover_vector *packages;

    PloverPackageSet *installed;

    PloverTransactionHelper *helper;

    struct run_remove_baton baton={0,};

    root=g_strdup("razor-test-dir-XXXXXX");

    g_assert(mkdtemp(root));

    file=g_file_new_for_path(root);

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    helper=get_transaction_helper();

    plover_transaction_helper_set_base(helper,"../yum-repo-test-dir");

    packages=

      plover_transaction_helper_group_get_default_packages(helper,"zappy",&err);

    if (!packages)

	g_error("zappy: %s",err->message);

    if (!plover_install("../yum-repo-test-dir",NULL,packages->strings,&err))

	g_error("plover_install: %s",err->message);

    plover_vector_free(packages);

    installed=plover_package_set_new();

    if (!plover_package_set_open(installed,root,TRUE,&err))

	g_error("%s: %s",root,err->message);

    plover_transaction_helper_set_installed(helper,installed);

    g_object_unref(installed);

    if (!plover_transaction_helper_remove_group(helper,"zappy",&err))

	g_error("zappy: %s",err->message);

    g_assert(!err);

    plover_transaction_helper_present(helper);

    baton.helper=helper;

    baton.eid=g_idle_add_full(G_PRIORITY_LOW,run_remove_tick,&baton,NULL);

    gtk_main();

    g_object_unref(helper);

    g_unsetenv("RAZOR_ROOT");

    g_free(root);

}

struct run_update_baton {

    enum {

	RU_STATE_INIT = 0,

	RU_STATE_SUMMARY,

	RU_STATE_PROGRESS,

	RU_STATE_PROGRESS_DELAY,

	RU_STATE_DONE,

	RU_STATE_FINISH

    } state;

    guint eid;		/* event ID (or 0) */

    PloverTransactionHelper *helper;

};

gboolean run_update_tick(gpointer data)

{

    gboolean retval=TRUE;

    struct run_update_baton *baton=data;

    GtkWidget *page;

    GtkAssistant *assistant=baton->helper->assistant;

    switch(baton->state)

    {

	case RU_STATE_INIT:

	    if (!assistant || !gtk_widget_get_visible(GTK_WIDGET(assistant)))

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->forward));

	    break;

	case RU_STATE_SUMMARY:

	    if (gtk_assistant_get_current_page(assistant)<1)

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->apply));

	    break;

	case RU_STATE_PROGRESS:

	    if (gtk_assistant_get_current_page(assistant)<2)

		return TRUE;

	    page=gtk_assistant_get_nth_page(assistant,2);

	    baton->eid=g_timeout_add_seconds(1,run_update_tick,baton);

	    if (!gtk_assistant_get_page_complete(assistant,page))

		return FALSE;

	    else

		retval=FALSE;

	    break;

	case RU_STATE_PROGRESS_DELAY:

	    retval=FALSE;

	    baton->eid=g_idle_add_full(G_PRIORITY_LOW,run_update_tick,baton,

	      NULL);

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->forward));

	    break;

	case RU_STATE_DONE:

	    if (gtk_assistant_get_current_page(assistant)<3)

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->close));

	    break;

	case RU_STATE_FINISH:

	    if (assistant && gtk_widget_get_visible(GTK_WIDGET(assistant)))

		return TRUE;

	    gtk_main_quit();

	    baton->eid=0;

	    return FALSE;

    }

    baton->state++;

    return retval;

}

static void test_run_update(void)

{

    gchar *root,*root_uri;

    GError *err=NULL;

    GFile *file;

    struct razor_importer *importer;

    struct razor_set *downgraded;

    struct razor_atomic *atomic;

    struct plover_vector *packages;

    PloverPackageSet *installed;

    PloverTransactionHelper *helper;

    struct run_update_baton baton={0,};

    root=g_strdup("razor-test-dir-XXXXXX");

    g_assert(mkdtemp(root));

    file=g_file_new_for_path(root);

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    helper=get_transaction_helper();

    plover_transaction_helper_set_base(helper,"../yum-repo-test-dir");

    installed=plover_package_set_new();

    if (!plover_package_set_open(installed,root,TRUE,&err))

	g_error("%s: %s",root,err->message);

    importer=razor_importer_create();

    razor_importer_begin_package(importer,"zappy","0-1","noarch");

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

    razor_importer_add_property(importer,"zappy",RAZOR_PROPERTY_PROVIDES,"0-1");

    razor_importer_finish_package(importer);

    downgraded=razor_importer_finish(importer);

    atomic=razor_atomic_open("Add downgraded packages");

    if (!plover_package_set_update(installed,downgraded,atomic) ||

      razor_atomic_commit(atomic))

	g_error("%s: %s",root,razor_atomic_get_error_msg(atomic));

    razor_atomic_destroy(atomic);

    razor_set_unref(downgraded);

    plover_transaction_helper_set_installed(helper,installed);

    g_object_unref(installed);

    if (!plover_transaction_helper_update(helper,&err))

	g_error("update: %s",err->message);

    g_assert(!err);

    plover_transaction_helper_present(helper);

    baton.helper=helper;

    baton.eid=g_idle_add_full(G_PRIORITY_LOW,run_update_tick,&baton,NULL);

    gtk_main();

    g_object_unref(helper);

    g_unsetenv("RAZOR_ROOT");

    g_free(root);

}

struct check_vendor_baton {

    enum {

	CV_STATE_INIT = 0,

	CV_STATE_SUMMARY,

	CV_STATE_PROGRESS,

	CV_STATE_PROGRESS_DELAY,

	CV_STATE_DONE,

	CV_STATE_FINISH

    } state;

    guint eid;		/* event ID (or 0) */

    PloverTransactionHelper *helper;

};

gboolean check_vendor_tick(gpointer data)

{

    gboolean retval=TRUE;

    struct check_vendor_baton *baton=data;

    GtkWidget *page,*w;

    GtkAssistant *assistant=baton->helper->assistant;

    switch(baton->state)

    {

	case CV_STATE_INIT:

	    if (!assistant || !gtk_widget_get_visible(GTK_WIDGET(assistant)))

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->forward));

	    break;

	case CV_STATE_SUMMARY:

	    if (gtk_assistant_get_current_page(assistant)<1)

		return TRUE;

	    g_assert(!gtk_widget_is_sensitive(assistant->apply));

	    w=GTK_WIDGET(gtk_builder_get_object(ui,"SIRemoveExisting"));

	    gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(w),TRUE);

	    g_assert(gtk_widget_is_sensitive(assistant->apply));

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->apply));

	    break;

	case CV_STATE_PROGRESS:

	    if (gtk_assistant_get_current_page(assistant)<2)

		return TRUE;

	    page=gtk_assistant_get_nth_page(assistant,2);

	    baton->eid=g_timeout_add_seconds(1,check_vendor_tick,baton);

	    if (!gtk_assistant_get_page_complete(assistant,page))

		return FALSE;

	    else

		retval=FALSE;

	    break;

	case CV_STATE_PROGRESS_DELAY:

	    retval=FALSE;

	    baton->eid=g_idle_add_full(G_PRIORITY_LOW,check_vendor_tick,baton,

	      NULL);

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->forward));

	    break;

	case CV_STATE_DONE:

	    if (gtk_assistant_get_current_page(assistant)<3)

		return TRUE;

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(assistant->close));

	    break;

	case CV_STATE_FINISH:

	    if (assistant && gtk_widget_get_visible(GTK_WIDGET(assistant)))

		return TRUE;

	    gtk_main_quit();

	    baton->eid=0;

	    return FALSE;

    }

    baton->state++;

    return retval;

}

static void test_check_vendor(void)

{

    int fh;

    gchar *root,*root_uri,*s;

    GFile *file;

    GError *err=NULL;

    struct razor_importer *importer;

    struct razor_set *downgraded;

    struct razor_atomic *atomic;

    struct plover_vector *packages;

    PloverPackageSet *installed;

    PloverTransactionHelper *helper;

    struct run_update_baton baton={0,};

    g_setenv("PLOVER_VENDOR_PREFIX","/srv",TRUE);

    root=g_strdup("razor-test-dir-XXXXXX");

    g_assert(mkdtemp(root));

    file=g_file_new_for_path(root);

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    helper=get_transaction_helper();

    plover_transaction_helper_set_check_vendor(helper,TRUE);

    plover_transaction_helper_set_base(helper,"../yum-repo-test-dir");

    installed=plover_package_set_new();

    if (!plover_package_set_open(installed,root,TRUE,&err))

	g_error("%s: %s",root,err->message);

    importer=razor_importer_create();

    razor_importer_begin_package(importer,"zappy","0-1","noarch");

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

    razor_importer_add_property(importer,"zappy",RAZOR_PROPERTY_PROVIDES,"0-1");

    razor_importer_add_install_prefix(importer,"/test");

    razor_importer_add_file(importer,"/test/bin/zappy");

    razor_importer_finish_package(importer);

    downgraded=razor_importer_finish(importer);

    atomic=razor_atomic_open("Add downgraded packages");

    razor_atomic_make_dirs(atomic,root_uri,"/test/bin/zappy");

    s=g_build_filename(root_uri,"test/bin/zappy",NULL);

    fh=razor_atomic_create_file(atomic,s,S_IRWXU|S_IRWXG|S_IRWXO);

    g_free(s);

    razor_atomic_close(atomic,fh);

    if (!plover_package_set_update(installed,downgraded,atomic) ||

      razor_atomic_commit(atomic))

	g_error("%s: %s",root,razor_atomic_get_error_msg(atomic));

    razor_atomic_destroy(atomic);

    razor_set_unref(downgraded);

    plover_transaction_helper_set_installed(helper,installed);

    g_object_unref(installed);

    if (!plover_transaction_helper_update(helper,&err))

	g_error("update: %s",err->message);

    g_assert(!err);

    plover_transaction_helper_present(helper);

    baton.helper=helper;

    baton.eid=g_idle_add_full(G_PRIORITY_LOW,check_vendor_tick,&baton,NULL);

    gtk_main();

    g_object_unref(helper);

    g_unsetenv("RAZOR_ROOT");

    g_free(root);

    g_free(root_uri);

    g_unsetenv("PLOVER_VENDOR_PREFIX");

}

struct set_error_baton {

    enum {

	SE_STATE_INIT = 0,

	SE_STATE_FINISH

    } state;

    PloverTransactionHelper *helper;

};

/*

 * This handler may be called as either an event (ie., idle or timeout)

 * or as a (swapped) signal. In the latter case, the return is ignored.

 */

gboolean set_error_tick(gpointer data)

{

    struct set_error_baton *baton=data;

    GtkDialog *dlg;

    GtkWidget *button;

    dlg=GTK_DIALOG(baton->helper->error_dialog);

    switch(baton->state)

    {

	case SE_STATE_INIT:

	    if (!dlg || !gtk_widget_get_visible(GTK_WIDGET(dlg)))

		return TRUE;

	    button=gtk_dialog_get_widget_for_response(dlg,GTK_RESPONSE_CLOSE);

	    if (!manual_mode)

		gtk_button_clicked(GTK_BUTTON(button));

	    break;

	case SE_STATE_FINISH:

	    if (dlg && gtk_widget_get_visible(GTK_WIDGET(dlg)))

		return TRUE;

	    gtk_main_quit();

	    return FALSE;

    }

    baton->state++;

    return TRUE;

}

static void test_set_error(void)

{

    gchar *root,*root_uri;

    const char *errmsg;

    GError *err=NULL;

    const GError *err2=NULL;

    GFile *file;

    PloverPackageSet *installed;

    PloverTransactionHelper *helper;

    struct set_error_baton baton={0,};

    root=g_strdup("razor-test-dir-XXXXXX");

    g_assert(mkdtemp(root));

    file=g_file_new_for_path(root);

    g_free(root);

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    helper=get_transaction_helper();

    installed=plover_package_set_new_from_installed("../razor-test-dir",&err);

    if (!installed)

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

    plover_transaction_helper_set_installed(helper,installed);

    g_object_unref(installed);

    plover_transaction_helper_set_base(helper,"../yum-repo-test-dir");

    if (!plover_transaction_helper_update(helper,&err))

	plover_transaction_helper_set_error(helper,err,"Expected error");

    g_assert(plover_transaction_helper_get_visible(helper));

    errmsg=plover_transaction_helper_get_error(helper,&err2);

    g_assert_cmpstr(errmsg,==,"Expected error");

    g_assert_error(err2,err->domain,err->code);

    plover_transaction_helper_present(helper);

    baton.helper=helper;

    g_idle_add_full(G_PRIORITY_LOW,set_error_tick,&baton,NULL);

    g_error_free(err);

    gtk_main();

    g_object_unref(helper);

    g_unsetenv("RAZOR_ROOT");

}

int main(int argc,char **argv)

{

    int retval;

    GError *err=NULL;

    /*

     * Note that because g_test_init() handles --help,

     * these options will not appear in the output.

     */

    GOptionEntry options[]={

	{"manual",0,0,G_OPTION_ARG_NONE,&manual_mode,

	  "Disable automatic mode",NULL},

	{NULL}

    };

    g_test_init(&argc,&argv,NULL);

    g_setenv("GTK_MODULES","",TRUE);

    g_setenv("GTK2_RC_FILES","/dev/null",TRUE);

    gtk_disable_setlocale();

    setlocale(LC_ALL,"C");

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

    if (!gtk_init_with_args(&argc,&argv,NULL,options,NULL,&err))

    {

	g_printf("%s\n",err->message);

	exit(0);

    }

    g_test_add_func("/transactionhelper/init",test_init);

    g_test_add_func("/transactionhelper/basic-properties",

      test_basic_properties);

    g_test_add_func("/transactionhelper/install-group",test_install_group);

    g_test_add_func("/transactionhelper/remove-group",test_remove_group);

    g_test_add_func("/transactionhelper/update",test_update);

    g_test_add_func("/transactionhelper/run-install",test_run_install);

    g_test_add_func("/transactionhelper/run-remove",test_run_remove);

    g_test_add_func("/transactionhelper/run-update",test_run_update);

    g_test_add_func("/transactionhelper/check-vendor",test_check_vendor);

    g_test_add_func("/transactionhelper/set-error",test_set_error);

    retval=g_test_run();

    return retval;

}