291 lines
8.6 KiB
Python
291 lines
8.6 KiB
Python
|
# -*- encoding: utf8 -*-
|
||
|
|
"""Tests for distutils.command.register."""
|
||
|
|
import os
|
||
|
|
import unittest
|
||
|
|
import getpass
|
||
|
|
import urllib2
|
||
|
|
import warnings
|
||
|
|
|
||
|
|
from test.test_support import check_warnings, run_unittest
|
||
|
|
|
||
|
|
from distutils.command import register as register_module
|
||
|
|
from distutils.command.register import register
|
||
|
|
from distutils.errors import DistutilsSetupError
|
||
|
|
|
||
|
|
from distutils.tests.test_config import PyPIRCCommandTestCase
|
||
|
|
|
||
|
|
try:
|
||
|
|
import docutils
|
||
|
|
except ImportError:
|
||
|
|
docutils = None
|
||
|
|
|
||
|
|
PYPIRC_NOPASSWORD = """\
|
||
|
|
[distutils]
|
||
|
|
|
||
|
|
index-servers =
|
||
|
|
server1
|
||
|
|
|
||
|
|
[server1]
|
||
|
|
username:me
|
||
|
|
"""
|
||
|
|
|
||
|
|
WANTED_PYPIRC = """\
|
||
|
|
[distutils]
|
||
|
|
index-servers =
|
||
|
|
pypi
|
||
|
|
|
||
|
|
[pypi]
|
||
|
|
username:tarek
|
||
|
|
password:password
|
||
|
|
"""
|
||
|
|
|
||
|
|
class RawInputs(object):
|
||
|
|
"""Fakes user inputs."""
|
||
|
|
def __init__(self, *answers):
|
||
|
|
self.answers = answers
|
||
|
|
self.index = 0
|
||
|
|
|
||
|
|
def __call__(self, prompt=''):
|
||
|
|
try:
|
||
|
|
return self.answers[self.index]
|
||
|
|
finally:
|
||
|
|
self.index += 1
|
||
|
|
|
||
|
|
class FakeOpener(object):
|
||
|
|
"""Fakes a PyPI server"""
|
||
|
|
def __init__(self):
|
||
|
|
self.reqs = []
|
||
|
|
|
||
|
|
def __call__(self, *args):
|
||
|
|
return self
|
||
|
|
|
||
|
|
def open(self, req):
|
||
|
|
self.reqs.append(req)
|
||
|
|
return self
|
||
|
|
|
||
|
|
def read(self):
|
||
|
|
return 'xxx'
|
||
|
|
|
||
|
|
class RegisterTestCase(PyPIRCCommandTestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super(RegisterTestCase, self).setUp()
|
||
|
|
# patching the password prompt
|
||
|
|
self._old_getpass = getpass.getpass
|
||
|
|
def _getpass(prompt):
|
||
|
|
return 'password'
|
||
|
|
getpass.getpass = _getpass
|
||
|
|
self.old_opener = urllib2.build_opener
|
||
|
|
self.conn = urllib2.build_opener = FakeOpener()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
getpass.getpass = self._old_getpass
|
||
|
|
urllib2.build_opener = self.old_opener
|
||
|
|
super(RegisterTestCase, self).tearDown()
|
||
|
|
|
||
|
|
def _get_cmd(self, metadata=None):
|
||
|
|
if metadata is None:
|
||
|
|
metadata = {'url': 'xxx', 'author': 'xxx',
|
||
|
|
'author_email': 'xxx',
|
||
|
|
'name': 'xxx', 'version': 'xxx'}
|
||
|
|
pkg_info, dist = self.create_dist(**metadata)
|
||
|
|
return register(dist)
|
||
|
|
|
||
|
|
def test_create_pypirc(self):
|
||
|
|
# this test makes sure a .pypirc file
|
||
|
|
# is created when requested.
|
||
|
|
|
||
|
|
# let's create a register instance
|
||
|
|
cmd = self._get_cmd()
|
||
|
|
|
||
|
|
# we shouldn't have a .pypirc file yet
|
||
|
|
self.assertTrue(not os.path.exists(self.rc))
|
||
|
|
|
||
|
|
# patching raw_input and getpass.getpass
|
||
|
|
# so register gets happy
|
||
|
|
#
|
||
|
|
# Here's what we are faking :
|
||
|
|
# use your existing login (choice 1.)
|
||
|
|
# Username : 'tarek'
|
||
|
|
# Password : 'password'
|
||
|
|
# Save your login (y/N)? : 'y'
|
||
|
|
inputs = RawInputs('1', 'tarek', 'y')
|
||
|
|
register_module.raw_input = inputs.__call__
|
||
|
|
# let's run the command
|
||
|
|
try:
|
||
|
|
cmd.run()
|
||
|
|
finally:
|
||
|
|
del register_module.raw_input
|
||
|
|
|
||
|
|
# we should have a brand new .pypirc file
|
||
|
|
self.assertTrue(os.path.exists(self.rc))
|
||
|
|
|
||
|
|
# with the content similar to WANTED_PYPIRC
|
||
|
|
f = open(self.rc)
|
||
|
|
try:
|
||
|
|
content = f.read()
|
||
|
|
self.assertEqual(content, WANTED_PYPIRC)
|
||
|
|
finally:
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
# now let's make sure the .pypirc file generated
|
||
|
|
# really works : we shouldn't be asked anything
|
||
|
|
# if we run the command again
|
||
|
|
def _no_way(prompt=''):
|
||
|
|
raise AssertionError(prompt)
|
||
|
|
register_module.raw_input = _no_way
|
||
|
|
|
||
|
|
cmd.show_response = 1
|
||
|
|
cmd.run()
|
||
|
|
|
||
|
|
# let's see what the server received : we should
|
||
|
|
# have 2 similar requests
|
||
|
|
self.assertEqual(len(self.conn.reqs), 2)
|
||
|
|
req1 = dict(self.conn.reqs[0].headers)
|
||
|
|
req2 = dict(self.conn.reqs[1].headers)
|
||
|
|
self.assertEqual(req2['Content-length'], req1['Content-length'])
|
||
|
|
self.assertTrue('xxx' in self.conn.reqs[1].data)
|
||
|
|
|
||
|
|
def test_password_not_in_file(self):
|
||
|
|
|
||
|
|
self.write_file(self.rc, PYPIRC_NOPASSWORD)
|
||
|
|
cmd = self._get_cmd()
|
||
|
|
cmd._set_config()
|
||
|
|
cmd.finalize_options()
|
||
|
|
cmd.send_metadata()
|
||
|
|
|
||
|
|
# dist.password should be set
|
||
|
|
# therefore used afterwards by other commands
|
||
|
|
self.assertEqual(cmd.distribution.password, 'password')
|
||
|
|
|
||
|
|
def test_registering(self):
|
||
|
|
# this test runs choice 2
|
||
|
|
cmd = self._get_cmd()
|
||
|
|
inputs = RawInputs('2', 'tarek', 'tarek@ziade.org')
|
||
|
|
register_module.raw_input = inputs.__call__
|
||
|
|
try:
|
||
|
|
# let's run the command
|
||
|
|
cmd.run()
|
||
|
|
finally:
|
||
|
|
del register_module.raw_input
|
||
|
|
|
||
|
|
# we should have send a request
|
||
|
|
self.assertEqual(len(self.conn.reqs), 1)
|
||
|
|
req = self.conn.reqs[0]
|
||
|
|
headers = dict(req.headers)
|
||
|
|
self.assertEqual(headers['Content-length'], '608')
|
||
|
|
self.assertTrue('tarek' in req.data)
|
||
|
|
|
||
|
|
def test_password_reset(self):
|
||
|
|
# this test runs choice 3
|
||
|
|
cmd = self._get_cmd()
|
||
|
|
inputs = RawInputs('3', 'tarek@ziade.org')
|
||
|
|
register_module.raw_input = inputs.__call__
|
||
|
|
try:
|
||
|
|
# let's run the command
|
||
|
|
cmd.run()
|
||
|
|
finally:
|
||
|
|
del register_module.raw_input
|
||
|
|
|
||
|
|
# we should have send a request
|
||
|
|
self.assertEqual(len(self.conn.reqs), 1)
|
||
|
|
req = self.conn.reqs[0]
|
||
|
|
headers = dict(req.headers)
|
||
|
|
self.assertEqual(headers['Content-length'], '290')
|
||
|
|
self.assertTrue('tarek' in req.data)
|
||
|
|
|
||
|
|
@unittest.skipUnless(docutils is not None, 'needs docutils')
|
||
|
|
def test_strict(self):
|
||
|
|
# testing the script option
|
||
|
|
# when on, the register command stops if
|
||
|
|
# the metadata is incomplete or if
|
||
|
|
# long_description is not reSt compliant
|
||
|
|
|
||
|
|
# empty metadata
|
||
|
|
cmd = self._get_cmd({})
|
||
|
|
cmd.ensure_finalized()
|
||
|
|
cmd.strict = 1
|
||
|
|
self.assertRaises(DistutilsSetupError, cmd.run)
|
||
|
|
|
||
|
|
# metadata are OK but long_description is broken
|
||
|
|
metadata = {'url': 'xxx', 'author': 'xxx',
|
||
|
|
'author_email': u'éxéxé',
|
||
|
|
'name': 'xxx', 'version': 'xxx',
|
||
|
|
'long_description': 'title\n==\n\ntext'}
|
||
|
|
|
||
|
|
cmd = self._get_cmd(metadata)
|
||
|
|
cmd.ensure_finalized()
|
||
|
|
cmd.strict = 1
|
||
|
|
self.assertRaises(DistutilsSetupError, cmd.run)
|
||
|
|
|
||
|
|
# now something that works
|
||
|
|
metadata['long_description'] = 'title\n=====\n\ntext'
|
||
|
|
cmd = self._get_cmd(metadata)
|
||
|
|
cmd.ensure_finalized()
|
||
|
|
cmd.strict = 1
|
||
|
|
inputs = RawInputs('1', 'tarek', 'y')
|
||
|
|
register_module.raw_input = inputs.__call__
|
||
|
|
# let's run the command
|
||
|
|
try:
|
||
|
|
cmd.run()
|
||
|
|
finally:
|
||
|
|
del register_module.raw_input
|
||
|
|
|
||
|
|
# strict is not by default
|
||
|
|
cmd = self._get_cmd()
|
||
|
|
cmd.ensure_finalized()
|
||
|
|
inputs = RawInputs('1', 'tarek', 'y')
|
||
|
|
register_module.raw_input = inputs.__call__
|
||
|
|
# let's run the command
|
||
|
|
try:
|
||
|
|
cmd.run()
|
||
|
|
finally:
|
||
|
|
del register_module.raw_input
|
||
|
|
|
||
|
|
# and finally a Unicode test (bug #12114)
|
||
|
|
metadata = {'url': u'xxx', 'author': u'\u00c9ric',
|
||
|
|
'author_email': u'xxx', u'name': 'xxx',
|
||
|
|
'version': u'xxx',
|
||
|
|
'description': u'Something about esszet \u00df',
|
||
|
|
'long_description': u'More things about esszet \u00df'}
|
||
|
|
|
||
|
|
cmd = self._get_cmd(metadata)
|
||
|
|
cmd.ensure_finalized()
|
||
|
|
cmd.strict = 1
|
||
|
|
inputs = RawInputs('1', 'tarek', 'y')
|
||
|
|
register_module.raw_input = inputs.__call__
|
||
|
|
# let's run the command
|
||
|
|
try:
|
||
|
|
cmd.run()
|
||
|
|
finally:
|
||
|
|
del register_module.raw_input
|
||
|
|
|
||
|
|
@unittest.skipUnless(docutils is not None, 'needs docutils')
|
||
|
|
def test_register_invalid_long_description(self):
|
||
|
|
description = ':funkie:`str`' # mimic Sphinx-specific markup
|
||
|
|
metadata = {'url': 'xxx', 'author': 'xxx',
|
||
|
|
'author_email': 'xxx',
|
||
|
|
'name': 'xxx', 'version': 'xxx',
|
||
|
|
'long_description': description}
|
||
|
|
cmd = self._get_cmd(metadata)
|
||
|
|
cmd.ensure_finalized()
|
||
|
|
cmd.strict = True
|
||
|
|
inputs = RawInputs('2', 'tarek', 'tarek@ziade.org')
|
||
|
|
register_module.raw_input = inputs
|
||
|
|
self.addCleanup(delattr, register_module, 'raw_input')
|
||
|
|
self.assertRaises(DistutilsSetupError, cmd.run)
|
||
|
|
|
||
|
|
def test_check_metadata_deprecated(self):
|
||
|
|
# makes sure make_metadata is deprecated
|
||
|
|
cmd = self._get_cmd()
|
||
|
|
with check_warnings() as w:
|
||
|
|
warnings.simplefilter("always")
|
||
|
|
cmd.check_metadata()
|
||
|
|
self.assertEqual(len(w.warnings), 1)
|
||
|
|
|
||
|
|
def test_suite():
|
||
|
|
return unittest.makeSuite(RegisterTestCase)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
run_unittest(test_suite())
|