mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-08 04:27:37 +00:00
74e936113a
--- testing/tps/tps/firefoxrunner.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) --HG-- extra : rebase_source : 7785bb543c0d44ce6dd3380c8ce1db3bff88cdad
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
|
|
import copy
|
|
import httplib2
|
|
import os
|
|
|
|
import mozfile
|
|
import mozinstall
|
|
from mozprofile import Profile
|
|
from mozrunner import FirefoxRunner
|
|
|
|
|
|
class TPSFirefoxRunner(object):
|
|
|
|
PROCESS_TIMEOUT = 240
|
|
|
|
def __init__(self, binary):
|
|
if binary is not None and ('http://' in binary or 'ftp://' in binary):
|
|
self.url = binary
|
|
self.binary = None
|
|
else:
|
|
self.url = None
|
|
self.binary = binary
|
|
|
|
self.installdir = None
|
|
|
|
def __del__(self):
|
|
if self.installdir:
|
|
mozfile.remove(self.installdir, True)
|
|
|
|
def download_url(self, url, dest=None):
|
|
h = httplib2.Http()
|
|
resp, content = h.request(url, 'GET')
|
|
if dest == None:
|
|
dest = os.path.basename(url)
|
|
|
|
local = open(dest, 'wb')
|
|
local.write(content)
|
|
local.close()
|
|
return dest
|
|
|
|
def download_build(self, installdir='downloadedbuild', appname='firefox'):
|
|
self.installdir = os.path.abspath(installdir)
|
|
buildName = os.path.basename(self.url)
|
|
pathToBuild = os.path.join(os.path.dirname(os.path.abspath(__file__)),
|
|
buildName)
|
|
|
|
# delete the build if it already exists
|
|
if os.access(pathToBuild, os.F_OK):
|
|
os.remove(pathToBuild)
|
|
|
|
# download the build
|
|
print 'downloading build'
|
|
self.download_url(self.url, pathToBuild)
|
|
|
|
# install the build
|
|
print 'installing %s' % pathToBuild
|
|
mozfile.remove(self.installdir, True)
|
|
binary = mozinstall.install(src=pathToBuild, dest=self.installdir)
|
|
|
|
# remove the downloaded archive
|
|
os.remove(pathToBuild)
|
|
|
|
return binary
|
|
|
|
def run(self, profile=None, timeout=PROCESS_TIMEOUT, env=None, args=None):
|
|
"""Runs the given FirefoxRunner with the given Profile, waits
|
|
for completion, then returns the process exit code
|
|
"""
|
|
if profile is None:
|
|
profile = Profile()
|
|
self.profile = profile
|
|
|
|
if self.binary is None and self.url:
|
|
self.binary = self.download_build()
|
|
|
|
runner = FirefoxRunner(profile=self.profile, binary=self.binary,
|
|
env=env, cmdargs=args)
|
|
|
|
runner.start(timeout=timeout)
|
|
return runner.wait()
|