diff options
| -rw-r--r-- | setuptools/tests/test_packageindex.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/setuptools/tests/test_packageindex.py b/setuptools/tests/test_packageindex.py index 63b92946..7c2f2c84 100644 --- a/setuptools/tests/test_packageindex.py +++ b/setuptools/tests/test_packageindex.py @@ -6,6 +6,7 @@ import distutils.errors from setuptools.extern import six from setuptools.extern.six.moves import urllib, http_client +import mock import pkg_resources import setuptools.package_index @@ -224,6 +225,41 @@ class TestPackageIndex: assert dists[1].version == vc + def test_download_git(self): + index = setuptools.package_index.PackageIndex( + hosts=('www.example.com',) + ) + + index._vcs_split_rev_from_url = mock.Mock() + url = 'https://example.com/bar' + rev = '2995' + + index._vcs_split_rev_from_url.return_value = (url, rev) + + filename = "somefile.py" + + with mock.patch("os.system") as os_system_mock: + + result = index._download_git(url, filename) + + os_system_mock.assert_called() + + assert os_system_mock.call_args_list[0][0] \ + == ("git clone --quiet %s %s" % (url, filename), ) + + assert os_system_mock.call_args_list[1][0] \ + == ("(cd %s && git checkout --quiet %s)" % (filename, rev), ) + assert result == filename + + index._vcs_split_rev_from_url.return_value = (url, None) + + with mock.patch("os.system") as os_system_mock: + + index._download_git(url, filename) + + os_system_mock.assert_called_once_with("git clone --quiet %s %s" % (url, filename)) + + class TestContentCheckers: def test_md5(self): checker = setuptools.package_index.HashChecker.from_url( |