/*

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

#include <lua.h>

#include <razor.h>

#include <plover/plover.h>

#include <plover/packageset.h>

#include <plover/transaction.h>

LUALIB_API int luaopen_posix(lua_State *L);

const char *zappy_packages[]={

    "zap","zappy","zappy-tools","zappy2"

};

static void test_installed_system_set(void)

{

    PloverTransaction *transaction;

    PloverPackageSet *package_set;

    struct razor_set *set;

    GError *err=NULL;

    transaction=plover_transaction_new();

    g_assert(PLOVER_IS_TRANSACTION(transaction));

    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);

    plover_transaction_set_installed(transaction,package_set);

    set=plover_transaction_get_system_set(transaction);

    g_assert(set == plover_package_set_get_razor(package_set));

    g_object_unref(package_set);

    g_object_unref(transaction);

}

static PloverTransaction *update_noop(void)

{

    PloverTransaction *transaction;

    struct razor_install_iterator *iter;

    struct razor_package *pkg;

    enum razor_install_action action;

    int count;

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_update("../yum-repo-test-dir","/test",

      NULL,&err);

    if (!transaction && err)

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

    g_assert(PLOVER_IS_TRANSACTION(transaction));

    g_assert(!err);

    g_assert(!plover_transaction_get_unsatisfied(transaction));

    iter=plover_transaction_get_install_iterator(transaction,&err);

    if (!iter && err)

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

    g_assert(iter);

    g_assert(!err);

    g_assert(!razor_install_iterator_next(iter,&pkg,&action,&count));

    return transaction;

}

static void test_update_noop(void)

{

    PloverTransaction *transaction;

    transaction=update_noop();

    g_object_unref(transaction);

}

static void test_update_nonexistant(void)

{

    PloverTransaction *transaction;

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

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_update("../yum-repo-test-dir","/test",

      pkgs,&err);

    g_assert(!transaction);

    g_assert(g_error_matches(err,PLOVER_GENERAL_ERROR,

      PLOVER_GENERAL_ERROR_NO_SUCH_PACKAGE));

    g_clear_error(&err);

}

static void test_install(void)

{

    PloverTransaction *transaction;

    struct razor_set *next;

    struct razor_install_iterator *iter;

    struct razor_package *pkg;

    enum razor_install_action action;

    int count;

    char *name;

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

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_install("../yum-repo-test-dir","/test",

      pkgs,&err);

    if (!transaction && err)

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

    g_assert(PLOVER_IS_TRANSACTION(transaction));

    g_assert(!err);

    g_assert(!plover_transaction_get_unsatisfied(transaction));

    next=plover_transaction_get_next_set(transaction,&err);

    if (!next && err)

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

    g_assert(next);

    g_assert(!err);

    iter=plover_transaction_get_install_iterator(transaction,&err);

    if (!iter && err)

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

    g_assert(iter);

    g_assert(!err);

    g_assert(razor_install_iterator_next(iter,&pkg,&action,&count));

    g_assert_cmpint(action,==,RAZOR_INSTALL_ACTION_ADD);

    g_assert_cmpint(count,==,1);

    razor_package_get_details(next,pkg,RAZOR_DETAIL_NAME,&name,

      RAZOR_DETAIL_LAST);

    g_assert_cmpstr(name,==,"zip");

    g_assert(!razor_install_iterator_next(iter,&pkg,&action,&count));

    g_object_unref(transaction);

}

static void test_install_nonexistant(void)

{

    PloverTransaction *transaction;

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

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_install("../yum-repo-test-dir","/test",

      pkgs,&err);

    g_assert(!transaction);

    g_assert(g_error_matches(err,PLOVER_GENERAL_ERROR,

      PLOVER_GENERAL_ERROR_NO_SUCH_PACKAGE));

    g_clear_error(&err);

}

static void test_install_uninstallable(void)

{

    PloverTransaction *transaction;

    struct razor_set *next;

    struct razor_install_iterator *iter;

    struct razor_package *pkg;

    enum razor_install_action action;

    int count;

    char *name;

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

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_install("../yum-repo-test-dir","/test",

      pkgs,&err);

    if (!transaction && err)

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

    g_assert(PLOVER_IS_TRANSACTION(transaction));

    g_assert(!err);

    g_assert_cmpstr(plover_transaction_get_unsatisfied(transaction),==,NULL);

    next=plover_transaction_get_next_set(transaction,&err);

    if (!next && err)

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

    g_assert(next);

    g_assert(!err);

    iter=plover_transaction_get_install_iterator(transaction,&err);

    if (!iter && err)

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

    g_assert(iter);

    g_assert(!err);

    g_assert(razor_install_iterator_next(iter,&pkg,&action,&count));

    g_assert_cmpint(action,==,RAZOR_INSTALL_ACTION_ADD);

    g_assert_cmpint(count,==,1);

    razor_package_get_details(next,pkg,RAZOR_DETAIL_NAME,&name,

      RAZOR_DETAIL_LAST);

    g_assert_cmpstr(name,==,"uninstallable");

    g_assert(!razor_install_iterator_next(iter,&pkg,&action,&count));

    g_assert(!plover_transaction_commit(transaction,NULL,&err));

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

    g_assert_cmpint(err->domain,==,PLOVER_RAZOR_ERROR);

    g_assert_cmpint(err->code,==,RAZOR_GENERAL_ERROR_FAILED);

    g_clear_error(&err);

    g_object_unref(transaction);

}

static void test_unsatisfied(void)

{

    PloverTransaction *transaction;

    struct razor_install_iterator *iter;

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

    const char *s;

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_install("../yum-repo-test-dir","/test",

      pkgs,&err);

    if (!transaction && err)

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

    g_assert(PLOVER_IS_TRANSACTION(transaction));

    g_assert(!err);

    iter=plover_transaction_get_install_iterator(transaction,&err);

    g_assert(!iter);

    g_assert(g_error_matches(err,PLOVER_GENERAL_ERROR,

      PLOVER_GENERAL_ERROR_REQUIREMENTS_NOT_MET));

    g_clear_error(&err);

    s=plover_transaction_get_unsatisfied(transaction);

    g_assert_cmpstr(s,!=,NULL);

    g_assert_cmpstr(s,!=,"");

    g_object_unref(transaction);

}

static void test_remove(void)

{

    PloverTransaction *transaction;

    struct razor_set *system;

    struct razor_install_iterator *iter;

    struct razor_package *pkg;

    enum razor_install_action action;

    int count;

    char *name;

    char *pkgs[]={"zappy-tools",NULL};

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_remove(pkgs,&err);

    if (!transaction && err)

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

    g_assert(PLOVER_IS_TRANSACTION(transaction));

    g_assert(!err);

    system=plover_transaction_get_system_set(transaction);

    g_assert(system);

    iter=plover_transaction_get_install_iterator(transaction,&err);

    if (!iter && err)

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

    g_assert(iter);

    g_assert(!err);

    g_assert(razor_install_iterator_next(iter,&pkg,&action,&count));

    g_assert_cmpint(action,==,RAZOR_INSTALL_ACTION_REMOVE);

    g_assert_cmpint(count,==,0);

    razor_package_get_details(system,pkg,RAZOR_DETAIL_NAME,&name,

      RAZOR_DETAIL_LAST);

    g_assert_cmpstr(name,==,"zappy-tools");

    g_assert(!razor_install_iterator_next(iter,&pkg,&action,&count));

    g_object_unref(transaction);

}

static void test_remove_nonexistant(void)

{

    PloverTransaction *transaction;

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

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_remove(pkgs,&err);

    g_assert(!transaction);

    g_assert(g_error_matches(err,PLOVER_GENERAL_ERROR,

      PLOVER_GENERAL_ERROR_NO_SUCH_PACKAGE));

    g_clear_error(&err);

}

GMainLoop *test_commit_mainloop;

static void test_commit_callback(GObject *source,GAsyncResult *result,

  gpointer user_data)

{

    PloverTransaction *transaction=user_data;

    GError *err=NULL;

    if (!plover_transaction_commit_finish(transaction,result,&err))

    {

	g_assert(err && err->message);

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

    }

    g_assert(!err);

    g_main_loop_quit(test_commit_mainloop);

}

static void test_commit(void)

{

    PloverTransaction *transaction;

    test_commit_mainloop=g_main_loop_new(NULL,FALSE);

    transaction=update_noop();

    plover_transaction_commit_async(transaction,NULL,test_commit_callback,

      transaction);

    g_main_loop_run(test_commit_mainloop);

    g_main_loop_unref(test_commit_mainloop);

    g_object_unref(transaction);

}

static void test_remove_with_leaves(void)

{

    PloverTransaction *transaction;

    struct razor_set *system;

    struct razor_install_iterator *iter;

    struct razor_package *pkg;

    enum razor_install_action action;

    int count;

    char *name;

    char *pkgs[]={"zappy-tools",NULL};

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_remove_with_leaves(pkgs,&err);

    if (!transaction && err)

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

    g_assert(PLOVER_IS_TRANSACTION(transaction));

    g_assert(!err);

    system=plover_transaction_get_system_set(transaction);

    g_assert(system);

    iter=plover_transaction_get_install_iterator(transaction,&err);

    if (!iter && err)

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

    g_assert(iter);

    g_assert(!err);

    g_assert(razor_install_iterator_next(iter,&pkg,&action,&count));

    g_assert_cmpint(action,==,RAZOR_INSTALL_ACTION_REMOVE);

    g_assert_cmpint(count,==,0);

    razor_package_get_details(system,pkg,RAZOR_DETAIL_NAME,&name,

      RAZOR_DETAIL_LAST);

    g_assert(!strcmp(name,"zappy-tools") || !strcmp(name,"zappy"));

    g_assert(razor_install_iterator_next(iter,&pkg,&action,&count));

    g_assert_cmpint(action,==,RAZOR_INSTALL_ACTION_REMOVE);

    g_assert_cmpint(count,==,0);

    razor_package_get_details(system,pkg,RAZOR_DETAIL_NAME,&name,

      RAZOR_DETAIL_LAST);

    g_assert(!strcmp(name,"zappy-tools") || !strcmp(name,"zappy"));

    g_assert(!razor_install_iterator_next(iter,&pkg,&action,&count));

    g_object_unref(transaction);

}

static void test_remove_nonexistant_with_leaves(void)

{

    PloverTransaction *transaction;

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

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_remove_with_leaves(pkgs,&err);

    g_assert(!transaction);

    g_assert(g_error_matches(err,PLOVER_GENERAL_ERROR,

      PLOVER_GENERAL_ERROR_NO_SUCH_PACKAGE));

    g_clear_error(&err);

}

static void test_remove_all(void)

{

    int i;

    PloverTransaction *transaction;

    struct razor_set *system;

    struct razor_install_iterator *iter;

    struct razor_package *pkg;

    enum razor_install_action action;

    int count;

    char *name;

    gchar *root_uri;

    GError *err=NULL;

    GFile *file;

    GList *expected=NULL,*lnk;

    file=g_file_new_for_path("../razor-test-dir");

    root_uri=g_file_get_uri(file);

    g_object_unref(file);

    g_setenv("RAZOR_ROOT",root_uri,TRUE);

    g_free(root_uri);

    transaction=plover_transaction_new_remove_with_leaves(NULL,&err);

    if (!transaction && err)

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

    g_assert(PLOVER_IS_TRANSACTION(transaction));

    g_assert(!err);

    system=plover_transaction_get_system_set(transaction);

    g_assert(system);

    for(i=0;i<G_N_ELEMENTS(zappy_packages);i++)

	expected=g_list_prepend(expected,

	  (gpointer)g_intern_string(zappy_packages[i]));

    iter=plover_transaction_get_install_iterator(transaction,&err);

    if (!iter && err)

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

    g_assert(iter);

    g_assert(!err);

    while(razor_install_iterator_next(iter,&pkg,&action,&count))

    {

	g_assert_cmpint(action,==,RAZOR_INSTALL_ACTION_REMOVE);

	g_assert_cmpint(count,==,0);

	razor_package_get_details(system,pkg,RAZOR_DETAIL_NAME,&name,

	  RAZOR_DETAIL_LAST);

	lnk=g_list_find(expected,g_intern_string(name));

	if (!lnk)

	    g_warning("Unexpected package to be removed: %s",name);

	else

	    expected=g_list_delete_link(expected,lnk);

    }

    if (expected)

	g_warning("%d package%s not removed, including %s",

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

	  expected->data);

    g_object_unref(transaction);

}

static void test_change_installed(void)

{

    PloverTransaction *transaction;

    PloverPackageSet *package_set;

    GError *err=NULL;

    transaction=plover_transaction_new();

    g_assert(PLOVER_IS_TRANSACTION(transaction));

    if (!plover_transaction_root_open(transaction,"../razor-test-dir",&err))

    {

	g_assert(err && err->message);

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

    }

    g_assert(!err);

    g_assert(plover_transaction_get_system_set(transaction));

    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);

    plover_transaction_set_installed(transaction,package_set);

    g_object_unref(package_set);

    g_object_unref(transaction);

}

int main(int argc,char **argv)

{

    g_test_init(&argc,&argv,NULL);

    razor_set_lua_loader("posix",(void (*)())luaopen_posix);

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

    g_test_add_func("/transaction/installed-system-set",

      test_installed_system_set);

    g_test_add_func("/transaction/update-noop",test_update_noop);

    g_test_add_func("/transaction/update-nonexistant",test_update_nonexistant);

    g_test_add_func("/transaction/install",test_install);

    g_test_add_func("/transaction/install-nonexistant",

      test_install_nonexistant);

    g_test_add_func("/transaction/install-uninstallable",

      test_install_uninstallable);

    g_test_add_func("/transaction/unsatisfied",test_unsatisfied);

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

    g_test_add_func("/transaction/remove-nonexistant",test_remove_nonexistant);

    g_test_add_func("/transaction/commit",test_commit);

    g_test_add_func("/transaction/remove-with-leaves",test_remove_with_leaves);

    g_test_add_func("/transaction/remove-nonexistant-with-leaves",

      test_remove_nonexistant_with_leaves);

    g_test_add_func("/transaction/remove-all",test_remove_all);

    g_test_add_func("/transaction/change-installed",test_change_installed);

    return g_test_run();

}