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_expect_atleast_once(self): self.client.expect(request(), response(), timesrange(atleast=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, 200) self.client.verify_expectations()
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_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_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_expect_atmost_once_fails(self): self.client.expect(request(), response(), timesrange(atmost=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) with self.assertRaises(AssertionError): self.client.verify_expectations()
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_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_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_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_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_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_http_requests(self): _client = mk.MockServerClient("http://localhost:1080") _client.expect(mk.request("GET", "/user"), mk.response(200, "I'm here"), mk.times(2))
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_code_response(self): self.client.stub(request(), response(code=418)) result = requests.get(MOCK_SERVER_URL) self.assertEqual(result.status_code, 418)
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_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_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_reset_should_clear_expectations(self): self.client.expect(request(), response(), times(1)) self.client.reset() self.client.verify_expectations()
def test_expect_once_not_called_fails(self): self.client.expect(request(), response(), times(1)) with self.assertRaises(AssertionError): self.client.verify_expectations()
def test_expect_never(self): self.client.expect(request(), response(), times(0)) self.client.verify_expectations()
def test_cookies_response(self): self.client.stub(request(), response(cookies={"i-am-cookie": "sweet-cookie"})) result = requests.get(MOCK_SERVER_URL) self.assertEqual(result.cookies["i-am-cookie"], "sweet-cookie")