def test_register(self):
with mock.patch('time.gmtime') as tt:
tt.return_value = time.struct_time((2013, 4, 30, 10, 39, 21, 1, 120, 0))
Entry.register(
Entry.GET,
'http://testme.org/get?a=1&b=2#test',
Response('{"a": "€"}', headers={'content-type': 'application/json'})
)
entries = Mocket._entries[('testme.org', 80)]
self.assertEqual(len(entries), 1)
entry = entries[0]
self.assertEqual(entry.method, 'GET')
self.assertEqual(entry.schema, 'http')
self.assertEqual(entry.path, '/get')
self.assertEqual(entry.query, 'a=1&b=2')
self.assertEqual(len(entry.responses), 1)
response = entry.responses[0]
self.assertEqual(response.body, b'{"a": "\xe2\x82\xac"}')
self.assertEqual(response.status, 200)
self.assertEqual(response.headers, {
'Status': '200',
'Date': 'Tue, 30 Apr 2013 10:39:21 GMT',
'Connection': 'close',
'Server': 'Python/Mocket',
'Content-Length': '12',
'Content-Type': 'application/json',
})
def test_multipart(self):
url = "http://httpbin.org/post"
data = '--xXXxXXyYYzzz\r\nContent-Disposition: form-data; name="content"\r\nContent-Type: text/plain; charset=utf-8\r\nContent-Length: 68\r\n\r\nAction: comment\nText: Comment with attach\nAttachment: x1.txt, x2.txt\r\n--xXXxXXyYYzzz\r\nContent-Disposition: form-data; name="attachment_2"; filename="x.txt"\r\nContent-Type: text/plain\r\nContent-Length: 4\r\n\r\nbye\n\r\n--xXXxXXyYYzzz\r\nContent-Disposition: form-data; name="attachment_1"; filename="x.txt"\r\nContent-Type: text/plain\r\nContent-Length: 4\r\n\r\nbye\n\r\n--xXXxXXyYYzzz--\r\n'
headers = {
"Content-Length": "495",
"Content-Type": "multipart/form-data; boundary=xXXxXXyYYzzz",
"Accept": "text/plain",
"User-Agent": "Mocket",
"Accept-encoding": "identity",
}
Entry.register(Entry.POST, url)
response = requests.post(url, data=data, headers=headers)
self.assertEqual(response.status_code, 200)
last_request = Mocket.last_request()
self.assertEqual(last_request.method, "POST")
self.assertEqual(last_request.path, "/post")
self.assertEqual(last_request.body, data)
sent_headers = dict(last_request.headers)
self.assertEqualHeaders(
sent_headers,
{
"accept": "text/plain",
"accept-encoding": "identity",
"content-length": "495",
"content-type": "multipart/form-data; boundary=xXXxXXyYYzzz",
"host": "httpbin.org",
"user-agent": "Mocket",
"connection": "keep-alive",
},
)
self.assertEqual(len(Mocket._requests), 1)
def test_register(self):
with mock.patch("time.gmtime") as tt:
tt.return_value = time.struct_time((2013, 4, 30, 10, 39, 21, 1, 120, 0))
Entry.register(
Entry.GET,
"http://testme.org/get?a=1&b=2#test",
Response('{"a": "€"}', headers={"content-type": "application/json"}),
)
entries = Mocket._entries[("testme.org", 80)]
self.assertEqual(len(entries), 1)
entry = entries[0]
self.assertEqual(entry.method, "GET")
self.assertEqual(entry.schema, "http")
self.assertEqual(entry.path, "/get")
self.assertEqual(entry.query, "a=1&b=2")
self.assertEqual(len(entry.responses), 1)
response = entry.responses[0]
self.assertEqual(response.body, b'{"a": "\xe2\x82\xac"}')
self.assertEqual(response.status, 200)
self.assertEqual(
response.headers,
{
"Status": "200",
"Date": "Tue, 30 Apr 2013 10:39:21 GMT",
"Connection": "close",
"Server": "Python/Mocket",
"Content-Length": "12",
"Content-Type": "application/json",
},
)
def test_multipart(self):
url = 'http://httpbin.org/post'
data = '--xXXxXXyYYzzz\r\nContent-Disposition: form-data; name="content"\r\nContent-Type: text/plain; charset=utf-8\r\nContent-Length: 68\r\n\r\nAction: comment\nText: Comment with attach\nAttachment: x1.txt, x2.txt\r\n--xXXxXXyYYzzz\r\nContent-Disposition: form-data; name="attachment_2"; filename="x.txt"\r\nContent-Type: text/plain\r\nContent-Length: 4\r\n\r\nbye\n\r\n--xXXxXXyYYzzz\r\nContent-Disposition: form-data; name="attachment_1"; filename="x.txt"\r\nContent-Type: text/plain\r\nContent-Length: 4\r\n\r\nbye\n\r\n--xXXxXXyYYzzz--\r\n'
headers = {
'Content-Length': '495',
'Content-Type': 'multipart/form-data; boundary=xXXxXXyYYzzz',
'Accept': 'text/plain',
'User-Agent': 'Mocket',
'Accept-encoding': 'identity',
}
Entry.register(Entry.POST, url)
response = requests.post(url, data=data, headers=headers)
self.assertEqual(response.status_code, 200)
last_request = Mocket.last_request()
self.assertEqual(last_request.method, 'POST')
self.assertEqual(last_request.path, '/post')
self.assertEqual(last_request.body, data)
sent_headers = dict(last_request.headers)
self.assertEqualHeaders(
sent_headers,
{
'accept': 'text/plain',
'accept-encoding': 'identity',
'content-length': '495',
'content-type': 'multipart/form-data; boundary=xXXxXXyYYzzz',
'host': 'httpbin.org',
'user-agent': 'Mocket',
'connection': 'keep-alive',
}
)
def test_multipart(self):
url = 'http://httpbin.org/post'
data = '--xXXxXXyYYzzz\r\nContent-Disposition: form-data; name="content"\r\nContent-Type: text/plain; charset=utf-8\r\nContent-Length: 68\r\n\r\nAction: comment\nText: Comment with attach\nAttachment: x1.txt, x2.txt\r\n--xXXxXXyYYzzz\r\nContent-Disposition: form-data; name="attachment_2"; filename="x.txt"\r\nContent-Type: text/plain\r\nContent-Length: 4\r\n\r\nbye\n\r\n--xXXxXXyYYzzz\r\nContent-Disposition: form-data; name="attachment_1"; filename="x.txt"\r\nContent-Type: text/plain\r\nContent-Length: 4\r\n\r\nbye\n\r\n--xXXxXXyYYzzz--\r\n'
headers = {
'Content-Length': '495',
'Content-Type': 'multipart/form-data; boundary=xXXxXXyYYzzz',
'Accept': 'text/plain',
'User-Agent': 'Mocket',
'Accept-encoding': 'identity',
}
Entry.register(Entry.POST, url)
response = requests.post(url, data=data, headers=headers)
self.assertEqual(response.status_code, 200)
last_request = Mocket.last_request()
self.assertEqual(last_request.method, 'POST')
self.assertEqual(last_request.path, '/post')
self.assertEqual(last_request.body, data)
sent_headers = dict(last_request.headers)
self.assertEqualHeaders(
sent_headers, {
'accept': 'text/plain',
'accept-encoding': 'identity',
'content-length': '495',
'content-type': 'multipart/form-data; boundary=xXXxXXyYYzzz',
'host': 'httpbin.org',
'user-agent': 'Mocket',
'connection': 'keep-alive',
})
self.assertEqual(len(Mocket._requests), 1)
def test_register(self):
with mock.patch('time.gmtime') as tt:
tt.return_value = time.struct_time(
(2013, 4, 30, 10, 39, 21, 1, 120, 0))
Entry.register(
Entry.GET, 'http://testme.org/get?a=1&b=2#test',
Response('{"a": "€"}',
headers={'content-type': 'application/json'}))
entries = Mocket._entries[('testme.org', 80)]
self.assertEqual(len(entries), 1)
entry = entries[0]
self.assertEqual(entry.method, 'GET')
self.assertEqual(entry.schema, 'http')
self.assertEqual(entry.path, '/get')
self.assertEqual(entry.query, 'a=1&b=2')
self.assertEqual(len(entry.responses), 1)
response = entry.responses[0]
self.assertEqual(response.body, b'{"a": "\xe2\x82\xac"}')
self.assertEqual(response.status, 200)
self.assertEqual(
response.headers, {
u'Status': u'200',
u'Date': u'Tue, 30 Apr 2013 10:39:21 GMT',
u'Connection': u'close',
u'Server': u'Python/Mocket',
u'Content-Length': u'12',
u'Content-Type': u'application/json',
})
def test_multi_register(self):
url = "http://foobar.com/path"
Entry.register(
Entry.POST,
url,
Response(body='{"foo":"bar0"}', status=200),
Response(body='{"foo":"bar1"}', status=201),
Response(body='{"foo":"bar2"}', status=202),
)
response = requests.post(url, json={"test": 0})
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json(), {"foo": "bar0"})
response = requests.post(url, json={"test": 1})
self.assertEqual(response.status_code, 201)
self.assertEqual(response.json(), {"foo": "bar1"})
response = requests.post(url, json={"test": 2})
self.assertEqual(response.status_code, 202)
self.assertEqual(response.json(), {"foo": "bar2"})
response = requests.post(url, json={"test": 22})
self.assertEqual(response.status_code, 202)
self.assertEqual(response.json(), {"foo": "bar2"})
def test_sendall_double(self):
Entry.register(Entry.GET, 'http://testme.org/', Response(status=404), Response())
self.assertRaises(HTTPError, urlopen, 'http://testme.org/')
response = urlopen('http://testme.org/')
self.assertEqual(response.code, 200)
response = urlopen('http://testme.org/')
self.assertEqual(response.code, 200)
self.assertEqual(len(Mocket._requests), 3)
def test_suggestion_for_register_and_status(self):
url = "http://foobar.com/path"
with self.assertRaises(AttributeError):
Entry.register(
Entry.POST,
url,
status=201,
)
def test_suggestion_for_register_and_body(self):
url = "http://foobar.com/path"
with self.assertRaises(AttributeError):
Entry.register(
Entry.POST,
url,
body='{"foo":"bar0"}',
)
def test_sendall_double(self):
Entry.register(Entry.GET, 'http://testme.org/', Response(status=404),
Response())
self.assertRaises(HTTPError, urlopen, 'http://testme.org/')
response = urlopen('http://testme.org/')
self.assertEqual(response.code, 200)
response = urlopen('http://testme.org/')
self.assertEqual(response.code, 200)
self.assertEqual(len(Mocket._requests), 3)
def test_file_object_with_no_lib_magic(self):
url = "http://github.com/fluidicon.png"
filename = "tests/fluidicon.png"
file_obj = open(filename, "rb")
Entry.register(Entry.GET, url, Response(body=file_obj, lib_magic=None))
r = requests.get(url)
remote_content = r.content
local_file_obj = open(filename, "rb")
local_content = local_file_obj.read()
self.assertEqual(remote_content, local_content)
self.assertEqual(len(remote_content), len(local_content))
self.assertEqual(int(r.headers["Content-Length"]), len(local_content))
with self.assertRaises(KeyError):
self.assertEqual(r.headers["Content-Type"], "image/png")
def test_file_object_with_no_lib_magic(self):
url = 'http://github.com/fluidicon.png'
filename = 'tests/fluidicon.png'
file_obj = open(filename, 'rb')
Entry.register(Entry.GET, url, Response(body=file_obj, lib_magic=None))
r = requests.get(url)
remote_content = r.content
local_file_obj = open(filename, 'rb')
local_content = local_file_obj.read()
self.assertEqual(remote_content, local_content)
self.assertEqual(len(remote_content), len(local_content))
self.assertEqual(int(r.headers['Content-Length']), len(local_content))
with self.assertRaises(KeyError):
self.assertEqual(r.headers['Content-Type'], 'image/png')
def test_wrongpath_truesendall(self):
Entry.register(Entry.GET, 'http://httpbin.org/user.agent', Response())
response = urlopen('http://httpbin.org/ip')
self.assertEqual(response.code, 200)
def test_wrongpath_truesendall(self):
Entry.register(Entry.GET, "http://httpbin.org/user.agent", Response(status=404))
response = urlopen("http://httpbin.org/ip")
self.assertEqual(response.code, 200)
def test_wrongpath_truesendall(self):
Entry.register(Entry.GET, 'http://httpbin.org/user.agent', Response())
response = urlopen('http://httpbin.org/ip')
self.assertEqual(response.code, 200)
