Adding upstream version 126.0.upstream/126.0

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 15 insertions, 1 deletions

diff --git a/testing/talos/mach_commands.py b/testing/talos/mach_commands.py
index c685219422..0b15af7a2c 100644
--- a/testing/talos/mach_commands.py
+++ b/testing/talos/mach_commands.py
@@ -12,6 +12,7 @@ import sys
 
 import six
 from mach.decorators import Command
+from mach.util import get_state_dir
 from mozbuild.base import BinaryNotFoundException, MozbuildObject
 
 HERE = os.path.dirname(os.path.realpath(__file__))
@@ -82,6 +83,7 @@ class TalosRunner(MozbuildObject):
                 "win32": "python3.manifest",
                 "win64": "python3_x64.manifest",
             },
+            "mozbuild_path": get_state_dir(),
         }
 
     def make_args(self):
@@ -117,6 +119,16 @@ def create_parser():
     return create_parser(mach_interface=True)
 
 
+def setup_toolchain_artifacts(args, command_context):
+    if not any(arg.lower() == "pdfpaint" for arg in args):
+        return
+
+    from mozbuild.bootstrap import bootstrap_toolchain
+
+    print("Setting up pdfpaint PDFs...")
+    bootstrap_toolchain("talos-pdfs")
+
+
 @Command(
     "talos-test",
     category="testing",
@@ -127,7 +139,9 @@ def run_talos_test(command_context, **kwargs):
     talos = command_context._spawn(TalosRunner)
 
     try:
-        return talos.run_test(sys.argv[2:])
+        args = sys.argv[2:]
+        setup_toolchain_artifacts(args, command_context)
+        return talos.run_test(args)
     except Exception as e:
         print(str(e))
         return 1