cast to real WorkingSet in test metadata pkg resources

Author: notatallshaw

tests/unit/metadata/test_metadata_pkg_resources.py

@@ -18,7 +18,7 @@
 )
 
 pkg_resources = pytest.importorskip("pip._vendor.pkg_resources")
-
+from pip._vendor.pkg_resources import WorkingSet
 
 def _dist_is_local(dist: mock.Mock) -> bool:
     return dist.kind != "global" and dist.kind != "user"
@@ -71,13 +71,13 @@ def require(self, name: str) -> None:
 )
 def test_get_distribution(ws: _MockWorkingSet, req_name: str) -> None:
     """Ensure get_distribution() finds all kinds of distributions."""
-    dist = Environment(ws).get_distribution(req_name)
+    dist = Environment(cast(WorkingSet, ws)).get_distribution(req_name)
     assert dist is not None
     assert cast(Distribution, dist)._dist.project_name == req_name
 
 
 def test_get_distribution_nonexist() -> None:
-    dist = Environment(workingset).get_distribution("non-exist")
+    dist = Environment(cast(WorkingSet, workingset)).get_distribution("non-exist")
     assert dist is None