def test_simple(self):
prefix = 'prefix'
version = 'version'
source_dir = 'source dir'
names = ['requirements.txt']
requirements = [
req.InstallRequirement.from_line('some-package>1'),
req.InstallRequirement.from_line('other-package'),
]
self.mox.StubOutWithMock(deps, 'get_requirement_filenames')
deps.get_requirement_filenames(
where=source_dir, only=None, group=prefix,
version=version).AndReturn(names)
self.mox.StubOutWithMock(req, 'parse_requirements')
req.parse_requirements(
'requirements.txt',
options=mox.IgnoreArg()).AndReturn(requirements)
self.mox.StubOutWithMock(os.path, 'exists')
os.path.exists('requirements.txt').AndReturn(True)
self.mox.ReplayAll()
reqs = deps.find_requirements(source_dir, group=prefix,
version=version)
self.mox.VerifyAll()
self.assertEqual(['some-package>1', 'other-package'], reqs)
def test_only_with_list(self):
names = deps.get_requirement_filenames(only=['a', 'b'], version='')
self.assertEqual([
'a-requirements.txt',
'b-requirements.txt',
], names)
def test_source_dir(self):
names = list(deps.get_requirement_filenames(where="/tmp",
version="2"))
self.assertEqual([
'/tmp/requirements.txt',
'/tmp/requirements-2.txt',
], names)
def test_version_as_str(self):
names = list(deps.get_requirement_filenames(version="2.1"))
self.assertEqual([
'requirements.txt',
'requirements-2.txt',
'requirements-21.txt',
], names)
def test_no_prefix(self):
names = list(deps.get_requirement_filenames(version=(2, 7, 3)))
self.assertEqual([
'requirements.txt',
'requirements-2.txt',
'requirements-27.txt',
'requirements-273.txt',
], names)
def test_zeros_should_be_included(self):
names = deps.get_requirement_filenames(version=[1, 0, 1])
self.assertEqual([
'requirements.txt',
'requirements-1.txt',
'requirements-10.txt',
'requirements-101.txt',
], names)
def test_version_info(self):
names = list(deps.get_requirement_filenames(group='prod',
version=sys.version_info))
self.assertIn('requirements.txt', names)
self.assertIn('requirements-%s.txt' % sys.version_info[0], names)
self.assertIn('prod-requirements.txt', names)
self.assertIn('prod-requirements-%s.txt' % sys.version_info[0],
names)
def test_prefix_and_version(self):
names = list(deps.get_requirement_filenames(group='devel',
version=(2, 7, 3)))
self.assertEqual([
'requirements.txt',
'requirements-2.txt',
'requirements-27.txt',
'requirements-273.txt',
'devel-requirements.txt',
'devel-requirements-2.txt',
'devel-requirements-27.txt',
'devel-requirements-273.txt',
], names)
self.assertIsInstance(names, list)
def test_empty(self):
names = list(deps.get_requirement_filenames(group='', version=''))
self.assertEqual(['requirements.txt'], names)
def test_only_and_group_ValueError(self):
with self.assertRaises(ValueError):
deps.get_requirement_filenames(only='a', group='b')
def test_only(self):
names = deps.get_requirement_filenames(only='setup', version='')
self.assertEqual([
'setup-requirements.txt',
], names)
