// Copyright Contributors to the DNF5 project.
// Copyright Contributors to the libdnf project.
// SPDX-License-Identifier: GPL-2.0-or-later
//
// This file is part of libdnf: https://github.com/rpm-software-management/libdnf/
//
// Libdnf 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.
//
// Libdnf 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 libdnf.  If not, see <https://www.gnu.org/licenses/>.


#include "test_advisory_module.hpp"

#include <filesystem>
#include <set>
#include <vector>


CPPUNIT_TEST_SUITE_REGISTRATION(AdvisoryAdvisoryModuleTest);

//This allows running only this single test suite, by using `getRegistry("AdvisoryAdvisoryTest_suite")` in run_tests.cpp
CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(AdvisoryAdvisoryModuleTest, "AdvisoryAdvisoryModuleTest_suite");

void AdvisoryAdvisoryModuleTest::setUp() {
    BaseTestCase::setUp();
    BaseTestCase::add_repo_repomd("repomd-repo1");
    libdnf5::advisory::AdvisoryQuery advisories(base);
    advisories.filter_name("DNF-2019-1");
    libdnf5::advisory::Advisory advisory = *advisories.begin();
    std::vector<libdnf5::advisory::AdvisoryCollection> collections = advisory.get_collections();
#ifdef WITH_MODULEMD
    modules = collections[0].get_modules();
#else
    CPPUNIT_ASSERT_THROW(collections[0].get_modules(), libdnf5::RuntimeError);
#endif
}

#ifdef WITH_MODULEMD
void AdvisoryAdvisoryModuleTest::test_get_name() {
    // Tests get_name method
    CPPUNIT_ASSERT_EQUAL(std::string("perl-DBI"), std::string(modules[0].get_name()));
}

void AdvisoryAdvisoryModuleTest::test_get_stream() {
    // Tests get_stream method
    CPPUNIT_ASSERT_EQUAL(std::string("master"), std::string(modules[0].get_stream()));
}

void AdvisoryAdvisoryModuleTest::test_get_version() {
    // Tests get_version method
    CPPUNIT_ASSERT_EQUAL(std::string("2"), std::string(modules[0].get_version()));
}

void AdvisoryAdvisoryModuleTest::test_get_context() {
    // Tests get_context method
    CPPUNIT_ASSERT_EQUAL(std::string("2a"), std::string(modules[0].get_context()));
}

void AdvisoryAdvisoryModuleTest::test_get_arch() {
    // Tests get_arch method
    CPPUNIT_ASSERT_EQUAL(std::string("x86_64"), std::string(modules[0].get_arch()));
}

void AdvisoryAdvisoryModuleTest::test_get_advisory_id() {
    // Tests get_advisory_id method
    libdnf5::advisory::AdvisoryId adv_id = modules[0].get_advisory_id();
    CPPUNIT_ASSERT(adv_id.id > 0);
}

void AdvisoryAdvisoryModuleTest::test_get_advisory() {
    // Tests get_advisory method
    libdnf5::advisory::Advisory a = modules[0].get_advisory();
    CPPUNIT_ASSERT_EQUAL(std::string("DNF-2019-1"), a.get_name());
}

void AdvisoryAdvisoryModuleTest::test_get_advisory_collection() {
    // Tests get_advisory_collection method
    libdnf5::advisory::AdvisoryCollection ac = modules[0].get_advisory_collection();
    std::vector<libdnf5::advisory::AdvisoryModule> out_mods = ac.get_modules();
    size_t target_size = 2;
    CPPUNIT_ASSERT_EQUAL(target_size, out_mods.size());
}
#endif
