Espresso test for loopback video quality testing

Update the AppRTCMobileTestStubbedVideoIO test to run on
phones without Internet connection. This is done by bringing up
a local instance of AppRTC on the Linux machine connected to
the Android device.

Running this test will need the webrtc.DEPS solution to be configured
for the checkout, since that will pull down the precompiled AppRTC
package that is needed.

Continued from http://crrev.com/2780493002#ps20001 (by kjellander@)
Continued from http://crrev.com/2741743002#ps180001 (by mandermo@)

BUG=webrtc:7185

Review-Url: https://codereview.webrtc.org/2825313002
Cr-Commit-Position: refs/heads/master@{#17838}
This commit is contained in:
oprypin
2017-04-24 04:15:13 -07:00
committed by Commit bot
parent 828a6e1f01
commit 30cda5ef98
7 changed files with 120 additions and 36 deletions

View File

@ -19,17 +19,34 @@ It assumes you have a Android device plugged in.
"""
import argparse
import atexit
import logging
import os
import shutil
import subprocess
import sys
import tempfile
import time
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
SRC_DIR = os.path.normpath(os.path.join(SCRIPT_DIR, os.pardir, os.pardir,
os.pardir))
WEBRTC_DEPS_INSTRUCTIONS = """Please add a solution to your .gclient file like
this and run gclient sync:
{
"name": "webrtc.DEPS",
"url": "https://chromium.googlesource.com/chromium/deps/webrtc/webrtc.DEPS",
},
"""
class Error(Exception):
pass
class VideoQualityTestError(Error):
pass
def _RunCommand(argv, cwd=SRC_DIR, **kwargs):
@ -37,6 +54,22 @@ def _RunCommand(argv, cwd=SRC_DIR, **kwargs):
subprocess.check_call(argv, cwd=cwd, **kwargs)
def _RunCommandWithOutput(argv, cwd=SRC_DIR, **kwargs):
logging.info('Running %r', argv)
return subprocess.check_output(argv, cwd=cwd, **kwargs)
def _RunBackgroundCommand(argv, cwd=SRC_DIR):
logging.info('Running %r', argv)
process = subprocess.Popen(argv, cwd=cwd)
atexit.register(process.terminate)
time.sleep(0.5)
status = process.poll()
if status: # is not None or 0
raise subprocess.CalledProcessError(status, argv)
return process
def _ParseArgs():
parser = argparse.ArgumentParser(description='Start loopback video analysis.')
parser.add_argument('build_dir_android',
@ -45,6 +78,7 @@ def _ParseArgs():
help='The path to the build directory for building locally.')
parser.add_argument('--temp_dir',
help='A temporary directory to put the output.')
parser.add_argument('--adb-path', help='Path to adb binary.', default='adb')
args = parser.parse_args()
return args
@ -58,6 +92,7 @@ def main():
build_dir_android = args.build_dir_android
build_dir_x86 = args.build_dir_x86
temp_dir = args.temp_dir
adb_path = args.adb_path
if not temp_dir:
temp_dir = tempfile.mkdtemp()
else:
@ -76,14 +111,49 @@ def main():
download_script = os.path.join(tools_dir, 'download_tools.py')
_RunCommand([sys.executable, download_script, toolchain_dir])
# Select an Android device in case multiple are connected
for line in _RunCommandWithOutput([adb_path, 'devices']).splitlines():
if line.endswith('\tdevice'):
android_device = line.split('\t')[0]
break
else:
raise VideoQualityTestError('Cannot find any connected Android device.')
# Start AppRTC Server
dev_appserver = os.path.join(SRC_DIR, 'out', 'apprtc', 'google_appengine',
'dev_appserver.py')
if not os.path.isfile(dev_appserver):
raise VideoQualityTestError('Cannot find %s.\n%s' %
(dev_appserver, WEBRTC_DEPS_INSTRUCTIONS))
appengine_dir = os.path.join(SRC_DIR, 'out', 'apprtc', 'out', 'app_engine')
_RunBackgroundCommand(['python', dev_appserver, appengine_dir,
'--port=9999', '--admin_port=9998',
'--skip_sdk_update_check', '--clear_datastore=yes'])
# Start Collider
collider_path = os.path.join(SRC_DIR, 'out', 'go-workspace', 'bin',
'collidermain')
if not os.path.isfile(collider_path):
raise VideoQualityTestError('Cannot find %s.\n%s' %
(collider_path, WEBRTC_DEPS_INSTRUCTIONS))
_RunBackgroundCommand([collider_path, '-tls=false',
'-port=8089', '-room-server=http://localhost:9999'])
# Start adb reverse forwarder
reverseforwarder_path = os.path.join(
SRC_DIR, 'build', 'android', 'adb_reverse_forwarder.py')
_RunBackgroundCommand([reverseforwarder_path, '--device', android_device,
'9999', '9999', '8089', '8089'])
# Run the Espresso code.
test_script = os.path.join(build_dir_android,
'bin', 'run_AppRTCMobileTestStubbedVideoIO')
_RunCommand([sys.executable, test_script])
_RunCommand([test_script, '--device', android_device])
# Pull the output video.
test_video = os.path.join(temp_dir, 'test_video.y4m')
_RunCommand(['adb', 'pull', '/sdcard/output.y4m', test_video])
_RunCommand([adb_path, '-s', android_device,
'pull', '/sdcard/output.y4m', test_video])
test_video_yuv = os.path.join(temp_dir, 'test_video.yuv')
@ -103,8 +173,7 @@ def main():
ConvertVideo(reference_video, reference_video_yuv)
# Run compare script.
compare_script = os.path.join(SRC_DIR, 'webrtc', 'tools',
'compare_videos.py')
compare_script = os.path.join(SRC_DIR, 'webrtc', 'tools', 'compare_videos.py')
zxing_path = os.path.join(toolchain_dir, 'linux', 'zxing')
# The frame_analyzer binary should be built for local computer and not for