def test_run_stack_info_run_multi(self, context: Context) -> None:
"""Check assigning stack info only once on multiple calls."""
prescription_str = """
name: SieveUnit
type: sieve
should_include:
times: 1
adviser_pipeline: true
match:
package_version:
name: flask
version: '>1.0,<=1.1.0'
index_url: 'https://pypi.org/simple'
run:
stack_info:
- type: WARNING
message: Some stack warning message printed by a sieve
link: https://thoth-station.ninja
"""
prescription = yaml.safe_load(prescription_str)
PRESCRIPTION_SIEVE_SCHEMA(prescription)
SievePrescription.set_prescription(prescription)
package_version = PackageVersion(
name="flask",
version="==1.1.0",
index=Source("https://pypi.org/simple"),
develop=False,
)
assert not context.stack_info
unit = SievePrescription()
unit.pre_run()
with unit.assigned_context(context):
result = list(unit.run((pv for pv in (package_version,))))
assert not result
result = list(unit.run((pv for pv in (package_version,))))
assert not result
assert self.verify_justification_schema(context.stack_info)
assert len(context.stack_info) == 1, "Stack information should be added only once"
assert context.stack_info == unit.run_prescription["stack_info"]
def test_sieve_not_index_url(self, context: Context) -> None:
"""Test sieving packages based on "not" index_url."""
prescription_str = """
name: SieveUnit
type: sieve
should_include:
times: 1
adviser_pipeline: true
match:
package_version:
name: flask
index_url:
not: https://pypi.org/simple
"""
prescription = yaml.safe_load(prescription_str)
PRESCRIPTION_SIEVE_SCHEMA(prescription)
SievePrescription.set_prescription(prescription)
package_versions = [
PackageVersion(
name="flask",
version="==1.1.0",
index=Source("https://thoth-station.ninja/simple"),
develop=False,
),
PackageVersion(
name="flask",
version="==1.1.2",
index=Source("https://pypi.org/simple"),
develop=False,
),
PackageVersion(
name="flask",
version="==1.0.2",
index=Source("https://pypi.org/simple"),
develop=False,
),
]
unit = SievePrescription()
unit.pre_run()
with unit.assigned_context(context):
result = list(unit.run((pv for pv in package_versions)))
assert len(result) == 2
assert result == [package_versions[1], package_versions[2]]
def test_sieve_develop_true(self, context: Context, develop: bool) -> None:
"""Test sieving packages based on develop flag set to true."""
prescription_str = f"""
name: SieveUnit
type: sieve
should_include:
times: 1
adviser_pipeline: true
match:
package_version:
name: flask
develop: {'true' if develop else 'false'}
"""
prescription = yaml.safe_load(prescription_str)
PRESCRIPTION_SIEVE_SCHEMA(prescription)
SievePrescription.set_prescription(prescription)
package_versions = [
PackageVersion(
name="flask",
version="==1.1.2",
index=Source("https://pypi.org/simple"),
develop=develop,
),
PackageVersion(
name="flask",
version="==1.0.2",
index=Source("https://pypi.org/simple"),
develop=not develop,
),
]
unit = SievePrescription()
unit.pre_run()
with unit.assigned_context(context):
result = list(unit.run((pv for pv in package_versions)))
assert len(result) == 1
assert [result[0]] == [pv for pv in package_versions if pv.develop != develop]
