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
|
from contextlib import contextmanager
from io import BytesIO, TextIOWrapper, BufferedIOBase
import random
from typing import cast
from unittest.mock import patch
import urllib.request
# Note: the casts of StringIO/BytesIO to BufferedIOBase are only necessary because of
# pycharm bug PY-28155
def random_text(n=1024):
return ''.join(chr(random.randint(32, 126)) for _ in range(n))
class MockStdout(object):
def __init__(self, name, as_bytes):
self.bytes_io = BytesIO()
object.__setattr__(self.bytes_io, 'name', name)
self.wrapper = TextIOWrapper(cast(BufferedIOBase, self.bytes_io))
self.wrapper.mode = 'w'
self.as_bytes = as_bytes
def getvalue(self):
self.wrapper.flush()
val = self.bytes_io.getvalue()
if not self.as_bytes:
val = val.decode()
return val
@contextmanager
def intercept_stdout(as_bytes=False):
i = MockStdout('<stdout>', as_bytes)
with patch('sys.stdout', i.wrapper):
yield i
@contextmanager
def intercept_stderr(as_bytes=False):
i = MockStdout('<stderr>', as_bytes)
with patch('sys.stderr', i.wrapper):
yield i
@contextmanager
def intercept_stdin(content, is_bytes=False):
if not is_bytes:
content = content.encode()
i = BytesIO()
object.__setattr__(i, 'name', '<stdin>')
i.write(content)
if not (is_bytes or content.endswith(b'\n')):
i.write(b'\n')
i.seek(0)
i = TextIOWrapper(cast(BufferedIOBase, i))
i.mode = 'r'
with patch('sys.stdin', i):
yield
def no_internet():
"""Test whether there's no internet connection available.
"""
try:
urllib.request.urlopen("https://github.com").info()
return False
except:
return True
|
