def _capture_annotated_out_reqlines(results, resolver, constraints,
format_control):
with NamedTemporaryFile(delete=False) as tmp_file:
tmp_file_name = tmp_file.name
try:
writer = OutputWriter(src_files=None,
dst_file=tmp_file_name,
dry_run=False,
emit_header=False,
emit_index=False,
emit_trusted_host=False,
annotate=True,
generate_hashes=False,
default_index_url=None,
index_urls=None,
trusted_hosts=(),
format_control=format_control)
writer.write(
results=results,
reverse_dependencies=resolver.reverse_dependencies(results),
unsafe_requirements=resolver.unsafe_constraints,
primary_packages={key_from_ireq(ireq)
for ireq in constraints},
markers={},
hashes=None)
with open(tmp_file_name, 'r') as tmp_file:
return tmp_file.readlines()
finally:
os.unlink(tmp_file_name)
def test_iter_lines__unsafe_dependencies(from_line, allow_unsafe):
writer = OutputWriter(
src_files=["src_file", "src_file2"], dst_file="dst_file",
dry_run=True,
emit_header=True, emit_index=True, emit_trusted_host=True,
annotate=True,
generate_hashes=False,
default_index_url=None, index_urls=[],
trusted_hosts=[],
format_control=FormatControl(set(), set()),
allow_unsafe=allow_unsafe,
)
ireq = [from_line('test==1.2')]
unsafe_req = [from_line('setuptools')]
reverse_dependencies = {'test': ['xyz']}
str_lines = list(writer._iter_lines(
ireq,
unsafe_req,
reverse_dependencies,
['test'],
{},
None,
))
assert comment('# The following packages are considered to be unsafe in a requirements file:') in str_lines
if allow_unsafe:
assert 'setuptools' in str_lines
else:
assert comment('# setuptools') in str_lines
assert 'test==1.2' in str_lines
def writer(tmpdir_cwd):
with open("src_file", "w"), open("src_file2", "w"):
pass
cli_args = [
"--dry-run",
"--output-file",
"requirements.txt",
"src_file",
"src_file2",
]
with cli.make_context("pip-compile", cli_args) as ctx:
writer = OutputWriter(
src_files=["src_file", "src_file2"],
dst_file=ctx.params["output_file"],
click_ctx=ctx,
dry_run=True,
emit_header=True,
emit_index=True,
emit_trusted_host=True,
annotate=True,
generate_hashes=False,
default_index_url=None,
index_urls=[],
trusted_hosts=[],
format_control=FormatControl(set(), set()),
allow_unsafe=False,
find_links=[],
emit_find_links=True,
)
yield writer
def writer():
return OutputWriter(src_files=["src_file", "src_file2"], dst_file="dst_file", dry_run=True,
emit_header=True, emit_index=True, annotate=True,
generate_hashes=False,
default_index_url=None, index_urls=[],
trusted_hosts=[],
format_control=FormatControl(set(), set()))
def writer():
return OutputWriter(src_file="src_file",
dry_run=True,
emit_header=True,
emit_index=True,
annotate=True,
default_index_url=None,
index_urls=[])
def writer(tmp_file):
return OutputWriter(
src_files=["src_file", "src_file2"],
dst_file=tmp_file,
dry_run=True,
emit_header=True,
emit_index=True,
emit_trusted_host=True,
annotate=True,
generate_hashes=False,
default_index_url=None,
index_urls=[],
trusted_hosts=[],
format_control=FormatControl(set(), set()),
allow_unsafe=False,
find_links=[],
)