# Copyright (C) 2018  Red Hat, Inc.
#
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions of
# the GNU General Public License v.2, or (at your option) any later version.
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY expressed or implied, including the implied warranties 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.  Any Red Hat trademarks that are incorporated in the
# source code or documentation are not subject to the GNU General Public
# License and may only be used or replicated with the express permission of
# Red Hat, Inc.
#

from __future__ import absolute_import
from __future__ import unicode_literals

import unittest

import dnf.cli
import dnf.sack

from tests.support import mock, PkgStub
import repodiff


PACKAGES_OLD = [
    PkgStub('toberemoved', '0', '1.0', '1', 'noarch', 'old-repo'),
    PkgStub('tobeupgraded', '0', '1.0', '1', 'noarch', 'old-repo'),
    PkgStub('tobedowngraded', '0', '1.0', '5', 'noarch', 'old-repo'),
    PkgStub('tobeobsoleted', '0', '1.0', '1', 'noarch', 'old-repo'),
    PkgStub('tostayput', '0', '1.0', '1', 'noarch', 'old-repo'),
]

PACKAGES_NEW = [
    PkgStub('tobeupgraded', '0', '1.0', '5', 'noarch', 'new-repo'),
    PkgStub('tobedowngraded', '0', '1.0', '1', 'noarch', 'new-repo'),
    PkgStub('tostayput', '0', '1.0', '1', 'noarch', 'new-repo'),
    PkgStub('added', '0', '1.0', '1', 'noarch', 'new-repo'),
    PkgStub('obsoleter', '0', '1.0', '1', 'noarch', 'new-repo',
            obsoletes=['tobeobsoleted']),
]


class QueryStub(list):
    def __init__(self, lst):
        super(QueryStub, self).__init__(lst)

    def filter(self, *args, **kwargs):
        filtered = []
        if 'obsoletes' in kwargs:
            for obsolete in kwargs['obsoletes']:
                for pkg in self:
                    if obsolete.name in pkg.obsoletes:
                        filtered.append(pkg)
        elif 'provides' in kwargs:
            for provide in kwargs['provides']:
                for pkg in self:
                    if pkg.name == provide:
                        filtered.append(pkg)
        return filtered


class RepodiffCommandTest(unittest.TestCase):

    def setUp(self):
        cli = mock.MagicMock()
        self.cmd = repodiff.RepoDiffCommand(cli)
        self.cmd.cli.base = dnf.cli.cli.BaseCli()
        self.cmd.cli.base._sack = dnf.sack.Sack()
        self.cmd.opts = mock.Mock()
        self.cmd.opts.compare_arch = False

        self.repodiff = self.cmd._repodiff(
            QueryStub(PACKAGES_OLD), QueryStub(PACKAGES_NEW))

    def test_added(self):
        added = sorted(self.repodiff['added'])
        self.assertEqual(
            [p.fullname for p in added],
            ['added-1.0-1.noarch', 'obsoleter-1.0-1.noarch']
        )

    def test_removed(self):
        removed = sorted(self.repodiff['removed'])
        self.assertEqual(
            [p.fullname for p in removed],
            ['tobeobsoleted-1.0-1.noarch', 'toberemoved-1.0-1.noarch']
        )

    def test_obsoletes(self):
        self.assertEqual(
            [(k, v.fullname) for k, v in self.repodiff['obsoletes'].items()],
            [('tobeobsoleted', 'obsoleter-1.0-1.noarch')])

    def test_upgraded(self):
        upgraded = sorted(self.repodiff['upgraded'])
        self.assertEqual(
            [(o.fullname, n.fullname) for o, n in upgraded],
            [('tobeupgraded-1.0-1.noarch', 'tobeupgraded-1.0-5.noarch')]
        )

    def test_downgraded(self):
        downgraded = sorted(self.repodiff['downgraded'])
        self.assertEqual(
            [(o.fullname, n.fullname) for o, n in downgraded],
            [('tobedowngraded-1.0-5.noarch', 'tobedowngraded-1.0-1.noarch')]
        )