annotate tests/mockutil.py @ 27:c898b4df0f29
Use context for html stripping, with support for custom URL sizes
For instance in Twitter URLs are 23 characters long since they use their
own URL shortening service. Without taking this into account, post lengths
would not be calculated correctly.
| author |
Ludovic Chabant <ludovic@chabant.com> |
| date |
Wed, 19 Apr 2023 12:46:58 -0700 |
| parents |
27543b2e73b9 |
| children |
b404445082f8 |
| rev |
line source |
|
2
|
1
|
|
|
2 def mock_urllib(m):
|
|
|
3 import urllib.request
|
|
|
4 m.setattr(urllib.request, 'urlretrieve', _patched_urlretrieve)
|
|
|
5 m.setattr(urllib.request, 'urlcleanup', _patched_urlcleanup)
|
|
|
6 return m
|
|
|
7
|
|
|
8
|
|
|
9 def _patched_urlretrieve(url):
|
|
|
10 return ('/retrieved/' + url.lstrip('/'), None)
|
|
|
11
|
|
|
12
|
|
|
13 def _patched_urlcleanup():
|
|
|
14 pass
|
