def test_restful_fail(monkeypatch):
"""
Test that failures in call_rest_api() result in HTTPError exception.
This is done only for the PUT HTTP verb.
"""
class MockResponse:
def p(self):
raise HTTPError("foo")
def __init__(self):
self.status_code = 400
self.raise_for_status = self.p
def mock_response(uri, headers, data, params, proxies, timeout):
return MockResponse()
with monkeypatch.context() as m:
m.setattr("requests.put", mock_response)
with pytest.raises(HTTPError):
call_rest_api(
ApiCall({
"uri": 'http://foo',
"method": 'PUT',
"data": 'data'
}))
def test_content_type(monkeypatch):
"""
Test HTTP Content-type header handling.
"""
for method in ["PUT", "POST", "DELETE"]:
text_plain_header = {CONTENT_TYPE: 'text/plain'}
for header_arg in [text_plain_header, None]:
call = {
"uri": "http://localhost:8080/source/api/v1/foo",
"method": method,
"data": "data",
"headers": header_arg
}
def mock_response(verb, uri, **kwargs):
headers = kwargs['headers']
if header_arg:
assert text_plain_header.items() <= headers.items()
else:
assert {CONTENT_TYPE: APPLICATION_JSON}.items() \
<= headers.items()
with monkeypatch.context() as m:
m.setattr("opengrok_tools.utils.restful.do_api_call",
mock_response)
call_rest_api(ApiCall(call))
def test_replacement(monkeypatch):
"""
Test replacement performed both in the URL and data.
"""
okay_status = 200
class MockResponse:
def p(self):
pass
def __init__(self):
self.status_code = okay_status
self.raise_for_status = self.p
def mock_do_api_call(verb, uri, **kwargs):
# Spying on mocked function is maybe too much so verify
# the arguments here.
assert uri == "http://localhost:8080/source/api/v1/BAR"
assert kwargs['data'] == '"fooBARbar"'
return MockResponse()
for verb in ["PUT", "POST", "DELETE"]:
call = {
"uri": "http://localhost:8080/source/api/v1/%FOO%",
"method": verb,
"data": "foo%FOO%bar"
}
pattern = "%FOO%"
value = "BAR"
with monkeypatch.context() as m:
m.setattr("opengrok_tools.utils.restful.do_api_call",
mock_do_api_call)
assert call_rest_api(ApiCall(call), {pattern: value}). \
status_code == okay_status
def test_headers_timeout(monkeypatch):
"""
Test that HTTP headers from command specification are united with
HTTP headers passed to call_res_api(). Also test timeout.
:param monkeypatch: monkey fixture
"""
headers = {'Tatsuo': 'Yasuko'}
expected_timeout = 42
expected_api_timeout = 24
call = {
"uri": "http://localhost:8080/source/api/v1/bar",
"method": "GET",
"data": "data",
"headers": headers
}
extra_headers = {'Mei': 'Totoro'}
def mock_do_api_call(verb, uri, **kwargs):
all_headers = headers
all_headers.update(extra_headers)
assert headers == all_headers
assert kwargs['timeout'] == expected_timeout
assert kwargs['api_timeout'] == expected_api_timeout
with monkeypatch.context() as m:
m.setattr("opengrok_tools.utils.restful.do_api_call", mock_do_api_call)
call_rest_api(ApiCall(call),
http_headers=extra_headers,
timeout=expected_timeout,
api_timeout=expected_api_timeout)
def test_invalid_command_bad_uri():
with pytest.raises(Exception):
call_rest_api(
ApiCall({
"uri": "foo",
"method": "PUT",
"data": "data",
"headers": "headers"
}))
def test_unknown_method():
call = {
"uri": "http://localhost:8080/source/api/v1/foo",
"method": "FOOBAR",
"data": "data"
}
pattern = "%FOO%"
value = "BAR"
with pytest.raises(Exception):
call_rest_api(ApiCall(call), {pattern: value})
def test_api_call_timeout_override(monkeypatch):
"""
Test that ApiCall object timeouts override timeouts passed as call_rest_api() arguments.
"""
expected_timeout = 42
expected_api_timeout = 24
call = {
"uri": "http://localhost:8080/source/api/v1/bar",
"method": "POST",
"data": "data",
"api_timeout": expected_timeout,
"async_api_timeout": expected_api_timeout
}
def mock_do_api_call(verb, uri, **kwargs):
assert kwargs['timeout'] == expected_timeout
assert kwargs['api_timeout'] == expected_api_timeout
with monkeypatch.context() as m:
m.setattr("opengrok_tools.utils.restful.do_api_call", mock_do_api_call)
call_rest_api(ApiCall(call),
timeout=expected_timeout + 1,
api_timeout=expected_api_timeout + 1)
def test_invalid_command_list():
with pytest.raises(Exception):
call_rest_api(ApiCall(["foo", "bar"]))
def test_invalid_command_uknown_key():
with pytest.raises(Exception):
call_rest_api(ApiCall({"foo": "bar"}))