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
95
96
97
98
|
from urllib.parse import parse_qsl
import requests_mock
import mechanicalsoup
"""
Utilities for testing MechanicalSoup.
"""
choose_submit_form = '''
<html>
<body>
<!-- vaguely based on Trac edit-page form -->
<form id="choose-submit-form" method="post" action="mock://form.com/post">
<textarea id="text" class="wikitext trac-resizable" name="text"
cols="80" rows="40">
</textarea>
<div class="field">
<label>Comment about this change (optional):<br />
<input id="comment" type="text" name="comment" size="80" value="" />
</label>
</div>
<div class="buttons">
<input type="submit" name="preview" value="Preview Page" />
<input type="submit" name="diff" value="Review Changes" />
<input type="submit" id="save" name="save" value="Submit changes" />
<button type="submit" name="cancel" value="Cancel" />
</div>
</form>
</body>
</html>
'''
def setup_mock_browser(expected_post=None, text=choose_submit_form):
url = 'mock://form.com'
browser, mock = prepare_mock_browser()
mock_get(mock, url, text)
if expected_post is not None:
mock_post(mock, url + '/post', expected_post)
return browser, url
def prepare_mock_browser(scheme='mock'):
mock = requests_mock.Adapter()
browser = mechanicalsoup.StatefulBrowser(requests_adapters={scheme: mock})
return browser, mock
def mock_get(mocked_adapter, url, reply, content_type='text/html', **kwargs):
headers = {'Content-Type': content_type}
if isinstance(reply, str):
kwargs['text'] = reply
else:
kwargs['content'] = reply
mocked_adapter.register_uri('GET', url, headers=headers, **kwargs)
def mock_post(mocked_adapter, url, expected, reply='Success!'):
def text_callback(request, context):
query = parse_qsl(request.text)
assert query == expected
return reply
mocked_adapter.register_uri('POST', url, text=text_callback)
class HttpbinRemote:
"""Drop-in replacement for pytest-httpbin's httpbin fixture
that uses the remote httpbin server instead of a local one."""
def __init__(self):
self.url = "http://httpbin.org"
def __add__(self, x):
return self.url + x
def open_legacy_httpbin(browser, httpbin):
"""Opens the start page of httpbin (given as a fixture). Tries the
legacy page (available only on recent versions of httpbin), and if
it fails fall back to the main page (which is JavaScript-only in
recent versions of httpbin hence usable for us only on old
versions).
"""
try:
response = browser.open(httpbin + "/legacy")
if response.status_code == 404:
# The line above may or may not have raised the exception
# depending on raise_on_404. Raise it unconditionally now.
raise mechanicalsoup.LinkNotFoundError()
return response
except mechanicalsoup.LinkNotFoundError:
return browser.open(httpbin.url)
|
