getdeps: configure testpilot to use collections and tag tests

Summary:
This should enable test pilot to skip broken/flakey tests.

The `--tag-new-tests` flag is only appropriate for code that has been
landed on master and is used by the FB infra to classify new tests
appropriately.

For continuous builds we use a test collection with different parameters
from the normal developer facing flow so that the infra can re-assess
their status and enable/disable high-signal/noisy tests.

Depends: D15495344

Reviewed By: Ben0mega

Differential Revision: D15500955

fbshipit-source-id: f3b7976cec6a5cf70f5d128b38bde11620b26918

build/fbcode_builder/getdeps/builder.py

@@ -329,21 +329,65 @@ class CMakeBuilder(BuilderBase):
 
             env.set("http_proxy", "")
             env.set("https_proxy", "")
-            self._run_cmd(
+
-                [
+            runs = []
-                    testpilot,
+
-                    # Need to force the repo type otherwise testpilot on windows
+            testpilot_args = [
-                    # can be confused (presumably sparse profile related)
+                testpilot,
-                    "--force-repo",
+                # Need to force the repo type otherwise testpilot on windows
-                    "fbcode",
+                # can be confused (presumably sparse profile related)
-                    "--force-repo-root",
+                "--force-repo",
-                    self.build_opts.fbsource_dir,
+                "fbcode",
-                    "--buck-test-info",
+                "--force-repo-root",
-                    buck_test_info_name,
+                self.build_opts.fbsource_dir,
-                ],
+                "--buck-test-info",
-                cwd=self.build_opts.fbcode_builder_dir,
+                buck_test_info_name,
-                env=env,
+            ]
-            )
+
+            if schedule_type == "continuous":
+                runs.append(
+                    [
+                        "--tag-new-tests",
+                        "--collection",
+                        "oss-continuous",
+                        "--purpose",
+                        "continuous",
+                    ]
+                )
+            elif schedule_type == "testwarden":
+                # One run to assess new tests
+                runs.append(
+                    [
+                        "--tag-new-tests",
+                        "--collection",
+                        "oss-new-test-stress",
+                        "--stress-runs",
+                        "10",
+                        "--purpose",
+                        "stress-run-new-test",
+                    ]
+                )
+                # And another for existing tests
+                runs.append(
+                    [
+                        "--tag-new-tests",
+                        "--collection",
+                        "oss-existing-test-stress",
+                        "--stress-runs",
+                        "10",
+                        "--purpose",
+                        "stress-run",
+                    ]
+                )
+            else:
+                runs.append(["--collection", "oss-diff", "--purpose", "diff"])
+
+            for run in runs:
+                self._run_cmd(
+                    testpilot_args + run,
+                    cwd=self.build_opts.fbcode_builder_dir,
+                    env=env,
+                )
         else:
             self._run_cmd(
                 [ctest, "--output-on-failure", "-j", str(self.build_opts.num_jobs)],