def test_run_match_develop(self, context: Context, state: State, develop: bool) -> None:
"""Test running this pipeline unit based on develop matching."""
prescription_str = f"""
name: StrideUnit
type: stride
should_include:
times: 1
adviser_pipeline: true
match:
state:
resolved_dependencies:
- name: flask
develop: {'true' if develop else 'false'}
run:
stack_info:
- type: INFO
message: This message will be shown
link: https://thoth-station.ninja
"""
prescription = yaml.safe_load(prescription_str)
PRESCRIPTION_STRIDE_SCHEMA(prescription)
StridePrescription.set_prescription(prescription)
package_version = PackageVersion(
name="flask",
version="==2.0.1",
index=Source("https://pypi.org/simple"),
develop=develop,
)
state.add_resolved_dependency(package_version.to_tuple())
context.register_package_version(package_version)
assert not context.stack_info
unit = StridePrescription()
unit.pre_run()
with unit.assigned_context(context):
assert unit.run(state) is None
# Run one more time to verify the stack info is added just once.
assert unit.run(state) is None
assert context.stack_info == [
{"type": "INFO", "message": "This message will be shown", "link": "https://thoth-station.ninja"}
]
def test_run_no_match(self, context: Context, state: State) -> None:
"""Test running this pipeline unit without match."""
prescription_str = """
name: StrideUnit
type: stride
should_include:
times: 1
adviser_pipeline: true
match:
state:
resolved_dependencies:
- name: flask
version: "<=1.0.0,>=0.12"
index_url: "https://pypi.org/simple"
- name: connexion
version: "==2.7.0"
index_url: "https://pypi.org/simple"
run:
stack_info:
- type: ERROR
message: This message will not be shown
link: https://pypi.org/project/connexion
"""
prescription = yaml.safe_load(prescription_str)
PRESCRIPTION_STRIDE_SCHEMA(prescription)
StridePrescription.set_prescription(prescription)
state.add_resolved_dependency(
("flask", "0.12", "https://pypi.org/simple"))
state.add_resolved_dependency(
("connexion", "2.0.0", "https://pypi.org/simple"))
assert not context.stack_info
unit = StridePrescription()
unit.pre_run()
with unit.assigned_context(context):
assert unit.run(state) is None
assert not context.stack_info