14bd0632c0
Since we only ever handle one requirement here, the parse_requirements generator (which btw needs a next m() is not the best choice. Also extracted install command cleanup into class method and added tests for that. cc @markwal
80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
# coding=utf-8
|
|
from __future__ import absolute_import
|
|
|
|
__license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
|
|
__copyright__ = "Copyright (C) 2017 The OctoPrint Project - Released under terms of the AGPLv3 License"
|
|
|
|
import unittest
|
|
import ddt
|
|
import mock
|
|
|
|
import octoprint.util.pip
|
|
|
|
import site
|
|
import pkg_resources
|
|
|
|
@ddt.ddt
|
|
class PipCallerTest(unittest.TestCase):
|
|
|
|
@ddt.data(
|
|
# remove --process-dependency-links for versions < 1.5
|
|
(["install", "--process-dependency-links", "http://example.com/foo.zip"], "1.1", True, False, False,
|
|
True,
|
|
["install", "http://example.com/foo.zip"]),
|
|
|
|
# keep --process-dependency-links for versions >= 1.5, --no-use-wheel for ==1.5.0
|
|
(["install", "--process-dependency-links", "http://example.com/foo.zip"], "1.5", True, False, False,
|
|
True,
|
|
["install", "--process-dependency-links", "http://example.com/foo.zip", "--no-use-wheel"]),
|
|
|
|
# keep --process-dependency-links for versions >= 1.5
|
|
(["install", "--process-dependency-links", "http://example.com/foo.zip"], "9.0.1", True, False, False,
|
|
True,
|
|
["install", "--process-dependency-links", "http://example.com/foo.zip"]),
|
|
|
|
# remove --user in virtual env
|
|
(["install", "--user", "http://example.com/foo.zip"], "9.0.1", True, False, False,
|
|
True,
|
|
["install", "http://example.com/foo.zip"]),
|
|
|
|
# ignore use_user in virtual env
|
|
(["install", "http://example.com/foo.zip"], "9.0.1", True, True, False,
|
|
True,
|
|
["install", "http://example.com/foo.zip"]),
|
|
|
|
# ignore force_user in virtual env
|
|
(["install", "http://example.com/foo.zip"], "9.0.1", True, False, True,
|
|
True,
|
|
["install", "http://example.com/foo.zip"]),
|
|
|
|
# remove --user with disabled user_site
|
|
(["install", "--user", "http://example.com/foo.zip"], "9.0.1", False, False, False,
|
|
False,
|
|
["install", "http://example.com/foo.zip"]),
|
|
|
|
# add --user when not in virtual env and use_user is True
|
|
(["install", "http://example.com/foo.zip"], "9.0.1", False, True, False,
|
|
True,
|
|
["install", "http://example.com/foo.zip", "--user"]),
|
|
|
|
# ignore use_user with disabled user_site
|
|
(["install", "http://example.com/foo.zip"], "9.0.1", False, True, False,
|
|
False,
|
|
["install", "http://example.com/foo.zip"]),
|
|
|
|
# add --user when not in virtual env and force_user is True
|
|
(["install", "http://example.com/foo.zip"], "9.0.1", False, False, True,
|
|
True,
|
|
["install", "http://example.com/foo.zip", "--user"]),
|
|
|
|
# ignore force_user with disabled user_site
|
|
(["install", "http://example.com/foo.zip"], "9.0.1", False, False, True,
|
|
False,
|
|
["install", "http://example.com/foo.zip"]),
|
|
)
|
|
@ddt.unpack
|
|
def test_clean_install_command(self, args, version, virtual_env, use_user, force_user, user_site, expected):
|
|
with mock.patch.object(site, "ENABLE_USER_SITE", user_site):
|
|
parsed = pkg_resources.parse_version(version)
|
|
actual = octoprint.util.pip.PipCaller.clean_install_command(args, parsed, virtual_env, use_user, force_user)
|
|
self.assertEquals(expected, actual)
|