class VerifyTest(TestCase):
def setUp(self):
self.server = StubServer(8998)
def test_verify_checks_all_expectations(self):
satisfied_expectation = self._MockExpectation(True)
unsatisfied_expectation = self._MockExpectation(False)
self.server._expectations = [
satisfied_expectation,
unsatisfied_expectation,
satisfied_expectation
]
self.assertRaises(Exception, self.server.verify)
def test_verify_clears_all_expectations(self):
satisfied_expectation = self._MockExpectation(True)
self.server._expectations = [
satisfied_expectation,
satisfied_expectation,
satisfied_expectation
]
self.server.verify()
self.assertEqual([], self.server._expectations)
class _MockExpectation(object):
def __init__(self, satisfied):
self.satisfied = satisfied
class VerifyTest(TestCase):
def setUp(self):
self.server = StubServer(8998)
def test_verify_checks_all_expectations(self):
satisfied_expectation = self._MockExpectation(True)
unsatisfied_expectation = self._MockExpectation(False)
self.server._expectations = [
satisfied_expectation, unsatisfied_expectation,
satisfied_expectation
]
self.assertRaises(Exception, self.server.verify)
def test_verify_clears_all_expectations(self):
satisfied_expectation = self._MockExpectation(True)
self.server._expectations = [
satisfied_expectation, satisfied_expectation, satisfied_expectation
]
self.server.verify()
self.assertEqual([], self.server._expectations)
class _MockExpectation(object):
def __init__(self, satisfied):
self.satisfied = satisfied
class WebTest(object):
def setUp(self):
self.server = StubServer(8998)
self.server.run()
def tearDown(self):
self.server.stop()
self.server.verify()
def _make_request(self, url, method="GET", payload="", headers={}):
self.opener = urllib2.OpenerDirector()
self.opener.add_handler(urllib2.HTTPHandler())
request = urllib2.Request(url, headers=headers, data=payload)
request.get_method = lambda: method
response = self.opener.open(request)
response_code = getattr(response, 'code', -1)
return (response, response_code)
def test_get_with_file_call(self):
f = open('data.txt', 'w')
f.write("test file")
f.close()
self.server.expect(method="GET", url="/address/\d+$").and_return(mime_type="text/xml", file_content="./data.txt")
response, response_code = self._make_request("http://localhost:8998/address/25", method="GET")
expected = open("./data.txt", "r").read()
try:
self.assertEquals(expected, response.read())
finally:
response.close()
def test_put_with_capture(self):
capture = {}
self.server.expect(method="PUT", url="/address/\d+$", data_capture=capture).and_return(reply_code=201)
f, reply_code = self._make_request("http://localhost:8998/address/45", method="PUT", payload=str({"hello": "world", "hi": "mum"}))
try:
self.assertEquals("", f.read())
captured = eval(capture["body"])
self.assertEquals("world", captured["hello"])
self.assertEquals("mum", captured["hi"])
self.assertEquals(201, reply_code)
finally:
f.close()
def test_post_with_data_and_no_body_response(self):
self.server.expect(method="POST", url="address/\d+/inhabitant", data='<inhabitant name="Chris"/>').and_return(reply_code=204)
f, reply_code = self._make_request("http://localhost:8998/address/45/inhabitant", method="POST", payload='<inhabitant name="Chris"/>')
self.assertEquals(204, reply_code)
def test_get_with_data(self):
self.server.expect(method="GET", url="/monitor/server_status$").and_return(content="<html><body>Server is up</body></html>", mime_type="text/html")
f, reply_code = self._make_request("http://localhost:8998/monitor/server_status", method="GET")
try:
self.assertTrue("Server is up" in f.read())
self.assertEquals(200, reply_code)
finally:
f.close()
class FunctionalTest(unittest.TestCase):
def setUp(self):
self.server = StubServer(8998)
self.server.run()
def tearDown(self):
self.server.stop()
self.server.verify()
def test_get_with_file_call(self):
self.server.expect(method="GET", url="/address/\w+$").and_return(mime_type="text/xml", content="<address><number>12</number><street>Early Drive</street><city>Calgary</city></address>")
address = Address.objects.get(city="Calgary")
self.assertEquals("Early Drive", address.street)
class FunctionalTest(unittest.TestCase):
def setUp(self):
self.server = StubServer(8998)
self.server.run()
def tearDown(self):
self.server.stop()
self.server.verify()
def test_get_with_file_call(self):
self.server.expect(method="GET", url="/address/\w+$").and_return(
mime_type="text/xml",
content=
"<address><number>12</number><street>Early Drive</street><city>Calgary</city></address>"
)
address = Address.objects.get(city="Calgary")
self.assertEqual("Early Drive", address.street)
class WebTest(TestCase):
def setUp(self):
self.server = StubServer(8998)
self.server.run()
def tearDown(self):
self.server.stop()
self.server.verify() # this is redundant because stop includes verify
def _make_request(self, url, method="GET", payload=None, headers={}):
self.opener = OpenerDirector()
self.opener.add_handler(HTTPHandler())
if payload is None:
request = Request(url, headers=headers)
else:
request = Request(url, headers=headers, data=payload.encode('utf-8'))
request.get_method = lambda: method
response = self.opener.open(request)
response_code = getattr(response, 'code', -1)
return (response, response_code)
def test_get_with_file_call(self):
with open('data.txt', 'w') as f:
f.write("test file")
self.server.expect(method="GET", url="/address/\d+$").and_return(mime_type="text/xml", file_content="./data.txt")
response, response_code = self._make_request("http://localhost:8998/address/25", method="GET")
with open("./data.txt", "r") as f:
expected = f.read().encode('utf-8')
try:
self.assertEqual(expected, response.read())
finally:
response.close()
def test_put_with_capture(self):
capture = {}
self.server.expect(method="PUT", url="/address/\d+$", data_capture=capture).and_return(reply_code=201)
f, reply_code = self._make_request("http://localhost:8998/address/45", method="PUT", payload=str({"hello": "world", "hi": "mum"}))
try:
self.assertEqual(b"", f.read())
captured = eval(capture["body"])
self.assertEqual("world", captured["hello"])
self.assertEqual("mum", captured["hi"])
self.assertEqual(201, reply_code)
finally:
f.close()
def test_post_with_wrong_data(self):
self.server.expect(method="POST", url="data/", data='Bob').and_return()
f, reply_code = self._make_request("http://localhost:8998/data/", method="POST", payload='Chris')
self.assertEqual(403, reply_code)
self.assertRaises(Exception, self.server.stop)
def test_post_with_multiple_expectations_wrong_data(self):
self.server.expect(method="POST", url="data/", data='Bob').and_return(reply_code=201)
self.server.expect(method="POST", url="data/", data='John').and_return(reply_code=202)
self.server.expect(method="POST", url="data/", data='Dave').and_return(reply_code=203)
f, reply_code = self._make_request("http://localhost:8998/data/", method="POST", payload='Dave')
self.assertEqual(203, reply_code)
f, reply_code = self._make_request("http://localhost:8998/data/", method="POST", payload='Bob')
self.assertEqual(201, reply_code)
f, reply_code = self._make_request("http://localhost:8998/data/", method="POST", payload='Chris')
self.assertEqual(403, reply_code)
self.assertRaises(Exception, self.server.stop)
def test_post_with_mixed_expectations(self):
self.server.expect(method="POST", url="data/").and_return(reply_code=202)
self.server.expect(method="POST", url="data/", data='John').and_return(reply_code=201)
f, reply_code = self._make_request("http://localhost:8998/data/", method="POST", payload='John')
self.assertEqual(201, reply_code)
f, reply_code = self._make_request("http://localhost:8998/data/", method="POST", payload='Dave')
self.assertEqual(202, reply_code)
def test_post_with_data_and_no_body_response(self):
self.server.expect(method="POST", url="address/\d+/inhabitant", data='<inhabitant name="Chris"/>').and_return(reply_code=204)
f, reply_code = self._make_request("http://localhost:8998/address/45/inhabitant", method="POST", payload='<inhabitant name="Chris"/>')
self.assertEqual(204, reply_code)
def test_multiple_expectations_identifies_correct_unmatched_request(self):
self.server.expect(method="POST", url="address/\d+/inhabitant", data='Twas brillig and the slithy toves').and_return(reply_code=204)
f, reply_code = self._make_request("http://localhost:8998/address/45/inhabitant", method="POST", payload='Twas brillig and the slithy toves')
self.assertEqual(204, reply_code)
self.server.expect(method="GET", url="/monitor/server_status$").and_return(content="Four score and seven years ago", mime_type="text/html")
try:
self.server.stop()
except Exception as e:
self.assertEqual(-1, str(e).find('brillig'), str(e))
def test_get_with_data(self):
self.server.expect(method="GET", url="/monitor/server_status$").and_return(content="<html><body>Server is up</body></html>", mime_type="text/html")
f, reply_code = self._make_request("http://localhost:8998/monitor/server_status", method="GET")
try:
self.assertTrue(b"Server is up" in f.read())
self.assertEqual(200, reply_code)
finally:
f.close()
def test_get_from_root(self):
self.server.expect(method="GET", url="/$").and_return(content="<html><body>Server is up</body></html>", mime_type="text/html")
f, reply_code = self._make_request("http://localhost:8998/", method="GET")
try:
self.assertTrue(b"Server is up" in f.read())
self.assertEqual(200, reply_code)
finally:
f.close()
def test_put_when_post_expected(self):
# set expectations
self.server.expect(method="POST", url="address/\d+/inhabitant", data='<inhabitant name="Chris"/>').and_return(
reply_code=204)
# try a different method
f, reply_code = self._make_request("http://localhost:8998/address/45/inhabitant", method="PUT",
payload='<inhabitant name="Chris"/>')
# Validate the response
self.assertEqual("Method not allowed", f.msg)
self.assertEqual(405, reply_code)
self.assertTrue(f.read().startswith(b"Method PUT not allowed."))
# And we have an unmet expectation which needs to mention the POST that didn't happen
try:
self.server.stop()
except Exception as e:
self.assertTrue(str(e).find("POST") > 0, str(e))
def test_unexpected_get(self):
f, reply_code = self._make_request("http://localhost:8998/address/45/inhabitant", method="GET")
self.assertEqual(404, reply_code)
self.server.stop()
def test_repeated_get(self):
self.server.expect(method="GET", url="counter$").and_return(content="1")
self.server.expect(method="GET", url="counter$").and_return(content="2")
self.server.expect(method="GET", url="counter$").and_return(content="3")
for i in range(1, 4):
f, reply_code = self._make_request("http://localhost:8998/counter", method="GET")
self.assertEqual(200, reply_code)
self.assertEqual(str(i).encode('utf-8'), f.read())
def test_extra_get(self):
self.server.expect(method="GET", url="counter$").and_return(content="1")
f, reply_code = self._make_request("http://localhost:8998/counter", method="GET")
self.assertEqual(200, reply_code)
self.assertEqual(b"1", f.read())
f, reply_code = self._make_request("http://localhost:8998/counter", method="GET")
self.assertEqual(400, reply_code)
self.assertEqual("Expectations exhausted",f.msg)
self.assertTrue(f.read().startswith(b"Expectations at this URL have already been satisfied.\n"))
def test_returns_additional_headers_for_expectation_without_data(self):
self.server.expect(method="GET", url="/api/endpoint").and_return(
headers=(("some_header", "foo"), ("some_other_header", "bar"),))
r = requests.get("http://localhost:8998/api/endpoint")
self.assertEqual(r.headers["some_header"], "foo")
self.assertEqual(r.headers["some_other_header"], "bar")
def test_returns_additional_headers_for_expectation_with_data(self):
self.server.expect(method="POST", url="/api/endpoint", data="expected data").and_return(
headers=(("some_header", "foo"), ("some_other_header", "bar"),))
r = requests.post("http://localhost:8998/api/endpoint", data="expected data")
self.assertEqual(r.headers["some_header"], "foo")
self.assertEqual(r.headers["some_other_header"], "bar")
class WebTest(TestCase):
def setUp(self):
self.server = StubServer(8998)
self.server.run()
def tearDown(self):
self.server.stop()
self.server.verify() # this is redundant because stop includes verify
def _make_request(self, url, method="GET", payload=None, headers={}):
self.opener = OpenerDirector()
self.opener.add_handler(HTTPHandler())
if payload is None:
request = Request(url, headers=headers)
else:
request = Request(url,
headers=headers,
data=payload.encode('utf-8'))
request.get_method = lambda: method
response = self.opener.open(request)
response_code = getattr(response, 'code', -1)
return (response, response_code)
def test_get_with_file_call(self):
with open('data.txt', 'w') as f:
f.write("test file")
self.server.expect(method="GET", url="/address/\d+$").and_return(
mime_type="text/xml", file_content="./data.txt")
response, response_code = self._make_request(
"http://localhost:8998/address/25", method="GET")
with open("./data.txt", "r") as f:
expected = f.read().encode('utf-8')
try:
self.assertEqual(expected, response.read())
finally:
response.close()
def test_put_with_capture(self):
capture = {}
self.server.expect(method="PUT",
url="/address/\d+$",
data_capture=capture).and_return(reply_code=201)
f, reply_code = self._make_request("http://localhost:8998/address/45",
method="PUT",
payload=str({
"hello": "world",
"hi": "mum"
}))
try:
self.assertEqual(b"", f.read())
captured = eval(capture["body"])
self.assertEqual("world", captured["hello"])
self.assertEqual("mum", captured["hi"])
self.assertEqual(201, reply_code)
finally:
f.close()
def test_post_with_wrong_data(self):
self.server.expect(method="POST", url="data/", data='Bob').and_return()
f, reply_code = self._make_request("http://localhost:8998/data/",
method="POST",
payload='Chris')
self.assertEqual(403, reply_code)
self.assertRaises(Exception, self.server.stop)
def test_post_with_multiple_expectations_wrong_data(self):
self.server.expect(method="POST", url="data/",
data='Bob').and_return(reply_code=201)
self.server.expect(method="POST", url="data/",
data='John').and_return(reply_code=202)
self.server.expect(method="POST", url="data/",
data='Dave').and_return(reply_code=203)
f, reply_code = self._make_request("http://localhost:8998/data/",
method="POST",
payload='Dave')
self.assertEqual(203, reply_code)
f, reply_code = self._make_request("http://localhost:8998/data/",
method="POST",
payload='Bob')
self.assertEqual(201, reply_code)
f, reply_code = self._make_request("http://localhost:8998/data/",
method="POST",
payload='Chris')
self.assertEqual(403, reply_code)
self.assertRaises(Exception, self.server.stop)
def test_post_with_mixed_expectations(self):
self.server.expect(method="POST",
url="data/").and_return(reply_code=202)
self.server.expect(method="POST", url="data/",
data='John').and_return(reply_code=201)
f, reply_code = self._make_request("http://localhost:8998/data/",
method="POST",
payload='John')
self.assertEqual(201, reply_code)
f, reply_code = self._make_request("http://localhost:8998/data/",
method="POST",
payload='Dave')
self.assertEqual(202, reply_code)
def test_post_with_data_and_no_body_response(self):
self.server.expect(
method="POST",
url="address/\d+/inhabitant",
data='<inhabitant name="Chris"/>').and_return(reply_code=204)
f, reply_code = self._make_request(
"http://localhost:8998/address/45/inhabitant",
method="POST",
payload='<inhabitant name="Chris"/>')
self.assertEqual(204, reply_code)
def test_multiple_expectations_identifies_correct_unmatched_request(self):
self.server.expect(
method="POST",
url="address/\d+/inhabitant",
data='Twas brillig and the slithy toves').and_return(
reply_code=204)
f, reply_code = self._make_request(
"http://localhost:8998/address/45/inhabitant",
method="POST",
payload='Twas brillig and the slithy toves')
self.assertEqual(204, reply_code)
self.server.expect(method="GET",
url="/monitor/server_status$").and_return(
content="Four score and seven years ago",
mime_type="text/html")
try:
self.server.stop()
except Exception as e:
self.assertEqual(-1, str(e).find('brillig'), str(e))
def test_get_with_data(self):
self.server.expect(
method="GET", url="/monitor/server_status$").and_return(
content="<html><body>Server is up</body></html>",
mime_type="text/html")
f, reply_code = self._make_request(
"http://localhost:8998/monitor/server_status", method="GET")
try:
self.assertTrue(b"Server is up" in f.read())
self.assertEqual(200, reply_code)
finally:
f.close()
def test_get_from_root(self):
self.server.expect(method="GET", url="/$").and_return(
content="<html><body>Server is up</body></html>",
mime_type="text/html")
f, reply_code = self._make_request("http://localhost:8998/",
method="GET")
try:
self.assertTrue(b"Server is up" in f.read())
self.assertEqual(200, reply_code)
finally:
f.close()
def test_put_when_post_expected(self):
# set expectations
self.server.expect(
method="POST",
url="address/\d+/inhabitant",
data='<inhabitant name="Chris"/>').and_return(reply_code=204)
# try a different method
f, reply_code = self._make_request(
"http://localhost:8998/address/45/inhabitant",
method="PUT",
payload='<inhabitant name="Chris"/>')
# Validate the response
self.assertEqual("Method not allowed", f.msg)
self.assertEqual(405, reply_code)
self.assertTrue(f.read().startswith(b"Method PUT not allowed."))
# And we have an unmet expectation which needs to mention the POST that didn't happen
try:
self.server.stop()
except Exception as e:
self.assertTrue(str(e).find("POST") > 0, str(e))
def test_unexpected_get(self):
f, reply_code = self._make_request(
"http://localhost:8998/address/45/inhabitant", method="GET")
self.assertEqual(404, reply_code)
self.server.stop()
def test_repeated_get(self):
self.server.expect(method="GET",
url="counter$").and_return(content="1")
self.server.expect(method="GET",
url="counter$").and_return(content="2")
self.server.expect(method="GET",
url="counter$").and_return(content="3")
for i in range(1, 4):
f, reply_code = self._make_request("http://localhost:8998/counter",
method="GET")
self.assertEqual(200, reply_code)
self.assertEqual(str(i).encode('utf-8'), f.read())
def test_extra_get(self):
self.server.expect(method="GET",
url="counter$").and_return(content="1")
f, reply_code = self._make_request("http://localhost:8998/counter",
method="GET")
self.assertEqual(200, reply_code)
self.assertEqual(b"1", f.read())
f, reply_code = self._make_request("http://localhost:8998/counter",
method="GET")
self.assertEqual(400, reply_code)
self.assertEqual("Expectations exhausted", f.msg)
self.assertTrue(f.read().startswith(
b"Expectations at this URL have already been satisfied.\n"))
class WebTest(TestCase):
def setUp(self):
self.server = StubServer(8998)
self.server.run()
def tearDown(self):
self.server.stop()
self.server.verify()
def _make_request(self, url, method="GET", payload="", headers={}):
self.opener = urllib2.OpenerDirector()
self.opener.add_handler(urllib2.HTTPHandler())
request = urllib2.Request(url, headers=headers, data=payload)
request.get_method = lambda: method
response = self.opener.open(request)
response_code = getattr(response, 'code', -1)
return (response, response_code)
def test_get_with_file_call(self):
f = open('data.txt', 'w')
f.write("test file")
f.close()
self.server.expect(method="GET", url="/address/\d+$").and_return(mime_type="text/xml", file_content="./data.txt")
response, response_code = self._make_request("http://localhost:8998/address/25", method="GET")
expected = open("./data.txt", "r").read()
try:
self.assertEquals(expected, response.read())
finally:
response.close()
def test_put_with_capture(self):
capture = {}
self.server.expect(method="PUT", url="/address/\d+$", data_capture=capture).and_return(reply_code=201)
f, reply_code = self._make_request("http://localhost:8998/address/45", method="PUT", payload=str({"hello": "world", "hi": "mum"}))
try:
self.assertEquals("", f.read())
captured = eval(capture["body"])
self.assertEquals("world", captured["hello"])
self.assertEquals("mum", captured["hi"])
self.assertEquals(201, reply_code)
finally:
f.close()
def test_post_with_data_and_no_body_response(self):
self.server.expect(method="POST", url="address/\d+/inhabitant", data='<inhabitant name="Chris"/>').and_return(reply_code=204)
f, reply_code = self._make_request("http://localhost:8998/address/45/inhabitant", method="POST", payload='<inhabitant name="Chris"/>')
self.assertEquals(204, reply_code)
def test_get_with_data(self):
self.server.expect(method="GET", url="/monitor/server_status$").and_return(content="<html><body>Server is up</body></html>", mime_type="text/html")
f, reply_code = self._make_request("http://localhost:8998/monitor/server_status", method="GET")
try:
self.assertTrue("Server is up" in f.read())
self.assertEquals(200, reply_code)
finally:
f.close()
def test_get_from_root(self):
self.server.expect(method="GET", url="/$").and_return(content="<html><body>Server is up</body></html>", mime_type="text/html")
f, reply_code = self._make_request("http://localhost:8998/", method="GET")
try:
self.assertTrue("Server is up" in f.read())
self.assertEquals(200, reply_code)
finally:
f.close()