"""Tests for distutils.version."""
import unittest
import distutils
from distutils.version import LooseVersion
from distutils.version import StrictVersion
from test.support import run_unittest


class VersionTestCase(unittest.TestCase):
    def setUp(self):
        self.ctx = distutils.version.suppress_known_deprecation()
        self.ctx.__enter__()

    def tearDown(self):
        self.ctx.__exit__(None, None, None)

    def test_prerelease(self):
        version = StrictVersion('1.2.3a1')
        self.assertEqual(version.version, (1, 2, 3))
        self.assertEqual(version.prerelease, ('a', 1))
        self.assertEqual(str(version), '1.2.3a1')

        version = StrictVersion('1.2.0')
        self.assertEqual(str(version), '1.2')

    def test_cmp_strict(self):
        versions = (
            ('1.5.1', '1.5.2b2', -1),
            ('161', '3.10a', ValueError),
            ('8.02', '8.02', 0),
            ('3.4j', '1996.07.12', ValueError),
            ('3.2.pl0', '3.1.1.6', ValueError),
            ('2g6', '11g', ValueError),
            ('0.9', '2.2', -1),
            ('1.2.1', '1.2', 1),
            ('1.1', '1.2.2', -1),
            ('1.2', '1.1', 1),
            ('1.2.1', '1.2.2', -1),
            ('1.2.2', '1.2', 1),
            ('1.2', '1.2.2', -1),
            ('0.4.0', '0.4', 0),
            ('1.13++', '5.5.kw', ValueError),
        )

        for v1, v2, wanted in versions:
            try:
                res = StrictVersion(v1)._cmp(StrictVersion(v2))
            except ValueError:
                if wanted is ValueError:
                    continue
                else:
                    raise AssertionError(
                        ("cmp(%s, %s) " "shouldn't raise ValueError") % (v1, v2)
                    )
            self.assertEqual(
                res, wanted, 'cmp(%s, %s) should be %s, got %s' % (v1, v2, wanted, res)
            )
            res = StrictVersion(v1)._cmp(v2)
            self.assertEqual(
                res, wanted, 'cmp(%s, %s) should be %s, got %s' % (v1, v2, wanted, res)
            )
            res = StrictVersion(v1)._cmp(object())
            self.assertIs(
                res,
                NotImplemented,
                'cmp(%s, %s) should be NotImplemented, got %s' % (v1, v2, res),
            )

    def test_cmp(self):
        versions = (
            ('1.5.1', '1.5.2b2', -1),
            ('161', '3.10a', 1),
            ('8.02', '8.02', 0),
            ('3.4j', '1996.07.12', -1),
            ('3.2.pl0', '3.1.1.6', 1),
            ('2g6', '11g', -1),
            ('0.960923', '2.2beta29', -1),
            ('1.13++', '5.5.kw', -1),
        )

        for v1, v2, wanted in versions:
            res = LooseVersion(v1)._cmp(LooseVersion(v2))
            self.assertEqual(
                res, wanted, 'cmp(%s, %s) should be %s, got %s' % (v1, v2, wanted, res)
            )
            res = LooseVersion(v1)._cmp(v2)
            self.assertEqual(
                res, wanted, 'cmp(%s, %s) should be %s, got %s' % (v1, v2, wanted, res)
            )
            res = LooseVersion(v1)._cmp(object())
            self.assertIs(
                res,
                NotImplemented,
                'cmp(%s, %s) should be NotImplemented, got %s' % (v1, v2, res),
            )


def test_suite():
    return unittest.TestLoader().loadTestsFromTestCase(VersionTestCase)


if __name__ == "__main__":
    run_unittest(test_suite())