Add supports_test_execution_caching attribute to python test

Summary: Add an attribute for turning on test execution caching. It's turned off everywhere and will just let us experiment a little bit. The caching is for RE executions only (no DICE caching), so it is safe.

Reviewed By: IanChilds

Differential Revision: D95567733

fbshipit-source-id: 16849b3d44fa430d25e74fa0f23dbc9ea0083ebb

Author: Ioan Budea

decls/python_rules.bzl

@@ -140,6 +140,7 @@ def _python_test_attrs():
     test_attrs["_test_main"] = attrs.source(default = "prelude//python/tools:__test_main__.py")
     test_attrs["implicit_test_library"] = attrs.option(attrs.dep(providers = [PythonLibraryInfo]), default = None)
     test_attrs["safer_lazy_imports"] = attrs.bool(default = False)  # TODO(T240038931) When enabling lazy imports by default, remove this line
+    test_attrs["supports_test_execution_caching"] = attrs.bool(default = False)
     test_attrs.update(re_test_common.test_args())
     return test_attrs
 

python/compute_providers.bzl

@@ -56,5 +56,6 @@ def compute_test_providers(ctx: AnalysisContext, exe: PexProviders) -> list[Prov
             run_from_project_root = re_executor != None,
             use_project_relative_paths = re_executor != None,
             network_access = getattr(ctx.attrs, "network_access", None),
+            supports_test_execution_caching = ctx.attrs.supports_test_execution_caching,
         ),
     ) + [make_default_info(exe)]

python/python_needed_coverage_test.bzl

@@ -56,6 +56,7 @@ def python_needed_coverage_test_impl(ctx: AnalysisContext) -> list[Provider]:
             # the cell root (e.g. fbcode root).
             run_from_project_root = re_executor != None,
             use_project_relative_paths = re_executor != None,
+            supports_test_execution_caching = ctx.attrs.supports_test_execution_caching,
         ),
     ) + [
         DefaultInfo(),

rules_impl.bzl

@@ -476,6 +476,7 @@ _python_extra_attributes = {
         env = attrs.dict(key = attrs.string(), value = attrs.arg(), sorted = False, default = {}),
         labels = attrs.list(attrs.string(), default = []),
         needed_coverage = attrs.list(attrs.tuple(attrs.int(), attrs.dep(), attrs.option(attrs.string())), default = []),
+        supports_test_execution_caching = attrs.bool(default = False),
         test = attrs.dep(providers = [ExternalRunnerTestInfo]),
         **(re_test_common.test_args() | buck.inject_test_env_arg()),
     ),