| # Copyright (c) 2014 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import errno |
| import hashlib |
| import logging |
| import os |
| |
| from autotest_lib.client.bin import utils |
| from autotest_lib.client.common_lib import error |
| from autotest_lib.client.common_lib import file_utils |
| from autotest_lib.client.cros import chrome_binary_test |
| |
| DOWNLOAD_BASE = 'http://commondatastorage.googleapis.com/chromiumos-test-assets-public/' |
| BINARY = 'video_encode_accelerator_unittest' |
| |
| def _remove_if_exists(filepath): |
| try: |
| os.remove(filepath) |
| except OSError, e: |
| if e.errno != errno.ENOENT: # no such file |
| raise |
| |
| |
| def _download_video(download_path, local_file): |
| url = '%s%s' % (DOWNLOAD_BASE, download_path) |
| logging.info('download "%s" to "%s"', url, local_file) |
| |
| file_utils.download_file(url, local_file) |
| |
| with open(local_file, 'r') as r: |
| md5sum = hashlib.md5(r.read()).hexdigest() |
| if md5sum not in download_path: |
| raise error.TestError('unmatched md5 sum: %s' % md5sum) |
| |
| |
| class video_VideoEncodeAccelerator(chrome_binary_test.ChromeBinaryTest): |
| """ |
| This test is a wrapper of the chrome unittest binary: |
| video_encode_accelerator_unittest. |
| """ |
| |
| version = 1 |
| |
| def get_filter_option(self): |
| """Get option of filtering test |
| """ |
| |
| blacklist = { |
| # board: [tests to skip...] |
| |
| # Kevin doesn't support HW encode for plane sizes not multiple |
| # of cache line |
| 'kevin': ['CacheLineUnalignedInputTest/*'] |
| } |
| |
| board = utils.get_current_board() |
| if board in blacklist: |
| return '-' + ':'.join(blacklist[board]) |
| |
| return '' |
| |
| @chrome_binary_test.nuke_chrome |
| def run_once(self, in_cloud, streams, profile, gtest_filter=None): |
| """Runs video_encode_accelerator_unittest on the streams. |
| |
| @param in_cloud: Input file needs to be downloaded first. |
| @param streams: The test streams for video_encode_accelerator_unittest. |
| @param profile: The profile to encode into. |
| @param gtest_filter: test case filter. |
| |
| @raises error.TestFail for video_encode_accelerator_unittest failures. |
| """ |
| |
| last_test_failure = None |
| for path, width, height, bit_rate in streams: |
| if in_cloud: |
| input_path = os.path.join(self.tmpdir, path.split('/')[-1]) |
| _download_video(path, input_path) |
| else: |
| input_path = os.path.join(self.cr_source_dir, path) |
| |
| output_path = os.path.join(self.tmpdir, |
| '%s.out' % input_path.split('/')[-1]) |
| |
| cmd_line_list = [] |
| cmd_line_list.append('--test_stream_data="%s:%s:%s:%s:%s:%s"' % ( |
| input_path, width, height, profile, output_path, bit_rate)) |
| if utils.is_freon(): |
| cmd_line_list.append('--ozone-platform=gbm') |
| |
| # Command line |gtest_filter| can override get_filter_option(). |
| predefined_filter = self.get_filter_option() |
| if gtest_filter and predefined_filter: |
| logging.warning('predefined gtest filter is suppressed: %s', |
| predefined_filter) |
| if not gtest_filter: |
| gtest_filter = predefined_filter |
| if gtest_filter: |
| cmd_line_list.append('--gtest_filter="%s"' % gtest_filter) |
| |
| cmd_line = ' '.join(cmd_line_list) |
| try: |
| self.run_chrome_test_binary(BINARY, cmd_line, as_chronos=False) |
| except error.TestFail as test_failure: |
| # Continue to run the remaining test streams and raise |
| # the last failure after finishing all streams. |
| logging.exception('error while encoding %s', input_path) |
| last_test_failure = test_failure |
| finally: |
| # Remove the downloaded video |
| if in_cloud: |
| _remove_if_exists(input_path) |
| _remove_if_exists(output_path) |
| |
| if last_test_failure: |
| raise last_test_failure |