| #!/usr/bin/env python |
| |
| """ |
| This file generates all video_VDAStress control files from a master list. |
| """ |
| |
| import collections |
| |
| Testdata = collections.namedtuple('Testdata', 'gs, decoder, access') |
| |
| TESTS = [ |
| Testdata('gs://chromeos-test-assets-private/VDA/', 'h264', 'private'), |
| Testdata('gs://chromeos-test-assets-private/VDA/', 'vp8', 'private'), |
| # TODO(ihf): Populate public bucket with test videos. |
| #Testdata('gs://chromiumos-test-assets-public/VDA/', 'h264', 'public'), |
| #Testdata('gs://chromiumos-test-assets-public/VDA/', 'vp8', 'public'), |
| #Testdata('gs://chromiumos-test-assets-public/VDA/', 'vp9', 'public'), |
| ] |
| |
| CONTROLFILE_TEMPLATE = ( |
| """# Copyright 2013 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| AUTHOR = 'Chrome OS Team, [email protected]' |
| NAME = 'video_VDAStress.{1}.{2}.{3}' |
| ATTRIBUTES = 'suite:video' |
| TIME = 'LENGTHY' |
| TEST_CATEGORY = 'Stress' |
| TEST_CLASS = 'video' |
| TEST_TYPE = 'server' |
| DEPENDENCIES = 'hw_video_acc_{1}' |
| |
| DOC = \"\"\" |
| VDA stress test to download and run with {1} test videos from cloud storage. |
| \"\"\" |
| |
| import shutil |
| import tempfile |
| |
| # Download the test videos from the gs bucket to the server. |
| server_videos_dir = tempfile.mkdtemp(dir=job.tmpdir) |
| videos = [] |
| job.run_test( |
| 'video_VDAStressSetup', |
| gs_bucket='{0}{1}/', |
| server_videos_dir=server_videos_dir, |
| videos=videos, |
| shard_number={3}, |
| shard_count={4}) |
| |
| |
| def run(machine): |
| job.run_test('video_VDAStress', |
| machine=machine, |
| server_videos_dir=server_videos_dir, |
| videos=videos) |
| |
| |
| job.parallel_on_machines(run, machines) |
| shutil.rmtree(server_videos_dir)""") |
| |
| |
| shard_count = 10 |
| for test in TESTS: |
| for shard_number in xrange(0, shard_count): |
| filename = 'control.%s.%s.%d' % (test.decoder, test.access, shard_number) |
| with open(filename, 'w+') as f: |
| content = ( |
| CONTROLFILE_TEMPLATE.format( |
| test.gs, test.decoder, test.access, |
| shard_number, shard_count)) |
| f.write(content) |
