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
|
#!/usr/bin/env python
import datetime
import os
from builtins import Exception
import mozinfo
import mozunit
import requests
from mozproxy import get_playback
from support import tempdir
here = os.path.dirname(__file__)
os.environ["MOZPROXY_DIR"] = os.path.join(here, "files")
def get_status_code(url, playback):
response = requests.get(
url=url, proxies={"http": "http://%s:%s/" % (playback.host, playback.port)}
)
return response.status_code
def test_record_and_replay(*args):
# test setup
basename = "recording"
suffix = datetime.datetime.now().strftime("%y%m%d_%H%M%S")
filename = "_".join([basename, suffix])
recording_file = os.path.join(here, "files", ".".join([filename, "zip"]))
# Record part
config = {
"playback_tool": "mitmproxy",
"recording_file": recording_file,
"playback_version": "8.1.1",
"platform": mozinfo.os,
"run_local": "MOZ_AUTOMATION" not in os.environ,
"binary": "firefox",
"app": "firefox",
"host": "127.0.0.1",
"record": True,
}
with tempdir() as obj_path:
config["obj_path"] = obj_path
record = get_playback(config)
assert record is not None
try:
record.start()
url = "https://m.media-amazon.com/images/G/01/csm/showads.v2.js"
assert get_status_code(url, record) == 200
finally:
record.stop()
# playback part
config["record"] = False
config["recording_file"] = None
config["playback_files"] = [recording_file]
playback = get_playback(config)
assert playback is not None
try:
playback.start()
url = "https://m.media-amazon.com/images/G/01/csm/showads.v2.js"
assert get_status_code(url, playback) == 200
finally:
playback.stop()
# Cleanup
try:
os.remove(recording_file)
os.remove(os.path.join("distribution", "policies.json"))
except Exception:
pass
if __name__ == "__main__":
mozunit.main(runwith="pytest")
|
