use top_level.txt

blueyed · Oct 27, 2020 · f5ce0f5 · f5ce0f5
1 parent 9db2705
commit f5ce0f5
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/_pytest/config/__init__.py b/src/_pytest/config/__init__.py
@@ -887,8 +887,13 @@ def _mark_plugins_for_rewrite(self, hook) -> None:
 
         for dist in importlib_metadata.distributions():
             for ep in dist.entry_points:
-                if ep.group == "pytest11":
-                    hook.mark_rewrite(ep.value)
+                if ep.group == "pytest11":  # type: ignore[attr-defined]
+                    top_level_lines = dist.read_text("top_level.txt")
+                    if top_level_lines:
+                        for top_level in top_level_lines.splitlines():
+                            hook.mark_rewrite(ep.value)
+                    else:
+                        hook.mark_rewrite(ep.value)
 
     def _validate_args(self, args: List[str], via: str) -> List[str]:
         """Validate known args."""