new Buck target using ConstrainedGangSpec

Reviewed By: IanChilds

Differential Revision: D94789611

fbshipit-source-id: 3edccfa8f71d671b814ac42b6ee718bfde323625

Author: alessandrolipa-meta

decls/re_test_common.bzl

@@ -19,6 +19,8 @@ def _opts_for_tests_arg() -> Attr:
     #     "use_case": str | None
     #     "remote_cache_enabled": bool | None
     #     "dependencies": list<Dict<str, str>> | []
+    #     "gang_workers": list<Dict<str, str>> | []
+    #     "gang": Dict<str, str | int | Dict<str, str>> | None
     #     "resource_units": int | None
     #     "remote_execution_dynamic_image": dict<str, str | list<str>> | None
     # }
@@ -31,6 +33,8 @@ def _opts_for_tests_arg() -> Attr:
                     value = attrs.one_of(
                         attrs.string(),
                         attrs.list(attrs.string()),
+                        attrs.dict(key = attrs.string(), value = attrs.string(), sorted = False),
+                        attrs.int(),
                     ),
                     sorted = False,
                 ),

tests/re_utils.bzl

@@ -82,6 +82,7 @@ def get_re_executors_from_props(ctx: AnalysisContext) -> ([CommandExecutorConfig
     remote_cache_enabled = re_props_copy.pop("remote_cache_enabled", None)
     re_dependencies = re_props_copy.pop("dependencies", [])
     re_gang_workers = re_props_copy.pop("gang_workers", [])
+    re_gang = re_props_copy.pop("gang", None)
     local_enabled = re_props_copy.pop("local_enabled", False)
     local_listing_enabled = re_props_copy.pop("local_listing_enabled", None)
     re_resource_units = re_props_copy.pop("resource_units", None)
@@ -91,6 +92,12 @@ def get_re_executors_from_props(ctx: AnalysisContext) -> ([CommandExecutorConfig
         unexpected_props = ", ".join(re_props_copy.keys())
         fail("found unexpected re props: " + unexpected_props)
 
+    meta_internal_extra_params = None
+    if re_gang != None:
+        meta_internal_extra_params = {
+            "remote_execution_gang": re_gang,
+        }
+
     default_executor = CommandExecutorConfig(
         local_enabled = local_enabled,
         remote_enabled = True,
@@ -101,6 +108,7 @@ def get_re_executors_from_props(ctx: AnalysisContext) -> ([CommandExecutorConfig
         remote_execution_gang_workers = re_gang_workers,
         remote_execution_resource_units = re_resource_units,
         remote_execution_dynamic_image = re_dynamic_image,
+        meta_internal_extra_params = meta_internal_extra_params,
     )
 
     listing_executor = default_executor