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
|
from twilio.base.exceptions import TwilioRestException
from twilio.http import HttpClient
from twilio.http.request import Request
import platform
from twilio import __version__
class Hologram(object):
def __init__(self, request, response):
self.request = request
self.response = response
class Holodeck(HttpClient):
def __init__(self):
self._holograms = []
self._requests = []
def mock(self, response, request=None):
request = request or Request()
self._holograms.append(Hologram(request, response))
@property
def requests(self):
return self._requests
def add_standard_headers(self, request):
standard_headers = {
'User-Agent': 'twilio-python/{} (Python {})'.format(
__version__,
platform.python_version()),
'X-Twilio-Client': 'python-{}'.format(__version__),
'Accept': 'application/json',
'Accept-Charset': 'utf-8'
}
if request.method == 'POST' and 'Content-Type' not in standard_headers:
standard_headers['Content-Type'] = 'application/x-www-form-urlencoded'
standard_headers.update(request.headers)
request.headers = standard_headers
return request
def assert_has_request(self, request):
for req in self.requests:
if req == request or req == self.add_standard_headers(request):
return
message = '\nHolodeck never received a request matching: \n + {}\n'.format(request)
if self._requests:
message += 'Requests received:\n'
message += '\n'.join(' * {}'.format(r) for r in self.requests)
else:
message += 'No Requests received'
raise AssertionError(message)
def request(self,
method,
url,
params=None,
data=None,
headers=None,
auth=None,
timeout=None,
allow_redirects=False):
request = Request(method, url, auth, params, data, headers)
self._requests.append(request)
for hologram in self._holograms:
if hologram.request == request:
return hologram.response
message = '\nHolodeck has no hologram for: {}\n'.format(request)
if self._holograms:
message += 'Holograms loaded:\n'
message += '\n'.join(' - {}'.format(h.request) for h in self._holograms)
else:
message += 'No Holograms loaded'
raise TwilioRestException(404, url, message, method=method)
|
