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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
#!/usr/bin/env python
import os
import platform
import subprocess
from pathlib import Path
from unittest import mock
import mozunit
import pytest
from mozperftest.system.macos import MacosDevice
from mozperftest.tests.support import DMG, get_running_env
def run_proc(*args, **kw):
if args[0][1] == "attach":
where = args[0][4]
bindir = Path(where, "firefox.app", "Contents", "MacOS")
os.makedirs(str(bindir))
firefox_bin = bindir / "firefox"
with firefox_bin.open("w") as f:
f.write("OK")
def mock_calls(test):
# on macOS we don't mock the system calls
# so we're mounting for real using hdiutil
if platform.system() == "Darwin":
return test
# on other platforms, we're unsing run_proc
@mock.patch("mozperftest.system.macos.MacosDevice._run_process", new=run_proc)
def wrapped(*args, **kw):
return test(*args, **kw)
@mock_calls
def test_mount_dmg():
mach_cmd, metadata, env = get_running_env(browsertime_binary=str(DMG))
device = MacosDevice(env, mach_cmd)
try:
device.run(metadata)
finally:
device.teardown()
target = Path(DMG.parent, "firefox", "Contents", "MacOS", "firefox")
assert env.get_arg("browsertime-binary") == str(target)
def run_fail(cmd):
def _run_fail(self, args):
run_cmd = " ".join(args)
if cmd not in run_cmd:
run_proc(args)
return
raise subprocess.CalledProcessError(returncode=2, cmd=" ".join(args))
return _run_fail
@mock.patch("mozperftest.system.macos.MacosDevice._run_process", new=run_fail("attach"))
def test_attach_fails():
mach_cmd, metadata, env = get_running_env(browsertime_binary=str(DMG))
device = MacosDevice(env, mach_cmd)
with pytest.raises(subprocess.CalledProcessError):
try:
device.run(metadata)
finally:
device.teardown()
@mock.patch("mozperftest.system.macos.MacosDevice._run_process", new=run_fail("detach"))
def test_detach_fails():
mach_cmd, metadata, env = get_running_env(browsertime_binary=str(DMG))
device = MacosDevice(env, mach_cmd)
# detaching will be swallowed
try:
device.run(metadata)
finally:
device.teardown()
target = Path(DMG.parent, "firefox", "Contents", "MacOS", "firefox")
assert env.get_arg("browsertime-binary") == str(target)
def test_no_op():
mach_cmd, metadata, env = get_running_env(browsertime_binary="notadmg")
device = MacosDevice(env, mach_cmd)
device.run(metadata)
if __name__ == "__main__":
mozunit.main()
|
