Only download necessary wprgo files in perf scripts

Bug: chromium:869762
Change-Id: I85a0b5f35465c5e2974248426dfb874e13259c8e
Reviewed-on: https://chromium-review.googlesource.com/1163045
Commit-Queue: Vovo Yang <vovoy@chromium.org>
Reviewed-by: Juan Antonio Navarro Pérez <perezju@chromium.org>
Reviewed-by: Ned Nguyen <nednguyen@google.com>
Cr-Commit-Position: refs/heads/master@{#580913}

tools/perf/benchmarks/system_health_smoke_test.py

@@ -178,7 +178,9 @@ def load_tests(loader, standard_tests, pattern):
     # Prefetch WPR archive needed by the stories set to avoid race condition
     # when feching them when tests are run in parallel.
     # See crbug.com/700426 for more details.
-    stories_set.wpr_archive_info.DownloadArchivesIfNeeded()
+    story_names = [s.name for s in stories_set if not s.is_local]
+    stories_set.wpr_archive_info.DownloadArchivesIfNeeded(
+        story_names=story_names)
 
     for story_to_smoke_test in stories_set.stories:
       suite.addTest(

tools/perf/fetch_benchmark_deps.py

@@ -35,8 +35,8 @@ def _FetchDependenciesIfNeeded(story_set):
     return
 
   # Download WPR files.
-  if any(not story.is_local for story in story_set):
-    story_set.wpr_archive_info.DownloadArchivesIfNeeded()
+  story_names = [s.name for s in story_set if not s.is_local]
+  story_set.wpr_archive_info.DownloadArchivesIfNeeded(story_names=story_names)
 
 
 def _EnumerateDependencies(story_set):