tests: add download-only to video_test_framework

Add download-only to the three main testing apps

Author: dabrain34

tests/libs/video_test_config_base.py

@@ -460,6 +460,36 @@ def download_sample_assets(samples, asset_type: str = "test") -> bool:
         return False
 
 
+def load_and_download_samples(sample_class, json_file: str,
+                              test_type: str) -> bool:
+    """Load samples from JSON and download their assets.
+
+    Args:
+        sample_class: Sample class with a from_dict() classmethod
+        json_file: Path to JSON file containing sample definitions
+        test_type: Type of test ("decode" or "encode")
+
+    Returns:
+        True if download succeeded or no samples found.
+    """
+    samples_data = load_samples_from_json(json_file, test_type=test_type)
+    if not samples_data:
+        print(f"No {test_type} samples found")
+        return True
+
+    samples = []
+    for data in samples_data:
+        try:
+            samples.append(sample_class.from_dict(data))
+        except (KeyError, ValueError, TypeError) as e:
+            print(f"  Warning: failed to load {test_type} sample "
+                  f"{data.get('name', 'unknown')}: {e}")
+
+    if not samples:
+        return True
+    return download_sample_assets(samples, f"{test_type} test")
+
+
 def _validate_zip_member(member_name: str, extract_dir: Path) -> Path:
     """
     Validate a zip member path to prevent zip slip attacks.

tests/libs/video_test_utils.py

@@ -162,6 +162,11 @@ def add_common_arguments(parser, codec_choices=None):
         action="store_true",
         help="Show skipped tests in summary output",
     )
+    parser.add_argument(
+        "--download-only",
+        action="store_true",
+        help="Download test resources and exit without running tests",
+    )
     return parser
 
 

tests/video_test_framework_codec.py

@@ -39,6 +39,7 @@
     VideoTestStatus,
     load_samples_from_json,
     download_sample_assets,
+    load_and_download_samples,
     load_skip_list,
     is_test_skipped,
     SkipFilter,
@@ -49,8 +50,14 @@
 )
 
 from tests.libs.video_test_utils import safe_main_wrapper, DEFAULT_TEST_TIMEOUT
-from tests.video_test_framework_encode import VulkanVideoEncodeTestFramework
-from tests.video_test_framework_decode import VulkanVideoDecodeTestFramework
+from tests.video_test_framework_encode import (
+    VulkanVideoEncodeTestFramework,
+    EncodeTestSample,
+)
+from tests.video_test_framework_decode import (
+    VulkanVideoDecodeTestFramework,
+    DecodeTestSample,
+)
 
 
 @dataclass
@@ -598,6 +605,10 @@ def create_argument_parser() -> argparse.ArgumentParser:
     parser.add_argument(
         "--no-auto-download", action="store_true",
         help="Skip automatic download of missing/corrupt sample files")
+    parser.add_argument(
+        "--download-only", action="store_true",
+        help="Download all test resources (encode and decode) and exit "
+             "without running tests")
     parser.add_argument(
         "--deviceID",
         help="Vulkan device ID to use for testing "
@@ -655,6 +666,29 @@ def find_executables(args: argparse.Namespace) -> Tuple[str, str]:
     return encoder_path, decoder_path
 
 
+def download_all_resources(args: argparse.Namespace) -> bool:
+    """Download all test resources (encode and decode) without running tests"""
+    print("=== Downloading All Test Resources ===\n")
+
+    decode_json = args.decode_test_suite or "decode_samples.json"
+    encode_json = args.encode_test_suite or "encode_samples.json"
+
+    decode_ok = load_and_download_samples(
+        DecodeTestSample, decode_json, "decode"
+    )
+    encode_ok = load_and_download_samples(
+        EncodeTestSample, encode_json, "encode"
+    )
+
+    success = decode_ok and encode_ok
+    if success:
+        print("\n✓ All resources downloaded successfully")
+    else:
+        print("\n✗ Some resources failed to download")
+
+    return success
+
+
 def run_framework_tests(args: argparse.Namespace, encoder_path: str,
                         decoder_path: str) -> bool:
     """Run the actual test framework"""
@@ -725,6 +759,11 @@ def main() -> int:
         list_all_samples(skip_list_path)
         return 0
 
+    # Handle --download-only option
+    if args.download_only:
+        success = download_all_resources(args)
+        return 0 if success else 1
+
     # Find executable paths
     encoder_path, decoder_path = find_executables(args)
 

tests/video_test_framework_decode.py

@@ -37,6 +37,7 @@
     VideoTestStatus,
     check_sample_resources,
     create_error_result,
+    load_and_download_samples,
     load_samples_from_json,
 )
 from tests.libs.video_test_framework_base import (  # noqa: E402
@@ -341,6 +342,14 @@ def main() -> int:
         list_decoder_samples()
         return 0
 
+    # Handle --download-only option
+    if args.download_only:
+        json_file = args.decode_test_suite or "decode_samples.json"
+        success = load_and_download_samples(
+            DecodeTestSample, json_file, "decode"
+        )
+        return 0 if success else 1
+
     # Find and resolve decoder executable path
     args.decoder = PlatformUtils.resolve_executable_path(
         args.decoder, args.verbose

tests/video_test_framework_encode.py

@@ -36,9 +36,10 @@
     CodecType,
     TestResult,
     VideoTestStatus,
+    check_sample_resources,
     create_error_result,
+    load_and_download_samples,
     load_samples_from_json,
-    check_sample_resources,
 )
 from tests.libs.video_test_framework_base import (  # noqa: E402
     VulkanVideoTestFrameworkBase,
@@ -497,6 +498,14 @@ def main() -> int:
         list_encoder_samples(test_suite)
         return 0
 
+    # Handle --download-only option
+    if args.download_only:
+        json_file = args.encode_test_suite or "encode_samples.json"
+        success = load_and_download_samples(
+            EncodeTestSample, json_file, "encode"
+        )
+        return 0 if success else 1
+
     # Find and resolve encoder executable path
     args.encoder = PlatformUtils.resolve_executable_path(
         args.encoder, args.verbose