From d33f1f8f8d17b77ef2efa499d4d5a0fae527f48f Mon Sep 17 00:00:00 2001 From: Kohsuke Kawaguchi Date: Fri, 5 Sep 2025 10:27:43 -0700 Subject: [PATCH 1/2] [AIENG-230] defined a switch to control the test selection behavior --- launchable/commands/subset.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/launchable/commands/subset.py b/launchable/commands/subset.py index e0ec48e4a..595f9232d 100644 --- a/launchable/commands/subset.py +++ b/launchable/commands/subset.py @@ -207,6 +207,12 @@ help="get subset list from git managed files", is_flag=True, ) +@click.option( + "--use-case", + "use_case", + type=str, + hidden=True, # control PTS v2 test selection behavior. Non-committed, so hidden for now. +) @click.pass_context def subset( context: click.core.Context, @@ -235,6 +241,7 @@ def subset( prioritized_tests_mapping_file: Optional[TextIO] = None, test_suite: Optional[str] = None, is_get_tests_from_guess: bool = False, + use_case: Optional[str] = None, ): app = context.obj tracking_client = TrackingClient(Command.SUBSET, app=app) @@ -513,6 +520,9 @@ def get_payload( if prioritized_tests_mapping_file: payload['prioritizedTestsMapping'] = json.load(prioritized_tests_mapping_file) + if use_case: + payload["changesUnderTest"] = use_case + return payload def _collect_potential_test_files(self): From 16afcef07ac921abb8d2edc18bfcfa48d5c401c4 Mon Sep 17 00:00:00 2001 From: Kohsuke Kawaguchi Date: Mon, 15 Sep 2025 10:24:28 +0900 Subject: [PATCH 2/2] [AIENG-230] better error diagnostics --- launchable/commands/subset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/launchable/commands/subset.py b/launchable/commands/subset.py index 595f9232d..3e05e0e71 100644 --- a/launchable/commands/subset.py +++ b/launchable/commands/subset.py @@ -210,7 +210,7 @@ @click.option( "--use-case", "use_case", - type=str, + type=click.Choice(["one-commit", "feature-branch", "recurring"]), hidden=True, # control PTS v2 test selection behavior. Non-committed, so hidden for now. ) @click.pass_context