def dependency_graph(
draw,
python_packages=lists(requirement(), unique_by=lambda x: x.name()),
external_dependencies=lists(external_dependency()),
is_runtime_dependency=booleans(),
selections=integers(),
):
graph = DependencyGraph()
packages = draw(python_packages)
if not packages:
return graph
for package in packages:
index = draw(selections) % len(packages)
try:
if draw(is_runtime_dependency):
graph.set_runtime_dependency(dependent=package,
dependency=packages[index])
else:
graph.set_buildtime_dependency(dependent=package,
dependency=packages[index])
except CyclicDependencyOccured:
continue
for dependency in draw(external_dependencies):
graph.set_external_dependency(dependent=packages[draw(selections) %
len(packages)],
dependency=dependency)
return graph
def test_can_set_runtime_dependencies(package_a: Requirement,
package_b: Requirement,
dependency_graph: DependencyGraph):
dependency_graph.set_runtime_dependency(dependent=package_a,
dependency=package_b)
assert dependency_graph.is_runtime_dependency(dependent=package_a,
dependency=package_b)
def test_cyclic_runtime_dependencies_not_allowed(
package_a: Requirement, package_b: Requirement,
dependency_graph: DependencyGraph):
dependency_graph.set_runtime_dependency(dependent=package_a,
dependency=package_b)
with pytest.raises(CyclicDependencyOccured):
dependency_graph.set_runtime_dependency(dependent=package_b,
dependency=package_a)
def test_cannot_add_circular_runtime_dependency_to_buildtime_dependency(
package_a: Requirement,
package_b: Requirement,
dependency_graph: DependencyGraph,
):
dependency_graph.set_buildtime_dependency(dependent=package_a,
dependency=package_b)
with pytest.raises(CyclicDependencyOccured):
dependency_graph.set_runtime_dependency(dependent=package_b,
dependency=package_a)
def test_build_time_dependencies_dont_show_up_as_runtime_dependencies(
package_a: Requirement,
package_b: Requirement,
package_c: Requirement,
dependency_graph: DependencyGraph,
):
dependency_graph.set_runtime_dependency(dependent=package_a,
dependency=package_b)
dependency_graph.set_buildtime_dependency(dependent=package_b,
dependency=package_c)
assert not dependency_graph.is_runtime_dependency(dependent=package_a,
dependency=package_c)
def test_can_detect_indirect_runtime_dependencies(
package_a: Requirement,
package_b: Requirement,
package_c: Requirement,
dependency_graph: DependencyGraph,
) -> None:
dependency_graph.set_runtime_dependency(dependent=package_a,
dependency=package_b)
dependency_graph.set_runtime_dependency(dependent=package_b,
dependency=package_c)
assert dependency_graph.is_runtime_dependency(dependent=package_a,
dependency=package_c)
def test_can_retrieve_external_dependencies_from_runtime_dependencies(
package_a: Requirement,
package_b: Requirement,
external_dependency_a: ExternalDependency,
dependency_graph: DependencyGraph,
):
dependency_graph.set_runtime_dependency(dependent=package_a,
dependency=package_b)
dependency_graph.set_external_dependency(dependent=package_b,
dependency=external_dependency_a)
assert dependency_graph.get_all_external_dependencies(
package=package_a) == {
external_dependency_a,
}
def test_can_retriev_all_runtime_dependency_names(
package_a: Requirement,
package_b: Requirement,
package_c: Requirement,
dependency_graph: DependencyGraph,
) -> None:
dependency_graph.set_runtime_dependency(dependent=package_a,
dependency=package_b)
dependency_graph.set_runtime_dependency(dependent=package_b,
dependency=package_c)
assert dependency_graph.get_all_runtime_dependency_names(package_a) == {
package_a.name(),
package_b.name(),
package_c.name(),
}
def test_can_add_two_dependencies_graphs_and_runtime_dependencies_are_also_added(
package_a: Requirement,
package_b: Requirement,
package_c: Requirement,
dependency_graph: DependencyGraph,
):
other_dependency_graph = copy(dependency_graph)
dependency_graph.set_runtime_dependency(dependent=package_a,
dependency=package_b)
other_dependency_graph.set_runtime_dependency(dependent=package_b,
dependency=package_c)
sum_graph = dependency_graph + other_dependency_graph
assert not dependency_graph.is_runtime_dependency(dependent=package_a,
dependency=package_c)
assert not other_dependency_graph.is_runtime_dependency(
dependent=package_a, dependency=package_c)
assert sum_graph.is_runtime_dependency(dependent=package_a,
dependency=package_c)
def test_can_retrieve_external_dependencies_after_adding_graphs(
package_a: Requirement,
package_b: Requirement,
external_dependency_a: ExternalDependency,
external_dependency_b: ExternalDependency,
dependency_graph: DependencyGraph,
):
other_dependency_graph = copy(dependency_graph)
dependency_graph.set_external_dependency(dependent=package_a,
dependency=external_dependency_a)
dependency_graph.set_runtime_dependency(dependent=package_a,
dependency=package_b)
other_dependency_graph.set_external_dependency(
dependent=package_b, dependency=external_dependency_b)
sum_graph = dependency_graph + other_dependency_graph
assert sum_graph.get_all_external_dependencies(package=package_a) == {
external_dependency_a,
external_dependency_b,
}