/*

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

#include <plover/repository.h>

const char *test_repo_packages[]={

    "zsh","zsh2","zip","zap","filesystem","zappy","zappy-tools","zappy2",

    "unsatisfiable","uninstallable","badpostun"

};

static void test_from_files(void)

{

    int i;

    struct razor_set *set;

    struct razor_package_iterator *iter;

    struct razor_package *package;

    PloverRepository *repository;

    PloverPackageSet *package_set;

    char **filenames;

    const char *name;

    GList *expected=NULL,*lnk;

    GError *err=NULL;

    filenames=g_new(char *,G_N_ELEMENTS(test_repo_packages)+1);

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

	filenames[i]=g_strconcat("../yum-repo-test-dir/rpms/",

	  test_repo_packages[i],"-1-1.noarch.rpm",NULL);

    filenames[i]=NULL;

    repository=plover_repository_new_from_files((const char **)filenames,&err);

    g_strfreev(filenames);

    if (!repository && err)

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

    g_assert(repository != NULL);

    g_assert(err == NULL);

    package_set=plover_repository_get_package_set(repository);

    g_assert(package_set != NULL);

    set=plover_package_set_get_razor(package_set);

    g_assert(set != NULL);

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

	expected=g_list_prepend(expected,

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

    iter=razor_package_iterator_create(set);

    while(razor_package_iterator_next(iter,&package,RAZOR_DETAIL_NAME,&name,

      RAZOR_DETAIL_LAST))

    {

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

	if (!lnk)

	    g_warning("Unexpected package in set: %s",name);

	else

	    expected=g_list_delete_link(expected,lnk);

    }

    razor_package_iterator_destroy(iter);

    if (expected)

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

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

	  expected->data);

    g_object_unref(repository);

}

static void test_from_yum(void)

{

    int i;

    struct razor_set *set;

    struct razor_package_iterator *iter;

    struct razor_package *package;

    PloverRepository *repository;

    PloverPackageSet *package_set;

    const char *name;

    GList *expected=NULL,*lnk;

    GError *err=NULL;

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

    if (!repository && err)

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

    g_assert(repository != NULL);

    g_assert(err == NULL);

    package_set=plover_repository_get_package_set(repository);

    g_assert(package_set != NULL);

    set=plover_package_set_get_razor(package_set);

    g_assert(set != NULL);

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

	expected=g_list_prepend(expected,

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

    iter=razor_package_iterator_create(set);

    while(razor_package_iterator_next(iter,&package,RAZOR_DETAIL_NAME,&name,

      RAZOR_DETAIL_LAST))

    {

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

	if (!lnk)

	    g_warning("Unexpected package in set: %s",name);

	else

	    expected=g_list_delete_link(expected,lnk);

    }

    razor_package_iterator_destroy(iter);

    if (expected)

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

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

	  expected->data);

    g_object_unref(repository);

}

static void test_open_rpm(void)

{

    struct razor_rpm *rpm;

    PloverRepository *repository;

    PloverPackageSet *package_set;

    PloverPackage *package;

    const char *name;

    GSList *packages,*lnk;

    GError *err=NULL;

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

    if (!repository)

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

    package_set=plover_repository_get_package_set(repository);

    packages=plover_package_set_get_packages(package_set);

    for(lnk=packages;lnk;lnk=lnk->next)

    {

	package=PLOVER_PACKAGE(lnk->data);

	rpm=plover_repository_open_rpm(repository,package,&err);

	if (!rpm && err)

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

	g_assert(rpm != NULL);

	g_assert(err == NULL);

	razor_rpm_get_details(rpm,RAZOR_DETAIL_NAME,&name,RAZOR_DETAIL_LAST);

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

	razor_rpm_close(rpm);

    }

    g_object_unref(repository);

}

int main(int argc,char **argv)

{

    g_test_init(&argc,&argv,NULL);

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

    g_test_add_func("/repository/from-files",test_from_files);

    g_test_add_func("/repository/from-yum",test_from_yum);

    g_test_add_func("/repository/open-rpm",test_open_rpm);

    return g_test_run();

}