Source code for tests.make_dist_tests

#!/usr/bin/python3
"""Tests for fixes module."""
#
# (C) Pywikibot team, 2022
#
# Distributed under the terms of the MIT license.
#
import os
import sys
import unittest

from pywikibot import __version__
from make_dist import handle_args

from tests.aspects import TestCase


class TestMakeDist(TestCase):

    """Test the make_dist script."""

    net = False

    def _test_argv(self):
        """Test argv."""
        if os.environ.get('PYWIKIBOT_TEST_RUNNING', '0') != '1':
            self.assertEqual(__file__, sys.argv[0])
        self.assertIn('sdist', sys.argv)
        self.assertIn('bdist_wheel', sys.argv)

    def test_handle_args_empty(self):
        """Test make_dist handle_args function."""
        args = handle_args()
        self.assertEqual(args, (False, ) * 5)
        self._test_argv()

    def test_handle_args_nodist(self):
        """Test make_dist handle_args function."""
        sys.argv += ['-local', '-nodist', '-remote']
        *args, nodist = handle_args()
        self.assertEqual(args, [False] * 4)
        self.assertTrue(nodist)
        self._test_argv()

    def test_handle_args(self):
        """Test make_dist handle_args function."""
        sys.argv += ['-clear', '-local', '-remote', '-upgrade']
        local, remote, clear, upgrade, nodist = handle_args()
        self.assertTrue(local)
        self.assertEqual(remote, 'dev' not in __version__)
        self.assertTrue(clear)
        self.assertTrue(upgrade)
        self.assertFalse(nodist)
        self._test_argv()


if __name__ == '__main__':  # pragma: no cover
    unittest.main()