def test_graph_bad_version_to_dot(self):
expected = (
('towel-stuff', 'bacon', 'bacon (<=0.2)'),
('grammar', 'bacon', 'truffles (>=1.2)'),
('choxie', 'towel-stuff', 'towel-stuff (0.1)'),
('banana', 'strawberry', 'strawberry (>=0.5)'),
)
dists = self.get_dists(
self.DISTROS_DIST + self.DISTROS_EGG + self.BAD_EGGS, True)
graph = make_graph(dists)
buf = StringIO()
graph.to_dot(buf)
buf.seek(0)
matches = []
lines = buf.readlines()
for line in lines[1:-1]: # skip the first and the last lines
if line[-1] == '\n':
line = line[:-1]
match = self.EDGE.match(line.strip())
self.assertIsNot(match, None)
matches.append(match.groups())
self.checkLists(matches, expected)
def test_requires_dist(self):
fields = {'name': 'project',
'version': '1.0',
'requires_dist': ['other', 'another (==1.0)']}
metadata = LegacyMetadata(mapping=fields)
self.assertEqual(metadata['Requires-Dist'],
['other', 'another (==1.0)'])
self.assertEqual(metadata['Metadata-Version'], '1.2')
self.assertNotIn('Provides', metadata)
self.assertEqual(metadata['Requires-Dist'],
['other', 'another (==1.0)'])
self.assertNotIn('Obsoletes', metadata)
# make sure write_file uses one RFC 822 header per item
fp = StringIO()
metadata.write_file(fp)
lines = fp.getvalue().split('\n')
self.assertIn('Requires-Dist: other', lines)
self.assertIn('Requires-Dist: another (==1.0)', lines)
# test warnings for invalid version constraints
# XXX this would cause no warnings if we used update (or the mapping
# argument of the constructor), see comment in LegacyMetadata.update
metadata = LegacyMetadata()
metadata['Requires-Dist'] = 'Funky (Groovie)'
metadata['Requires-Python'] = '1a-4'
self.assertEqual(len(self.get_logs()), 2)
# test multiple version matches
metadata = LegacyMetadata()
# XXX check PEP and see if 3 == 3.0
metadata['Requires-Python'] = '>=2.6, <3.0'
metadata['Requires-Dist'] = ['Foo (>=2.6, <3.0)']
self.assertEqual(self.get_logs(), [])
def test_graph_bad_version_to_dot(self):
expected = (
('towel-stuff', 'bacon', 'bacon (<=0.2)'),
('grammar', 'bacon', 'truffles (>=1.2)'),
('choxie', 'towel-stuff', 'towel-stuff (0.1)'),
('banana', 'strawberry', 'strawberry (>=0.5)'),
)
dists = self.get_dists(self.DISTROS_DIST + self.DISTROS_EGG +
self.BAD_EGGS, True)
graph = make_graph(dists)
buf = StringIO()
graph.to_dot(buf)
buf.seek(0)
matches = []
lines = buf.readlines()
for line in lines[1:-1]: # skip the first and the last lines
if line[-1] == '\n':
line = line[:-1]
match = self.EDGE.match(line.strip())
self.assertIsNot(match, None)
matches.append(match.groups())
self.checkLists(matches, expected)
def test_read_metadata(self):
fields = {
'name': 'project',
'version': '1.0',
'description': 'desc',
'summary': 'xxx',
'download_url': 'http://example.com',
'keywords': ['one', 'two'],
'requires_dist': ['foo']
}
metadata = LegacyMetadata(mapping=fields)
PKG_INFO = StringIO()
metadata.write_file(PKG_INFO)
PKG_INFO.seek(0)
metadata = LegacyMetadata(fileobj=PKG_INFO)
self.assertEqual(metadata['name'], 'project')
self.assertEqual(metadata['version'], '1.0')
self.assertEqual(metadata['summary'], 'xxx')
self.assertEqual(metadata['download_url'], 'http://example.com')
self.assertEqual(metadata['keywords'], ['one', 'two'])
self.assertEqual(metadata['platform'], [])
self.assertEqual(metadata['obsoletes'], [])
self.assertEqual(metadata['requires-dist'], ['foo'])
def test_graph_disconnected_to_dot(self):
dependencies_expected = (
('towel-stuff', 'bacon', 'bacon (<=0.2)'),
('grammar', 'bacon', 'truffles (>=1.2)'),
('choxie', 'towel-stuff', 'towel-stuff (0.1)'),
('banana', 'strawberry', 'strawberry (>=0.5)'),
)
disconnected_expected = ('cheese', 'bacon', 'strawberry')
dists = self.get_dists(self.DISTROS_DIST + self.DISTROS_EGG, True)
graph = make_graph(dists)
buf = StringIO()
graph.to_dot(buf, skip_disconnected=False)
buf.seek(0)
lines = buf.readlines()
dependencies_lines = []
disconnected_lines = []
# First sort output lines into dependencies and disconnected lines.
# We also skip the attribute lines, and don't include the "{" and "}"
# lines.
disconnected_active = False
for line in lines[1:-1]: # Skip first and last line
if line.startswith('subgraph disconnected'):
disconnected_active = True
continue
if line.startswith('}') and disconnected_active:
disconnected_active = False
continue
if disconnected_active:
# Skip the 'label = "Disconnected"', etc. attribute lines.
if ' = ' not in line:
disconnected_lines.append(line)
else:
dependencies_lines.append(line)
dependencies_matches = []
for line in dependencies_lines:
if line[-1] == '\n':
line = line[:-1]
match = self.EDGE.match(line.strip())
self.assertIsNot(match, None)
dependencies_matches.append(match.groups())
disconnected_matches = []
for line in disconnected_lines:
if line[-1] == '\n':
line = line[:-1]
line = line.strip('"')
disconnected_matches.append(line)
self.checkLists(dependencies_matches, dependencies_expected)
self.checkLists(disconnected_matches, disconnected_expected)
def test_graph_disconnected_to_dot(self):
dependencies_expected = (
('towel-stuff', 'bacon', 'bacon (<=0.2)'),
('grammar', 'bacon', 'truffles (>=1.2)'),
('choxie', 'towel-stuff', 'towel-stuff (0.1)'),
('banana', 'strawberry', 'strawberry (>=0.5)'),
)
disconnected_expected = ('cheese', 'bacon', 'strawberry')
dists = self.get_dists(self.DISTROS_DIST + self.DISTROS_EGG, True)
graph = make_graph(dists)
buf = StringIO()
graph.to_dot(buf, skip_disconnected=False)
buf.seek(0)
lines = buf.readlines()
dependencies_lines = []
disconnected_lines = []
# First sort output lines into dependencies and disconnected lines.
# We also skip the attribute lines, and don't include the "{" and "}"
# lines.
disconnected_active = False
for line in lines[1:-1]: # Skip first and last line
if line.startswith('subgraph disconnected'):
disconnected_active = True
continue
if line.startswith('}') and disconnected_active:
disconnected_active = False
continue
if disconnected_active:
# Skip the 'label = "Disconnected"', etc. attribute lines.
if ' = ' not in line:
disconnected_lines.append(line)
else:
dependencies_lines.append(line)
dependencies_matches = []
for line in dependencies_lines:
if line[-1] == '\n':
line = line[:-1]
match = self.EDGE.match(line.strip())
self.assertIsNot(match, None)
dependencies_matches.append(match.groups())
disconnected_matches = []
for line in disconnected_lines:
if line[-1] == '\n':
line = line[:-1]
line = line.strip('"')
disconnected_matches.append(line)
self.checkLists(dependencies_matches, dependencies_expected)
self.checkLists(disconnected_matches, disconnected_expected)
def test_metadata_read_write(self):
PKG_INFO = os.path.join(HERE, 'PKG-INFO')
metadata = LegacyMetadata(PKG_INFO)
out = StringIO()
metadata.write_file(out)
out.seek(0)
res = LegacyMetadata()
res.read_file(out)
self.assertEqual(metadata.values(), res.values())
def test_metadata_read_write(self):
PKG_INFO = os.path.join(HERE, 'PKG-INFO')
metadata = LegacyMetadata(PKG_INFO)
out = StringIO()
metadata.write_file(out)
out.seek(0)
res = LegacyMetadata()
res.read_file(out)
self.assertEqual(metadata.values(), res.values())
def test_instantiation(self):
PKG_INFO = os.path.join(HERE, 'PKG-INFO')
f = codecs.open(PKG_INFO, 'r', encoding='utf-8')
try:
contents = f.read()
finally:
f.close()
fp = StringIO(contents)
m = LegacyMetadata()
self.assertRaises(MetadataUnrecognizedVersionError, m.items)
m = LegacyMetadata(PKG_INFO)
self.assertEqual(len(m.items()), 22)
m = LegacyMetadata(fileobj=fp)
self.assertEqual(len(m.items()), 22)
m = LegacyMetadata(mapping=dict(name='Test', version='1.0'))
self.assertEqual(len(m.items()), 17)
d = dict(m.items())
self.assertRaises(TypeError, LegacyMetadata, PKG_INFO, fileobj=fp)
self.assertRaises(TypeError, LegacyMetadata, PKG_INFO, mapping=d)
self.assertRaises(TypeError, LegacyMetadata, fileobj=fp, mapping=d)
self.assertRaises(TypeError,
LegacyMetadata,
PKG_INFO,
mapping=m,
fileobj=fp)
def test_description_folding(self):
# make sure the indentation is preserved
out = StringIO()
desc = dedent("""\
example::
We start here
and continue here
and end here.
""")
metadata = Metadata()
metadata['description'] = desc
metadata.write_file(out)
folded_desc = desc.replace('\n', '\n' + (7 * ' ') + '|')
self.assertIn(folded_desc, out.getvalue())
def test_project_url(self):
metadata = LegacyMetadata()
metadata['Project-URL'] = [('one', 'http://ok')]
self.assertEqual(metadata['Project-URL'], [('one', 'http://ok')])
metadata.set_metadata_version()
self.assertEqual(metadata['Metadata-Version'], '1.2')
# make sure this particular field is handled properly when written
fp = StringIO()
metadata.write_file(fp)
self.assertIn('Project-URL: one,http://ok', fp.getvalue().split('\n'))
fp.seek(0)
metadata = LegacyMetadata()
metadata.read_file(fp)
self.assertEqual(metadata['Project-Url'], [('one', 'http://ok')])
def test_project_url(self):
metadata = LegacyMetadata()
metadata['Project-URL'] = [('one', 'http://ok')]
self.assertEqual(metadata['Project-URL'], [('one', 'http://ok')])
metadata.set_metadata_version()
self.assertEqual(metadata['Metadata-Version'], '1.2')
# make sure this particular field is handled properly when written
fp = StringIO()
metadata.write_file(fp)
self.assertIn('Project-URL: one,http://ok', fp.getvalue().split('\n'))
fp.seek(0)
metadata = LegacyMetadata()
metadata.read_file(fp)
self.assertEqual(metadata['Project-Url'], [('one', 'http://ok')])
def test_description_folding(self):
# make sure the indentation is preserved
out = StringIO()
desc = dedent("""\
example::
We start here
and continue here
and end here.
""")
metadata = LegacyMetadata()
metadata['description'] = desc
metadata.write_file(out)
# folded_desc = desc.replace('\n', '\n' + (7 * ' ') + '|')
folded_desc = desc.replace('\n', '\n' + (8 * ' '))
self.assertIn(folded_desc, out.getvalue())
def test_tarfile_vs_tar(self):
tmpdir, tmpdir2, base_name = self._create_files()
old_dir = os.getcwd()
os.chdir(tmpdir)
try:
_make_tarball(base_name, 'dist')
finally:
os.chdir(old_dir)
# check if the compressed tarball was created
tarball = base_name + '.tar.gz'
self.assertTrue(os.path.exists(tarball))
# now create another tarball using `tar`
tarball2 = os.path.join(tmpdir, 'archive2.tar.gz')
tar_cmd = ['tar', '-cf', 'archive2.tar', 'dist']
gzip_cmd = ['gzip', '-f9', 'archive2.tar']
old_dir = os.getcwd()
old_stdout = sys.stdout
os.chdir(tmpdir)
sys.stdout = StringIO()
try:
spawn(tar_cmd)
spawn(gzip_cmd)
finally:
os.chdir(old_dir)
sys.stdout = old_stdout
self.assertTrue(os.path.exists(tarball2))
# let's compare both tarballs
self.assertEqual(self._tarinfo(tarball), self._tarinfo(tarball2))
# trying an uncompressed one
base_name = os.path.join(tmpdir2, 'archive')
old_dir = os.getcwd()
os.chdir(tmpdir)
try:
_make_tarball(base_name, 'dist', compress=None)
finally:
os.chdir(old_dir)
tarball = base_name + '.tar'
self.assertTrue(os.path.exists(tarball))
# now for a dry_run
base_name = os.path.join(tmpdir2, 'archive')
old_dir = os.getcwd()
os.chdir(tmpdir)
try:
_make_tarball(base_name, 'dist', compress=None, dry_run=True)
finally:
os.chdir(old_dir)
tarball = base_name + '.tar'
self.assertTrue(os.path.exists(tarball))
def test_description(self):
content = self.get_file_contents('PKG-INFO')
metadata = LegacyMetadata()
metadata.read_file(StringIO(content))
# see if we can read the description now
DESC = os.path.join(HERE, 'LONG_DESC.txt')
f = open(DESC)
try:
wanted = f.read()
finally:
f.close()
self.assertEqual(wanted, metadata['Description'])
# save the file somewhere and make sure we can read it back
out = StringIO()
metadata.write_file(out)
out.seek(0)
out.seek(0)
metadata = LegacyMetadata()
metadata.read_file(out)
self.assertEqual(wanted, metadata['Description'])
def test_requires_dist(self):
fields = {
'name': 'project',
'version': '1.0',
'requires_dist': ['other', 'another (==1.0)']
}
metadata = LegacyMetadata(mapping=fields)
self.assertEqual(metadata['Requires-Dist'],
['other', 'another (==1.0)'])
self.assertEqual(metadata['Metadata-Version'], '1.2')
self.assertNotIn('Provides', metadata)
self.assertEqual(metadata['Requires-Dist'],
['other', 'another (==1.0)'])
self.assertNotIn('Obsoletes', metadata)
# make sure write_file uses one RFC 822 header per item
fp = StringIO()
metadata.write_file(fp)
lines = fp.getvalue().split('\n')
self.assertIn('Requires-Dist: other', lines)
self.assertIn('Requires-Dist: another (==1.0)', lines)
# test warnings for invalid version constraints
# XXX this would cause no warnings if we used update (or the mapping
# argument of the constructor), see comment in LegacyMetadata.update
metadata = LegacyMetadata()
metadata['Requires-Dist'] = 'Funky (Groovie)'
metadata['Requires-Python'] = '1a-4'
self.assertEqual(len(self.get_logs()), 2)
# test multiple version matches
metadata = LegacyMetadata()
# XXX check PEP and see if 3 == 3.0
metadata['Requires-Python'] = '>=2.6, <3.0'
metadata['Requires-Dist'] = ['Foo (>=2.6, <3.0)']
self.assertEqual(self.get_logs(), [])
def test_read_metadata(self):
fields = {'name': 'project',
'version': '1.0',
'description': 'desc',
'summary': 'xxx',
'download_url': 'http://example.com',
'keywords': ['one', 'two'],
'requires_dist': ['foo']}
metadata = LegacyMetadata(mapping=fields)
PKG_INFO = StringIO()
metadata.write_file(PKG_INFO)
PKG_INFO.seek(0)
metadata = LegacyMetadata(fileobj=PKG_INFO)
self.assertEqual(metadata['name'], 'project')
self.assertEqual(metadata['version'], '1.0')
self.assertEqual(metadata['summary'], 'xxx')
self.assertEqual(metadata['download_url'], 'http://example.com')
self.assertEqual(metadata['keywords'], ['one', 'two'])
self.assertEqual(metadata['platform'], [])
self.assertEqual(metadata['obsoletes'], [])
self.assertEqual(metadata['requires-dist'], ['foo'])
def test_graph_to_dot(self):
expected = (
("towel-stuff", "bacon", "bacon (<=0.2)"),
("grammar", "bacon", "truffles (>=1.2)"),
("choxie", "towel-stuff", "towel-stuff (0.1)"),
("banana", "strawberry", "strawberry (>=0.5)"),
)
dists = self.get_dists(self.DISTROS_DIST + self.DISTROS_EGG, True)
graph = make_graph(dists)
buf = StringIO()
graph.to_dot(buf)
buf.seek(0)
matches = []
lines = buf.readlines()
for line in lines[1:-1]: # skip the first and the last lines
if line[-1] == "\n":
line = line[:-1]
match = self.EDGE.match(line.strip())
self.assertIsNot(match, None)
matches.append(match.groups())
self.checkLists(matches, expected)
def test_mapping_api(self):
content = self.get_file_contents('PKG-INFO')
metadata = LegacyMetadata(fileobj=StringIO(content))
self.assertIn('Version', metadata.keys())
self.assertIn('0.5', metadata.values())
self.assertIn(('Version', '0.5'), metadata.items())
metadata.update({'version': '0.6'})
self.assertEqual(metadata['Version'], '0.6')
metadata.update([('version', '0.7')])
self.assertEqual(metadata['Version'], '0.7')
# use a kwarg to update
metadata.update(version='0.6')
self.assertEqual(metadata['Version'], '0.6')
# make sure update method checks values like the set method does
metadata.update({'version': '1--2'})
self.assertEqual(len(self.get_logs()), 1)
self.assertEqual(list(metadata), metadata.keys())
def test_description(self):
content = self.get_file_contents('PKG-INFO')
metadata = LegacyMetadata()
metadata.read_file(StringIO(content))
# see if we can read the description now
DESC = os.path.join(HERE, 'LONG_DESC.txt')
f = open(DESC)
try:
wanted = f.read()
finally:
f.close()
self.assertEqual(wanted, metadata['Description'])
# save the file somewhere and make sure we can read it back
out = StringIO()
metadata.write_file(out)
out.seek(0)
out.seek(0)
metadata = LegacyMetadata()
metadata.read_file(out)
self.assertEqual(wanted, metadata['Description'])
def test_main(self):
# just making sure _main() runs and returns things in the stdout
self.addCleanup(setattr, sys, 'stdout', sys.stdout)
sys.stdout = StringIO()
_main()
self.assertGreater(len(sys.stdout.getvalue().split('\n')), 0)
def test_attribute_access(self):
content = self.get_file_contents('PKG-INFO')
metadata = LegacyMetadata(fileobj=StringIO(content))
for attr in _ATTR2FIELD:
self.assertEqual(getattr(metadata, attr), metadata[attr])
