#include "tests.pch.h"

#include <tests.utils.h>

using namespace Microsoft::VisualStudio::CppUnitTestFramework;

using namespace vcpkg;
using namespace vcpkg::Update;

namespace UnitTest1
{
    using Pgh = std::vector<std::unordered_map<std::string, std::string>>;

    class UpdateTests : public TestClass<UpdateTests>
    {
        TEST_METHOD(find_outdated_packages_basic)
        {
            std::vector<std::unique_ptr<StatusParagraph>> status_paragraphs;
            status_paragraphs.push_back(make_status_pgh("a"));
            status_paragraphs.back()->package.version = "2";

            StatusParagraphs status_db(std::move(status_paragraphs));

            std::unordered_map<std::string, SourceControlFileLocation> map;
            auto scf = unwrap(SourceControlFile::parse_control_file(Pgh{{{"Source", "a"}, {"Version", "0"}}}));
            map.emplace("a", SourceControlFileLocation { std::move(scf), "" });
            Dependencies::MapPortFileProvider provider(map);

            auto pkgs = SortedVector<OutdatedPackage>(Update::find_outdated_packages(provider, status_db),
                                                      &OutdatedPackage::compare_by_name);

            Assert::AreEqual(size_t(1), pkgs.size());
            Assert::AreEqual("2", pkgs[0].version_diff.left.to_string().c_str());
            Assert::AreEqual("0", pkgs[0].version_diff.right.to_string().c_str());
        }

        TEST_METHOD(find_outdated_packages_features)
        {
            std::vector<std::unique_ptr<StatusParagraph>> status_paragraphs;
            status_paragraphs.push_back(make_status_pgh("a"));
            status_paragraphs.back()->package.version = "2";

            status_paragraphs.push_back(make_status_feature_pgh("a", "b"));
            status_paragraphs.back()->package.version = "2";

            StatusParagraphs status_db(std::move(status_paragraphs));

            std::unordered_map<std::string, SourceControlFileLocation> map;
            auto scf = unwrap(SourceControlFile::parse_control_file(Pgh{{{"Source", "a"}, {"Version", "0"}}}));
            map.emplace("a", SourceControlFileLocation { std::move(scf), "" });
            Dependencies::MapPortFileProvider provider(map);

            auto pkgs = SortedVector<OutdatedPackage>(Update::find_outdated_packages(provider, status_db),
                                                      &OutdatedPackage::compare_by_name);

            Assert::AreEqual(size_t(1), pkgs.size());
            Assert::AreEqual("2", pkgs[0].version_diff.left.to_string().c_str());
            Assert::AreEqual("0", pkgs[0].version_diff.right.to_string().c_str());
        }

        TEST_METHOD(find_outdated_packages_features_2)
        {
            std::vector<std::unique_ptr<StatusParagraph>> status_paragraphs;
            status_paragraphs.push_back(make_status_pgh("a"));
            status_paragraphs.back()->package.version = "2";

            status_paragraphs.push_back(make_status_feature_pgh("a", "b"));
            status_paragraphs.back()->package.version = "0";
            status_paragraphs.back()->state = InstallState::NOT_INSTALLED;
            status_paragraphs.back()->want = Want::PURGE;

            StatusParagraphs status_db(std::move(status_paragraphs));

            std::unordered_map<std::string, SourceControlFileLocation> map;
            auto scf = unwrap(SourceControlFile::parse_control_file(Pgh{{{"Source", "a"}, {"Version", "0"}}}));
            map.emplace("a", SourceControlFileLocation{ std::move(scf), "" });
            Dependencies::MapPortFileProvider provider(map);

            auto pkgs = SortedVector<OutdatedPackage>(Update::find_outdated_packages(provider, status_db),
                                                      &OutdatedPackage::compare_by_name);

            Assert::AreEqual(size_t(1), pkgs.size());
            Assert::AreEqual("2", pkgs[0].version_diff.left.to_string().c_str());
            Assert::AreEqual("0", pkgs[0].version_diff.right.to_string().c_str());
        }

        TEST_METHOD(find_outdated_packages_none)
        {
            std::vector<std::unique_ptr<StatusParagraph>> status_paragraphs;
            status_paragraphs.push_back(make_status_pgh("a"));
            status_paragraphs.back()->package.version = "2";

            StatusParagraphs status_db(std::move(status_paragraphs));

            std::unordered_map<std::string, SourceControlFileLocation> map;
            auto scf = unwrap(SourceControlFile::parse_control_file(Pgh{{{"Source", "a"}, {"Version", "2"}}}));
            map.emplace("a", SourceControlFileLocation{ std::move(scf), "" });
            Dependencies::MapPortFileProvider provider(map);

            auto pkgs = SortedVector<OutdatedPackage>(Update::find_outdated_packages(provider, status_db),
                                                      &OutdatedPackage::compare_by_name);

            Assert::AreEqual(size_t(0), pkgs.size());
        }
    };
}