| # -*- encoding: utf8 -*- | |
| """Tests for distutils.command.upload.""" | |
| import os | |
| import unittest | |
| from test.test_support import run_unittest | |
| from distutils.command import upload as upload_mod | |
| from distutils.command.upload import upload | |
| from distutils.core import Distribution | |
| from distutils.tests.test_config import PYPIRC, PyPIRCCommandTestCase | |
| PYPIRC_LONG_PASSWORD = """\ | |
| [distutils] | |
| index-servers = | |
| server1 | |
| server2 | |
| [server1] | |
| username:me | |
| password:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa | |
| [server2] | |
| username:meagain | |
| password: secret | |
| realm:acme | |
| repository:http://another.pypi/ | |
| """ | |
| PYPIRC_NOPASSWORD = """\ | |
| [distutils] | |
| index-servers = | |
| server1 | |
| [server1] | |
| username:me | |
| """ | |
| class FakeOpen(object): | |
| def __init__(self, url): | |
| self.url = url | |
| if not isinstance(url, str): | |
| self.req = url | |
| else: | |
| self.req = None | |
| self.msg = 'OK' | |
| def getcode(self): | |
| return 200 | |
| class uploadTestCase(PyPIRCCommandTestCase): | |
| def setUp(self): | |
| super(uploadTestCase, self).setUp() | |
| self.old_open = upload_mod.urlopen | |
| upload_mod.urlopen = self._urlopen | |
| self.last_open = None | |
| def tearDown(self): | |
| upload_mod.urlopen = self.old_open | |
| super(uploadTestCase, self).tearDown() | |
| def _urlopen(self, url): | |
| self.last_open = FakeOpen(url) | |
| return self.last_open | |
| def test_finalize_options(self): | |
| # new format | |
| self.write_file(self.rc, PYPIRC) | |
| dist = Distribution() | |
| cmd = upload(dist) | |
| cmd.finalize_options() | |
| for attr, waited in (('username', 'me'), ('password', 'secret'), | |
| ('realm', 'pypi'), | |
| ('repository', 'http://pypi.python.org/pypi')): | |
| self.assertEqual(getattr(cmd, attr), waited) | |
| def test_saved_password(self): | |
| # file with no password | |
| self.write_file(self.rc, PYPIRC_NOPASSWORD) | |
| # make sure it passes | |
| dist = Distribution() | |
| cmd = upload(dist) | |
| cmd.finalize_options() | |
| self.assertEqual(cmd.password, None) | |
| # make sure we get it as well, if another command | |
| # initialized it at the dist level | |
| dist.password = 'xxx' | |
| cmd = upload(dist) | |
| cmd.finalize_options() | |
| self.assertEqual(cmd.password, 'xxx') | |
| def test_upload(self): | |
| tmp = self.mkdtemp() | |
| path = os.path.join(tmp, 'xxx') | |
| self.write_file(path) | |
| command, pyversion, filename = 'xxx', '2.6', path | |
| dist_files = [(command, pyversion, filename)] | |
| self.write_file(self.rc, PYPIRC_LONG_PASSWORD) | |
| # lets run it | |
| pkg_dir, dist = self.create_dist(dist_files=dist_files, author=u'dédé') | |
| cmd = upload(dist) | |
| cmd.ensure_finalized() | |
| cmd.run() | |
| # what did we send ? | |
| self.assertIn('dédé', self.last_open.req.data) | |
| headers = dict(self.last_open.req.headers) | |
| self.assertEqual(headers['Content-length'], '2085') | |
| self.assertTrue(headers['Content-type'].startswith('multipart/form-data')) | |
| self.assertEqual(self.last_open.req.get_method(), 'POST') | |
| self.assertEqual(self.last_open.req.get_full_url(), | |
| 'http://pypi.python.org/pypi') | |
| self.assertTrue('xxx' in self.last_open.req.data) | |
| auth = self.last_open.req.headers['Authorization'] | |
| self.assertFalse('\n' in auth) | |
| def test_suite(): | |
| return unittest.makeSuite(uploadTestCase) | |
| if __name__ == "__main__": | |
| run_unittest(test_suite()) |