def test_http_request_error_code_cases(mox, response, exception_caught,
prefix):
"""Test the cases in which http_request throws an exception as a result of receiving an error status."""
creds = Credentials({'url': 'https://example.com', 'token': 'ABCDEFGH'})
conn = Connection(creds)
mox.StubOutWithMock(conn.session, 'request')
conn.session.request('GET',
'https://example.com/path',
headers=IgnoreArg(),
verify=True,
proxies=conn.proxies,
timeout=conn._timeout).AndReturn(response)
mox.ReplayAll()
with pytest.raises(exception_caught) as excinfo:
conn.http_request('get', '/path')
assert excinfo.value.message.startswith(prefix)
mox.VerifyAll()
def test_http_request_exception_cases(mox, exception_raised, exception_caught,
prefix):
"""Test the cases in which the underlying session object throws an exception as a result of a request."""
creds = Credentials({'url': 'https://example.com', 'token': 'ABCDEFGH'})
conn = Connection(creds)
mox.StubOutWithMock(conn.session, 'request')
conn.session.request('GET',
'https://example.com/path',
headers=IgnoreArg(),
verify=True,
proxies=conn.proxies,
timeout=conn._timeout).AndRaise(exception_raised)
mox.ReplayAll()
with pytest.raises(exception_caught) as excinfo:
conn.http_request('get', '/path')
if prefix:
assert excinfo.value.message.startswith(prefix)
mox.VerifyAll()
def test_session_ignore_system_proxy():
"""Test to make sure the ignore system proxy parameter has the right effect."""
creds = Credentials({
'url': 'https://example.com',
'token': 'ABCDEFGH',
'ignore_system_proxy': True
})
conn = Connection(creds)
assert conn.proxies['http'] == ''
assert conn.proxies['https'] == ''
assert conn.proxies['no'] == 'pass'
def test_session_cert_file_and_proxies():
"""Test to make sure the certificate file and proxy parameters get stashed in the right place."""
creds = Credentials({
'url': 'https://example.com',
'token': 'ABCDEFGH',
'ssl_cert_file': 'blort',
'proxy': 'foobie.bletch.com'
})
conn = Connection(creds)
assert conn.ssl_verify == 'blort'
assert conn.proxies['http'] == 'foobie.bletch.com'
assert conn.proxies['https'] == 'foobie.bletch.com'
def test_http_request_happy_path(mox):
"""Test the happy-path case of http_request."""
def validate_headers(hdrs):
assert hdrs['X-Auth-Token'] == 'ABCDEFGH'
assert hdrs['User-Agent'].startswith("CBC_SDK/")
assert hdrs['X-Test'] == 'yes'
return True
creds = Credentials({'url': 'https://example.com', 'token': 'ABCDEFGH'})
conn = Connection(creds)
mox.StubOutWithMock(conn.session, 'request')
conn.session.request('GET',
'https://example.com/path',
headers=Func(validate_headers),
verify=True,
proxies=conn.proxies,
timeout=conn._timeout).AndReturn(
StubResponse({'ok': True}))
mox.ReplayAll()
resp = conn.http_request('get', '/path', headers={'X-Test': 'yes'})
assert resp.json()['ok']
mox.VerifyAll()
def test_session_adapter_creation_failure(mox, adapter_raises, msg_prefix):
"""Test the failure cases that come from the failure to create the session adapter."""
creds = Credentials({'url': 'https://example.com', 'token': 'ABCDEFGH'})
import cbc_sdk.connection
mox.StubOutWithMock(cbc_sdk.connection,
'CBCSDKSessionAdapter',
use_mock_anything=True)
cbc_sdk.connection.CBCSDKSessionAdapter(force_tls_1_2=False, max_retries=IgnoreArg(), verify_hostname=True)\
.AndRaise(adapter_raises)
mox.ReplayAll()
with pytest.raises(ApiError) as excinfo:
Connection(creds)
assert excinfo.value.message.startswith(msg_prefix)
mox.VerifyAll()
def test_session_custom_session():
"""Test to make sure custom session is passed"""
import requests
session = requests.Session()
creds = Credentials({
'url': 'https://example.com',
'token': 'ABCDEFGH',
'ssl_cert_file': 'blort',
'proxy': None
})
session.proxies = {
'http': 'foobie.bletch.com',
'https': 'foobie.bletch.com'
}
conn = Connection(creds, proxy_session=session)
assert conn.ssl_verify == 'blort'
assert conn.proxies['http'] == 'foobie.bletch.com'
assert conn.proxies['https'] == 'foobie.bletch.com'
def test_request_helper_methods(mox):
"""Test the four helper methods of http_request."""
creds = Credentials({'url': 'https://example.com', 'token': 'ABCDEFGH'})
conn = Connection(creds)
mox.StubOutWithMock(conn.session, 'request')
conn.session.request('GET',
'https://example.com/getpath',
headers=IgnoreArg(),
verify=True,
proxies=conn.proxies,
timeout=conn._timeout).AndReturn(
StubResponse({'get': True}))
conn.session.request('POST',
'https://example.com/postpath',
headers=IgnoreArg(),
verify=True,
proxies=conn.proxies,
timeout=conn._timeout).AndReturn(
StubResponse({'post': True}))
conn.session.request('PUT',
'https://example.com/putpath',
headers=IgnoreArg(),
verify=True,
proxies=conn.proxies,
timeout=conn._timeout).AndReturn(
StubResponse({'put': True}))
conn.session.request('DELETE',
'https://example.com/delpath',
headers=IgnoreArg(),
verify=True,
proxies=conn.proxies,
timeout=conn._timeout).AndReturn(
StubResponse({'delete': True}))
mox.ReplayAll()
resp = conn.get('/getpath')
assert resp.json()['get']
resp = conn.post('/postpath')
assert resp.json()['post']
resp = conn.put('/putpath')
assert resp.json()['put']
resp = conn.delete('/delpath')
assert resp.json()['delete']
mox.VerifyAll()
def test_initial_connection_error(cdata, msg_prefix):
"""Test the initial raising of a ConnectionError by the connection."""
creds = Credentials(cdata)
with pytest.raises(ConnectionError) as excinfo:
Connection(creds)
assert excinfo.value.message.startswith(msg_prefix)
