def test_expect_atmost_once(self):
self.client.expect(request(), response(), timesrange(atmost=1))
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 200)
self.client.verify_expectations()
def test_delay_response_with_special_unit(self):
self.client.stub(request(), response(delay=milliseconds(500)))
start = time.time()
requests.get(MOCK_SERVER_URL)
elapsed = time.time() - start
self.assertGreater(elapsed, 0.5)
def test_time_to_live_stubbing(self):
self.client.stub(request(), response(), time_to_live=1)
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 200)
time.sleep(1)
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 404)
def test_delay_response(self):
self.client.stub(request(), response(delay=1))
start = time.time()
requests.get(MOCK_SERVER_URL)
elapsed = time.time() - start
self.assertGreater(elapsed, 1)
def test_path_stubbing(self):
self.client.stub(request(path="/path"), response(code=200))
result = requests.get(MOCK_SERVER_URL + "/whatrver")
self.assertEqual(result.status_code, 404)
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 200)
def test_expect_atleast_twice_fails(self):
self.client.expect(request(), response(), timesrange(atleast=2))
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 200)
with self.assertRaises(AssertionError):
self.client.verify_expectations()
def test_expect_never_fails(self):
self.client.expect(request(), response(), times(0))
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 404)
with self.assertRaises(AssertionError):
self.client.verify_expectations()
def test_expect_once_called_twice_fails(self):
self.client.expect(request(), response(), times(1))
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 200)
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 404)
def test_default_count_for_stubs_is_unlimited(self):
self.client.stub(request(), response())
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 200)
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 200)
def test_count_stubbing(self):
self.client.stub(request(), response(), times(1))
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 200)
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 404)
def test_body_response(self):
self.client.stub(
request(),
response(body="hey there")
)
result = requests.get(MOCK_SERVER_URL)
self.assertEqual(result.content.decode(), "hey there")
def test_headers_response(self):
self.client.stub(
request(),
response(headers={"i-like": "i-like"})
)
result = requests.get(MOCK_SERVER_URL)
self.assertEqual(result.headers["i-like"], "i-like")
def test_body_stubbing(self):
self.client.stub(request(body="hey there"), response(code=200))
result = requests.post(MOCK_SERVER_URL, data="sup?")
self.assertEqual(result.status_code, 404)
result = requests.post(MOCK_SERVER_URL, data="hey there")
self.assertEqual(result.status_code, 200)
def test_code_response(self):
self.client.stub(
request(),
response(code=418)
)
result = requests.get(MOCK_SERVER_URL)
self.assertEqual(result.status_code, 418)
def test_method_stubbing(self):
self.client.stub(request(method="POST"), response(code=200))
result = requests.get(MOCK_SERVER_URL)
self.assertEqual(result.status_code, 404)
result = requests.post(MOCK_SERVER_URL)
self.assertEqual(result.status_code, 200)
def test_catch_all(self):
self.client.stub(
request(),
response()
)
result = requests.get(MOCK_SERVER_URL + "/whatever")
self.assertEqual(result.status_code, 200)
def get_authn_cookies(self, email):
self.client.stub(request(method="GET", path="/o/oauth2/auth"),
response(code=200, body="fake google auth"))
url = f"{self.base}/login"
r = requests.get(url)
assert r.status_code == 200
# Retrieve the state that our app sent to the mock server when it redirected
# to the google oauth2 auth endpoint.
# http://www.mock-server.com/mock_server/mockserver_clients.html#rest-api shows
# the mock-server endpoint that can retrieve requests it received.
# Requested this feature be added to the mock-server python client here
# https://github.com/internap/python-mockserver-client/issues/16.
data = json.dumps({"path": "/o/oauth2/auth", "method": "GET"})
r = requests.put("http://mockserver:1080/retrieve?type=REQUESTS",
data=data)
assert 200 == r.status_code
state = r.json()[-1]["queryStringParameters"]["state"][0]
assert "" != state
# Mock out the app attempting to get an access token using the state and code from Google
tokenJson = json.dumps({
"access_token": "myfakeaccesstoken",
"token_type": "Bearer",
"expires_in": 0,
})
self.client.stub(
request(method="POST", path="/o/oauth2/token"),
response(code=200,
body=tokenJson,
headers={"Content-Type": "application/json"}))
# Mock out the app attempting to get profile data using the access token
data = json.dumps({"email": email})
self.client.stub(request(method="GET", path="/oauth2/v2/userinfo"),
response(code=200, body=data))
# Perform the callback from Google to finish the authentication with the app, return the cookie
code = "supersecretcode"
r = requests.get(f"{self.base}/callback?state={state}&code={code}")
assert 204 == r.status_code
cookie = r.cookies['dndtextapisession']
return dict(dndtextapisession=cookie)
def test_simple_body(self):
self.client.stub(request(),
json_response(code=207, body={"pay": {
"lo": "ad"
}}))
result = requests.get(MOCK_SERVER_URL)
self.assertEqual(result.status_code, 207)
self.assertEqual(result.json(), {"pay": {"lo": "ad"}})
self.assertEqual(result.headers["Content-Type"], "application/json")
def test_headers_stubbing(self):
self.client.stub(request(headers={"i-am-special": "yeah you are"}),
response(code=200))
result = requests.get(MOCK_SERVER_URL)
self.assertEqual(result.status_code, 404)
result = requests.get(MOCK_SERVER_URL,
headers={"i-am-special": "yeah you are"})
self.assertEqual(result.status_code, 200)
def test_cookies_stubbing(self):
self.client.stub(request(cookies={"i-am-cookie": "sweet-cookie"}),
response(code=200))
result = requests.get(MOCK_SERVER_URL)
self.assertEqual(result.status_code, 404)
result = requests.get(MOCK_SERVER_URL,
cookies={"i-am-cookie": "sweet-cookie"})
self.assertEqual(result.status_code, 200)
def test_form_request(self):
self.client.stub(request(body=form({
"a": "b",
"c[0]": "d"
})), response())
result = requests.post(MOCK_SERVER_URL, data={"a": "b"})
self.assertEqual(result.status_code, 404)
result = requests.post(MOCK_SERVER_URL, data={"a": "b", "c[0]": "d"})
self.assertEqual(result.status_code, 200)
def test_content_type_header_is_added_to_specified(self):
self.client.stub(
request(),
json_response(body={"pay": {
"lo": "ad"
}},
headers={"i-like": "i-like"}))
result = requests.get(MOCK_SERVER_URL)
self.assertEqual(result.headers["i-like"], "i-like")
self.assertEqual(result.headers["Content-Type"], "application/json")
def test_querystring_stubbing(self):
self.client.stub(request(querystring={
"a": "b",
"c[0]": "d"
}), response(code=200))
result = requests.get(MOCK_SERVER_URL + "/?e=f")
self.assertEqual(result.status_code, 404)
result = requests.get(MOCK_SERVER_URL + "/?a=b&c[0]=d")
self.assertEqual(result.status_code, 200)
def test_expect_json_contains(self):
self.client.expect(request(body=json_contains({"key": "value"})),
response(), times(1))
result = requests.get(MOCK_SERVER_URL + "/path",
json={"key1": "value2"})
self.assertEquals(result.status_code, 404)
result = requests.get(MOCK_SERVER_URL + "/path",
json={
"key1": "value2",
"key": "value"
})
self.assertEquals(result.status_code, 200)
def test_expect_with_ttl(self):
self.client.expect(request(), response(), times(1), seconds(10))
result = requests.get(MOCK_SERVER_URL + "/path")
self.assertEqual(result.status_code, 200)
def test_verify_request_received_once(self):
self.client.stub(request(), response())
requests.get(MOCK_SERVER_URL)
self.client.verify(request(), times(1))
def test_verify_all_expectations(self):
self.client.expect(request(), response(), times(1))
requests.get(MOCK_SERVER_URL)
self.client.verify_expectations()
def test_verify_request_not_received_fail(self):
with self.assertRaises(AssertionError):
self.client.verify(request(), times(1))
def test_verify_request_never_received(self):
self.client.verify(request(), times(0))
def test_reset_should_clear_expectations(self):
self.client.expect(request(), response(), times(1))
self.client.reset()
self.client.verify_expectations()
