def test_should_raise_value_error(self):
HTTPretty.register_uri(HTTPretty.PATCH, self.url,
body=self.json_payload,
content_type="application/json")
with self.assertRaises(ValueError):
http_request(self.url, payload=self.json_payload,
http_verb='PATCH')
def on_get(remaining_url):
# reject GET requests for search until token added to GET in Kibana
if "_search" in remaining_url.lower():
return "Unauthorized", 401
try:
response = http_request(
url=_build_url(remaining_url, None), http_verb="GET", request_timeout=_CONFIG.elasticsearch.timeout
)
return response.content
except Exception as ex:
_LOG.exception(ex)
return "", 500
def on_get(remaining_url):
#reject GET requests for search until token added to GET in Kibana
if "_search" in remaining_url.lower():
return 'Unauthorized', 401
try:
response = http_request(url=_build_url(remaining_url, None),
http_verb='GET',
request_timeout=_CONFIG.elasticsearch.timeout)
return response.content
except Exception as ex:
_LOG.exception(ex)
return '', 500
def test_should_return_http_200_on_all_http_verbs(self):
httpretty_verbs = {
'POST': HTTPretty.POST,
'GET': HTTPretty.GET
}
for http_verb in HTTP_VERBS:
HTTPretty.register_uri(httpretty_verbs[http_verb],
self.url,
body=self.json_payload,
content_type="application/json",
status=200)
self.assertTrue(http_request(self.url,
payload=self.json_payload,
http_verb=http_verb), 200)
def on_post(remaining_url):
try:
token = request.headers.get(X_AUTH_TOKEN)
tenant_name = request.headers.get(X_TENANT_NAME)
if token is None or tenant_name is None:
return "Unauthorized", 401
token_is_valid, tenant_id = _token_is_valid(token, tenant_name)
if token_is_valid:
response = http_request(
url=_build_url(remaining_url, tenant_id),
payload=request.data,
http_verb='POST',
request_timeout=_CONFIG.elasticsearch.timeout)
return response.content
except Exception as ex:
_LOG.exception(ex)
return '', 500
return 'Unauthorized', 401
def on_post(remaining_url):
try:
token = request.headers.get(X_AUTH_TOKEN)
tenant_name = request.headers.get(X_TENANT_NAME)
if token is None or tenant_name is None:
return "Unauthorized", 401
token_is_valid, tenant_id = _token_is_valid(token, tenant_name)
if token_is_valid:
response = http_request(
url=_build_url(remaining_url, tenant_id),
payload=request.data,
http_verb="POST",
request_timeout=_CONFIG.elasticsearch.timeout,
)
return response.content
except Exception as ex:
_LOG.exception(ex)
return "", 500
return "Unauthorized", 401
def test_should_cause_a_request_exception(self):
with patch.object(requests, 'get') as mock_method:
with self.assertRaises(requests.RequestException):
mock_method.side_effect = requests.RequestException
http_request(self.url, payload=self.json_payload)
