def _(): @fixture def parent_a(): pass @fixture def parent_b(): pass @fixture def child(a=parent_a, b=parent_b): pass assert Fixture(child).parents() == [Fixture(parent_a), Fixture(parent_b)]
def _(): @fixture def a(): pass @fixture def b(): pass @fixture def c(a=a, b=b): pass @fixture def d(a=a, c=c): pass fa, fb, fc, fd = fixtures = [Fixture(f) for f in (a, b, c, d)] to_parents, to_children = fixture_parents_and_children(fixtures) assert to_parents == {fa: [], fb: [], fc: [fa, fb], fd: [fa, fc]} assert to_children == { fa: [fc, fd], fb: [fc], fc: [fd], fd: [], }
def _(): def foo(fixture=dummy_fixture, not_fixture=5): pass t = Test(foo, module_name="") assert t.resolver.fixtures == {"fixture": Fixture(dummy_fixture)}
def output_fixture_dependency_tree(fixture: Fixture, show_scopes: bool, max_depth: Optional[int], depth: int = 0): if max_depth is not None and depth >= max_depth: return deps = [Fixture(par.default) for par in fixture.deps().values()] if not deps: return if depth == 0: yield indent(f"depends on", INDENT) for idx, dep in enumerate(deps): prefix = "├─" if idx != len(deps) - 1 else "└─" yield indent( f"{prefix} {format_fixture_header(dep, show_scopes)}", make_indent(depth + 1), ) child_lines = list( output_fixture_dependency_tree( dep, show_scopes, max_depth, depth=depth + 1, )) for child_idx, line in enumerate(child_lines): if idx < len(deps) - 1: line = line.replace(" └─", "│ └─") yield " " + line
def _(): @fixture def f(): pass t = Test(lambda f=f: None, module_name="") assert fixtures_used_directly_by_tests([t]) == {Fixture(f): [t]}
def _(): def foo( a=each(dummy_fixture, "foo", dummy_fixture, "fourth"), b=each("bar", dummy_fixture, dummy_fixture, "fourth"), ): pass t = Test(foo, module_name="") first, second, third, fourth = t.get_parameterised_instances() assert first.resolver.fixtures == {"a": Fixture(dummy_fixture)} assert second.resolver.fixtures == {"b": Fixture(dummy_fixture)} assert third.resolver.fixtures == { "a": Fixture(dummy_fixture), "b": Fixture(dummy_fixture), } assert fourth.resolver.fixtures == {}
def _resolve_single_fixture( self, fixture_fn: Callable, cache: FixtureCache ) -> Fixture: fixture = Fixture(fixture_fn) if fixture.key in cache: cached_fixture = cache[fixture.key] if fixture.scope == Scope.Global: return cached_fixture elif fixture.scope == Scope.Module: if cached_fixture.last_resolved_module_name == self.module_name: return cached_fixture elif fixture.scope == Scope.Test: if cached_fixture.last_resolved_test_id == self.id: return cached_fixture # Cache miss, so update the fixture metadata before we resolve and cache it fixture.last_resolved_test_id = self.id fixture.last_resolved_module_name = self.module_name has_deps = len(fixture.deps()) > 0 is_generator = fixture.is_generator_fixture if not has_deps: try: if is_generator: fixture.gen = fixture_fn() fixture.resolved_val = next(fixture.gen) else: fixture.resolved_val = fixture_fn() except Exception as e: raise FixtureError(f"Unable to resolve fixture '{fixture.name}'") from e cache.cache_fixture(fixture) return fixture signature = inspect.signature(fixture_fn) children_defaults = signature.bind_partial() children_defaults.apply_defaults() children_resolved = {} for name, child_fixture in children_defaults.arguments.items(): child_resolved = self._resolve_single_fixture(child_fixture, cache) children_resolved[name] = child_resolved try: if is_generator: fixture.gen = fixture_fn( **self._resolve_fixture_values(children_resolved) ) fixture.resolved_val = next(fixture.gen) else: fixture.resolved_val = fixture_fn( **self._resolve_fixture_values(children_resolved) ) except Exception as e: raise FixtureError(f"Unable to resolve fixture '{fixture.name}'") from e cache.cache_fixture(fixture) return fixture
def _(): @fixture def parent(): pass @fixture def child(): pass @fixture def not_used(): pass t1 = Test(lambda c=child: None, module_name="") t2 = Test(lambda p=parent, c=child: None, module_name="") t3 = Test(lambda _: None, module_name="") assert fixtures_used_directly_by_tests([t1, t2, t3]) == { Fixture(child): [t1, t2], Fixture(parent): [t2], }
def _(): @fixture def parent(): pass @fixture def child(): pass t = Test(lambda c=child: None, module_name="") assert fixtures_used_directly_by_tests([t]) == {Fixture(child): [t]}
def _resolve_single_arg(self, arg: Callable, cache: FixtureCache) -> Union[Any, Fixture]: """ Get the fixture return value If the fixture has been cached, return the value from the cache. Otherwise, call the fixture function and return the value. """ if not hasattr(arg, "ward_meta"): return arg fixture = Fixture(arg) if cache.contains(fixture, fixture.scope, self.scope_key_from(fixture.scope)): return cache.get(fixture.key, fixture.scope, self.scope_key_from(fixture.scope)) has_deps = len(fixture.deps()) > 0 is_generator = fixture.is_generator_fixture if not has_deps: try: if is_generator: fixture.gen = arg() fixture.resolved_val = next(fixture.gen) else: fixture.resolved_val = arg() except Exception as e: raise FixtureError( f"Unable to resolve fixture '{fixture.name}'") from e scope_key = self.scope_key_from(fixture.scope) cache.cache_fixture(fixture, scope_key) return fixture children_defaults = self._get_default_args(func=arg) children_resolved = {} for name, child_fixture in children_defaults.items(): child_resolved = self._resolve_single_arg(child_fixture, cache) children_resolved[name] = child_resolved try: args_to_inject = self._unpack_resolved(children_resolved) if is_generator: fixture.gen = arg(**args_to_inject) fixture.resolved_val = next(fixture.gen) else: fixture.resolved_val = arg(**args_to_inject) except Exception as e: raise FixtureError( f"Unable to resolve fixture '{fixture.name}'") from e scope_key = self.scope_key_from(fixture.scope) cache.cache_fixture(fixture, scope_key) return fixture
def output_fixtures( show_scopes: bool, show_docstrings: bool, show_direct_dependencies: bool, show_dependency_trees: bool, ): fixtures = [Fixture(f) for f in _FIXTURES] for fixture in fixtures: output_fixture_information( fixture, show_scopes=show_scopes, show_docstrings=show_docstrings, show_direct_dependencies=show_direct_dependencies, show_dependency_trees=show_dependency_trees, )
def _resolve_single_arg(self, arg: Callable, cache: FixtureCache) -> Union[Any, Fixture]: """ Get the fixture return value If the fixture has been cached, return the value from the cache. Otherwise, call the fixture function and return the value. """ if not hasattr(arg, "ward_meta"): return arg fixture = Fixture(arg) if cache.contains(fixture, fixture.scope, self.test.scope_key_from(fixture.scope)): return cache.get(fixture.key, fixture.scope, self.test.scope_key_from(fixture.scope)) children_defaults = self.get_default_args(func=arg) children_resolved = {} for name, child_fixture in children_defaults.items(): child_resolved = self._resolve_single_arg(child_fixture, cache) children_resolved[name] = child_resolved try: args_to_inject = self._unpack_resolved(children_resolved) if fixture.is_generator_fixture: fixture.gen = arg(**args_to_inject) fixture.resolved_val = next( fixture.gen) # type: ignore[arg-type] elif fixture.is_async_generator_fixture: fixture.gen = arg(**args_to_inject) awaitable = fixture.gen.__anext__() # type: ignore[union-attr] fixture.resolved_val = asyncio.get_event_loop( ).run_until_complete(awaitable) elif fixture.is_coroutine_fixture: fixture.resolved_val = asyncio.get_event_loop( ).run_until_complete(arg(**args_to_inject)) else: fixture.resolved_val = arg(**args_to_inject) except (Exception, SystemExit) as e: raise FixtureError( f"Unable to resolve fixture '{fixture.name}'") from e scope_key = self.test.scope_key_from(fixture.scope) cache.cache_fixture(fixture, scope_key) return fixture
def _(path=each(THIS_FILE.parent / "the-fixture-is-not-in-this-file.py")): fixtures = [Fixture(f) for f in [named_fixture]] assert list(filter_fixtures(fixtures, paths=[path])) == []
def _(path=each(THIS_FILE, THIS_FILE.parent, THIS_FILE.parent.parent)): fixtures = [Fixture(f) for f in [named_fixture]] assert list(filter_fixtures(fixtures, paths=[path])) == fixtures
def _(query=each("echo", "foobar", "wizbang")): fixtures = [Fixture(f) for f in [named_fixture, marker_fixture]] assert list(filter_fixtures(fixtures, query=query)) == []
def _(query=each("marker", "mark", "ret", "return", '"')): fixtures = [Fixture(f) for f in [named_fixture, marker_fixture]] assert list(filter_fixtures(fixtures, query=query)) == [Fixture(marker_fixture)]
def _(query=each("test", "test_collect", "collect", "t_coll")): fixtures = [Fixture(f) for f in [named_fixture]] assert list(filter_fixtures(fixtures, query=query)) == fixtures
def _(query=each("named_fixture", "named", "fixture", "med_fix")): fixtures = [Fixture(f) for f in [named_fixture]] assert list(filter_fixtures(fixtures, query=query)) == fixtures
def _(): fixtures = [Fixture(f) for f in [named_fixture, marker_fixture]] assert list(filter_fixtures(fixtures)) == fixtures
def exception_raising_fixture(): def i_raise_an_exception(): raise ZeroDivisionError() return Fixture(fn=i_raise_an_exception)
def fixtures(self) -> Dict[str, Fixture]: return { name: Fixture(arg) for name, arg in self._get_args_for_iteration().items() if is_fixture(arg) }
def fixtures(a=fixture_a, b=fixture_b): return {"fixture_a": Fixture(fn=a), "fixture_b": Fixture(fn=b)}
def _(): @fixture def fix(): pass assert len(Fixture(fix).parents()) == 0