pynsist/nsist/tests/test_local_wheels.py

import unittest
import os
import platform
import subprocess
import glob

from testpath.tempdir import TemporaryDirectory
from testpath import assert_isfile, assert_isdir
from nsist.pypi import fetch_pypi_wheels

class TestLocalWheels(unittest.TestCase):
    def test_matching_one_pattern(self):
        with TemporaryDirectory() as td1:
            subprocess.call(['pip', 'wheel', 'requests==2.19.1', '-w', td1])

            with TemporaryDirectory() as td2:
                fetch_pypi_wheels([], [os.path.join(td1, '*.whl')], td2, platform.python_version(), 64)

                assert_isdir(os.path.join(td2, 'requests'))
                assert_isfile(os.path.join(td2, 'requests-2.19.1.dist-info', 'METADATA'))

                assert_isdir(os.path.join(td2, 'urllib3'))
                self.assertTrue(glob.glob(os.path.join(td2, 'urllib3*.dist-info')))

    def test_duplicate_wheel_files_raise(self):
        with TemporaryDirectory() as td1:
            subprocess.call(['pip', 'wheel', 'requests==2.19.1', '-w', td1])

            with TemporaryDirectory() as td2:
                with self.assertRaisesRegex(ValueError, 'wheel distribution requests already included'):
                    fetch_pypi_wheels(['requests==2.19.1'], [os.path.join(td1, '*.whl')], td2, platform.python_version(), 64)

    def test_invalid_wheel_file_raise(self):
        with TemporaryDirectory() as td1:
            open(os.path.join(td1, 'notawheel.txt'), 'w+')

            with TemporaryDirectory() as td2:
                with self.assertRaisesRegex(ValueError, 'Invalid wheel file name: notawheel.txt'):
                    fetch_pypi_wheels([], [os.path.join(td1, '*')], td2, platform.python_version(), 64)

    def test_incompatible_plateform_wheel_file_raise(self):
        with TemporaryDirectory() as td1:
            open(os.path.join(td1, 'incompatiblewheel-1.0.0-py2.py3-none-linux_x86_64.whl'), 'w+')

            with TemporaryDirectory() as td2:
                with self.assertRaisesRegex(ValueError, '{0} is not compatible with Python {1} for Windows'
                .format('incompatiblewheel-1.0.0-py2.py3-none-linux_x86_64.whl', platform.python_version())):
                    fetch_pypi_wheels([], [os.path.join(td1, '*.whl')], td2, platform.python_version(), 64)

    def test_incompatible_python_wheel_file_raise(self):
        with TemporaryDirectory() as td1:
            open(os.path.join(td1, 'incompatiblewheel-1.0.0-py26-none-any.whl'), 'w+')

            with TemporaryDirectory() as td2:
                with self.assertRaisesRegex(ValueError, '{0} is not compatible with Python {1} for Windows'
                .format('incompatiblewheel-1.0.0-py26-none-any.whl', platform.python_version())):
                    fetch_pypi_wheels([], [os.path.join(td1, '*.whl')], td2, platform.python_version(), 64)

    def test_useless_wheel_glob_path_raise(self):
        with TemporaryDirectory() as td1:
            with TemporaryDirectory() as td2:
                with self.assertRaisesRegex(ValueError, 'does not match any wheel file'):
                    fetch_pypi_wheels([], [os.path.join(td1, '*.whl')], td2, platform.python_version(), 64)


# To exclude these, run:  nosetests -a '!network'
TestLocalWheels.test_matching_one_pattern.network = 1
TestLocalWheels.test_duplicate_wheel_files_raise.network = 1