/*

  * 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>

 LUALIB_API int luaopen_posix(lua_State *L);

 static void test_install(void)

 {

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

     gchar *root,*root_uri;

     GError *err=NULL;

     GFile *file;

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

     if (!plover_install("../yum-repo-test-dir","/test",pkgs,&err))

     {

 	g_assert(err && err->message);

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

     }

     g_assert(!err);

     g_unsetenv("RAZOR_ROOT");

     g_free(root);

 }

 static void test_install_bad_rpm(void)

 {

     gboolean installed_bad_rpm;

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

     gchar *root;

     GError *err=NULL;

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

     g_assert(mkdtemp(root));

     g_setenv("RAZOR_ROOT",root,TRUE);

     chmod("../yum-repo-test-dir/rpms/zip-1-1.noarch.rpm",0);

     installed_bad_rpm=plover_install("../yum-repo-test-dir","/test",pkgs,&err);

     chmod("../yum-repo-test-dir/rpms/zip-1-1.noarch.rpm",0666);

     g_assert(!installed_bad_rpm);

     g_assert(err && err->message);

     g_unsetenv("RAZOR_ROOT");

     g_free(root);

 }

 static void test_update_noop(void)

 {

     gchar *root,*root_uri;

     GError *err=NULL;

     GFile *file;

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

     if (!plover_update("../yum-repo-test-dir","/test",NULL,&err))

     {

 	g_assert(err && err->message);

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

     }

     g_assert(!err);

     g_unsetenv("RAZOR_ROOT");

     g_free(root);

 }

 static void test_update_nonexistant(void)

 {

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

     gchar *root,*root_uri;

     GError *err=NULL;

     GFile *file;

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

     g_assert(!plover_update("../yum-repo-test-dir","/test",pkgs,&err));

     g_assert(g_error_matches(err,PLOVER_GENERAL_ERROR,

       PLOVER_GENERAL_ERROR_NO_SUCH_PACKAGE));

     g_clear_error(&err);

     g_unsetenv("RAZOR_ROOT");

     g_free(root);

 }

 static void test_remove(void)

 {

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

     gchar *root,*root_uri;

     GError *err=NULL;

     GFile *file;

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

     if (!plover_install("../yum-repo-test-dir","/test",pkgs,&err))

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

     if (!plover_remove(pkgs,&err))

     {

 	g_assert(err && err->message);

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

     }

     g_assert(!err);

     g_unsetenv("RAZOR_ROOT");

     g_free(root);

 }

 static void test_remove_noop(void)

 {

     gchar *root,*root_uri;

     GError *err=NULL;

     GFile *file;

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

     if (!plover_remove(NULL,&err))

     {

 	g_assert(err && err->message);

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

     }

     g_assert(!err);

     g_unsetenv("RAZOR_ROOT");

     g_free(root);

 }

 static void test_remove_nonexistant(void)

 {

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

     gchar *root,*root_uri;

     GError *err=NULL;

     GFile *file;

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

     g_assert(!plover_remove(pkgs,&err));

     g_assert(g_error_matches(err,PLOVER_GENERAL_ERROR,

       PLOVER_GENERAL_ERROR_NO_SUCH_PACKAGE));

     g_clear_error(&err);

     g_unsetenv("RAZOR_ROOT");

     g_free(root);

 }

 static void test_remove_warning(void)

 {

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

     gchar *root,*root_uri;

     GError *err=NULL;

     GFile *file;

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

     if (!plover_install("../yum-repo-test-dir","/test",pkgs,&err))

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

     if (!plover_remove(pkgs,&err))

     {

 	g_assert(err && err->message);

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

     }

     g_assert(!err);

     g_unsetenv("RAZOR_ROOT");

     g_free(root);

 }

 static void test_prefix(void)

 {

     gchar *root_uri;

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

     g_assert(plover_installed_files_match_prefix("/usr//"));

     g_assert(!plover_installed_files_match_prefix("/tmp"));

     g_unsetenv("RAZOR_ROOT");

 }

 static void test_prefix_none(void)

 {

     gchar *root,*root_uri;

     GFile *file;

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

     g_assert(plover_installed_files_match_prefix("/any-prefix"));

     g_unsetenv("RAZOR_ROOT");

     g_free(root);

 }

 int main(int argc,char **argv)

 {

     int retval;

     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("/razor/install",test_install);

     g_test_add_func("/razor/install-bad-rpm",test_install_bad_rpm);

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

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

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

     g_test_add_func("/razor/remove-noop",test_remove_noop);

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

     g_test_add_func("/razor/remove-warning",test_remove_warning);

     g_test_add_func("/razor/prefix",test_prefix);

     g_test_add_func("/razor/prefix-none",test_prefix_none);

     retval=g_test_run();

     return retval;

 }