1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
# 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 os
import mozinfo
import tempfile
import subprocess
from mozlog.formatters.process import strstatus
def printstatus(name, returncode):
"""
print the status of a command exit code, formatted for tbpl.
Note that mozlog structured action "process_exit" should be used
instead of that in new code.
"""
print "TEST-INFO | %s: %s" % (name, strstatus(returncode))
def dump_screen(utilityPath, log):
"""dumps a screenshot of the entire screen to a directory specified by
the MOZ_UPLOAD_DIR environment variable"""
is_structured_log = hasattr(log, 'process_exit')
# Need to figure out which OS-dependent tool to use
if mozinfo.isUnix:
utility = [os.path.join(utilityPath, "screentopng")]
utilityname = "screentopng"
elif mozinfo.isMac:
utility = ['/usr/sbin/screencapture', '-C', '-x', '-t', 'png']
utilityname = "screencapture"
elif mozinfo.isWin:
utility = [os.path.join(utilityPath, "screenshot.exe")]
utilityname = "screenshot"
# Get dir where to write the screenshot file
parent_dir = os.environ.get('MOZ_UPLOAD_DIR', None)
if not parent_dir:
log.info('Failed to retrieve MOZ_UPLOAD_DIR env var')
return
# Run the capture
try:
tmpfd, imgfilename = tempfile.mkstemp(
prefix='mozilla-test-fail-screenshot_',
suffix='.png', dir=parent_dir
)
os.close(tmpfd)
if is_structured_log:
log.process_start(utilityname)
returncode = subprocess.call(utility + [imgfilename])
if is_structured_log:
log.process_exit(utilityname, returncode)
else:
printstatus(utilityname, returncode)
except OSError, err:
log.info("Failed to start %s for screenshot: %s"
% (utility[0], err.strerror))
|
