#
# Chris Lumens <clumens@redhat.com>
#
# Copyright 2009 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.  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.
#
import unittest
from tests.baseclass import CommandTest
from pykickstart.errors import KickstartParseError, KickstartDeprecationWarning
from pykickstart.commands.timezone import FC3_Timezone, FC6_Timezone, F18_Timezone, F25_Timezone, RHEL7_Timezone


class Timezone_TestCase(unittest.TestCase):
    def runTest(self):
        for cmd_class in [FC6_Timezone, F25_Timezone, RHEL7_Timezone]:
            cmd = cmd_class()
            op = cmd._getParser()
            for action in op._actions:
                if '--isUtc' in action.option_strings:
                    self.assertFalse(action.default)

        cmd = F18_Timezone()
        self.assertEqual(cmd.__str__(), '')


class FC3_TestCase(CommandTest):
    command = "timezone"

    def runTest(self):
        # assert defaults
        self.assertFalse(FC3_Timezone().isUtc)
        self.assertFalse(F18_Timezone().nontp)

        # pass
        self.assert_parse("timezone Eastern", "timezone  Eastern\n")

        # On FC6 and later, we write out --isUtc regardless of what the input was.
        if self.__class__.__name__ == "FC3_TestCase":
            self.assert_parse("timezone --utc Eastern", "timezone --utc Eastern\n")
        else:
            self.assert_parse("timezone --utc Eastern", "timezone --isUtc Eastern\n")

        # fail
        self.assert_parse_error("timezone")
        self.assert_parse_error("timezone Eastern Central")
        self.assert_parse_error("timezone --blah Eastern")
        self.assert_parse_error("timezone --utc")
        self.assert_parse_error("timezone --bogus-option")

        # extra test coverage
        cmd = self.handler().commands[self.command]
        cmd.timezone = None
        self.assertEqual(cmd.__str__(), "")

class FC6_TestCase(FC3_TestCase):
    def runTest(self):
        FC3_TestCase.runTest(self)

        # pass
        self.assert_parse("timezone --isUtc Eastern", "timezone --isUtc Eastern\n")

        # fail
        self.assert_parse_error("timezone --isUtc")

class F18_TestCase(FC6_TestCase):
    def runTest(self):
        # pass
        self.assert_parse("timezone --utc Europe/Prague")
        self.assert_parse("timezone --isUtc Europe/Prague\n")
        self.assert_parse("timezone --isUtc Eastern", "timezone Eastern --isUtc\n")
        self.assert_parse("timezone Europe/Prague")
        self.assert_parse("timezone Europe/Prague --nontp",
                          "timezone Europe/Prague --nontp\n")
        self.assert_parse("timezone Europe/Prague "
                          "--ntpservers=ntp.cesnet.cz,tik.nic.cz")
        self.assert_parse("timezone Europe/Prague --ntpservers=ntp.cesnet.cz",
                          "timezone Europe/Prague --ntpservers=ntp.cesnet.cz\n")

        # fail
        self.assert_parse_error("timezone")
        self.assert_parse_error("timezone Eastern Central")
        self.assert_parse_error("timezone --blah Eastern")
        self.assert_parse_error("timezone --utc")
        self.assert_parse_error("timezone --isUtc")
        self.assert_parse_error("timezone Europe/Prague --nontp "
                                "--ntpservers=ntp.cesnet.cz")
        self.assert_parse_error("timezone Europe/Prague --ntpservers="
                                "ntp.cesnet.cz, tik.nic.cz")

class F23_TestCase(F18_TestCase):
    def runTest(self):
        # should keep multiple instances of the same URL
        self.assert_parse("timezone --utc Europe/Prague --ntpservers=ntp.cesnet.cz,0.fedora.pool.ntp.org," +
                          "0.fedora.pool.ntp.org,0.fedora.pool.ntp.org,0.fedora.pool.ntp.org",
                          "timezone Europe/Prague --isUtc --ntpservers=ntp.cesnet.cz,0.fedora.pool.ntp.org," +
                          "0.fedora.pool.ntp.org,0.fedora.pool.ntp.org,0.fedora.pool.ntp.org\n")
        self.assert_parse("timezone --utc Europe/Sofia --ntpservers=,0.fedora.pool.ntp.org,")

        # fail
        self.assert_parse_error("timezone Europe/Sofia --nontp --ntpservers=0.fedora.pool.ntp.org,1.fedora.pool.ntp.org")


class RHEL7_TestCase(F18_TestCase):
    def runTest(self):
        # since RHEL7 command version the timezone command can be used
        # without a timezone specification
        self.assert_parse("timezone --utc")
        self.assert_parse("timezone Europe/Sofia")
        self.assert_parse("timezone --isUtc")
        self.assert_parse("timezone --ntpservers=ntp.cesnet.cz")
        self.assert_parse("timezone --ntpservers=ntp.cesnet.cz,tik.nic.cz")
        # unknown argument
        self.assert_parse_error("timezone --blah")
        # more than two timezone specs
        self.assert_parse_error("timezone foo bar", KickstartParseError, 'One or zero arguments are expected for the timezone command')
        self.assert_parse_error("timezone --utc foo bar", exception=KickstartParseError)
        # just "timezone" without any arguments is also wrong as it really dosn't make sense
        self.assert_parse_error("timezone", KickstartParseError, 'At least one option and/or an argument are expected for the timezone command')

        # fail
        self.assert_parse_error("timezone Europe/Sofia --nontp --ntpservers=0.fedora.pool.ntp.org,1.fedora.pool.ntp.org")


class F25_TestCase(F23_TestCase):
    def runTest(self):
        # since RHEL7 command version the timezone command can be used
        # without a timezone specification
        self.assert_parse("timezone --utc", "timezone --isUtc\n")
        self.assert_parse("timezone --isUtc", "timezone --isUtc\n")
        self.assert_parse("timezone --ntpservers=ntp.cesnet.cz", "timezone --ntpservers=ntp.cesnet.cz\n")
        self.assert_parse("timezone --ntpservers=ntp.cesnet.cz,tik.nic.cz", "timezone --ntpservers=ntp.cesnet.cz,tik.nic.cz\n")
        # normal usage
        self.assert_parse("timezone Europe/Oslo --nontp", "timezone Europe/Oslo --nontp\n")
        # unknown argument
        self.assert_parse_error("timezone --blah")
        # more than two timezone specs
        self.assert_parse_error("timezone foo bar", KickstartParseError, 'One or zero arguments are expected for the timezone command')
        self.assert_parse_error("timezone --utc foo bar", exception=KickstartParseError)
        # just "timezone" without any arguments is also wrong as it really dosn't make sense
        self.assert_parse_error("timezone", KickstartParseError, 'At least one option and/or an argument are expected for the timezone command')

        # fail
        self.assert_parse_error("timezone Europe/Sofia --nontp --ntpservers=0.fedora.pool.ntp.org,1.fedora.pool.ntp.org")


class F32_TestCase(F25_TestCase):
    command = "timezone"

    def failures(self):
        # Failures
        # unknown argument
        self.assert_parse_error("timezone --blah")
        # more than two timezone specs
        self.assert_parse_error("timezone foo bar", KickstartParseError, 'One or zero arguments are expected for the timezone command')
        self.assert_parse_error("timezone --utc foo bar", exception=KickstartParseError)
        # no options
        self.assert_parse_error("timezone", KickstartParseError, 'At least one option and/or an argument are expected for the timezone command')
        # contradictory options
        self.assert_parse_error("timezone Europe/Sofia --nontp --ntpservers=0.fedora.pool.ntp.org,1.fedora.pool.ntp.org")

    def runTest(self):
        self.failures()

        # Successes
        # normal contents
        self.assert_parse("timezone Europe/Prague --ntpservers=ntp.cesnet.cz",
                          "timezone Europe/Prague --ntpservers=ntp.cesnet.cz\n")
        # no timezone spec
        self.assert_parse("timezone --ntpservers=ntp.cesnet.cz",
                          "timezone --ntpservers=ntp.cesnet.cz\n")
        # no ntp wanted
        self.assert_parse("timezone Europe/Oslo --nontp",
                          "timezone Europe/Oslo --nontp\n")

        # New in F32: any variant of UTC should be returned as --utc (again - was in FC3)
        # --utc should result in no warnings
        self.assert_parse("timezone --utc Europe/Bratislava",
                          "timezone Europe/Bratislava --utc\n")
        # but --isUtc should now give warning
        with self.assertWarns(KickstartDeprecationWarning):
            self.assert_parse("timezone --isUtc Europe/Bratislava")


class F33_TestCase(F32_TestCase):
    command = "timezone"

    def runTest(self):
        F32_TestCase.runTest(self)
        # As of Fedora 33 the --ntpservers and --nontp options are considered deprecated.

        # Check using --ntpservers returns appropriate deprecation warning
        with self.assertWarns(KickstartDeprecationWarning):
            self.assert_parse("timezone --ntpservers foo,bar,baz")

        # Check using --ntpservers returns appropriate deprecation warning
        with self.assertWarns(KickstartDeprecationWarning):
            self.assert_parse("timezone --nontp")

class F40_TestCase(F33_TestCase):
    command = "timezone"

    def runTest(self):
        F33_TestCase.runTest(self)

        # deprecated
        self.assert_deprecated("timezone", "--isUtc")
        self.assert_deprecated("timezone", "--ntpservers")
        self.assert_deprecated("timezone", "--nontp")

class RHEL10_TestCase(F32_TestCase):
    command = "timezone"

    def runTest(self):
        # removed
        self.assert_removed("timezone", "--isUtc")
        self.assert_removed("timezone", "--ntpservers")
        self.assert_removed("timezone", "--nontp")

        # Test failures
        F32_TestCase.failures(self)

        self.assert_parse("timezone --utc Europe/Bratislava",
                          "timezone Europe/Bratislava --utc\n")
        self.assert_parse("timezone --utc", "timezone --utc\n")
        self.assert_parse("timezone Pacific", "timezone Pacific\n")

if __name__ == "__main__":
    unittest.main()
