/*
 * 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/Packages/",
	  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();
}