def test_recover_from_expired_token(aggregator):
# First api answers with 403 to force the check to re-authenticate
unauthentified_response = MagicMock(status_code=403)
# Api answer when a request is being made to the login endpoint
login_response = MagicMock()
# Third api answer, when the check retries the initial endpoint but is now authenticated
valid_response = MagicMock()
valid_response.json = MagicMock(return_value={"foo": "bar"})
http = MagicMock()
http.post = MagicMock(side_effect=[login_response])
http.get = MagicMock(side_effect=[unauthentified_response, valid_response])
session_wrapper = SessionWrapper(aci_url=common.ACI_URL,
http=http,
log=MagicMock())
session_wrapper.apic_cookie = "cookie"
api = Api(common.ACI_URLS, http, common.USERNAME, password=common.PASSWORD)
api.sessions = [session_wrapper]
data = api.make_request("")
# Assert that we retrieved the value from `valid_response.json()`
assert data == {"foo": "bar"}
get_calls = http.get._mock_call_args_list
post_calls = http.post._mock_call_args_list
# Assert that the first call was to the ACI_URL
assert get_calls[0].args[0] == common.ACI_URL
# Assert that the second call was to the login endpoint
assert 'aaaLogin.xml' in post_calls[0].args[0]
# Assert that the last call was to the ACI_URL again
assert get_calls[1].args[0] == common.ACI_URL
def test_cisco(aggregator, session_mock):
cisco_aci_check = CiscoACICheck(CHECK_NAME, {}, {})
api = Api(ACI_URLS, USERNAME, PASSWORD, log=cisco_aci_check.log)
api.sessions = [session_mock]
api._refresh_sessions = False
cisco_aci_check._api_cache[hash_mutable(CONFIG)] = api
cisco_aci_check.check(CONFIG)
def test_recover_from_expired_token(aggregator, api_kwargs):
# First api answers with 403 to force the check to re-authenticate
unauthentified_response = MagicMock(status_code=403)
# Api answer when a request is being made to the login endpoint
login_response = MagicMock()
# Third api answer, when the check retries the initial endpoint but is now authenticated
valid_response = MagicMock()
valid_response.json = MagicMock(return_value={"foo": "bar"})
http = MagicMock()
http.post = MagicMock(side_effect=[login_response])
http.get = MagicMock(side_effect=[unauthentified_response, valid_response])
session_wrapper = SessionWrapper(aci_url=common.ACI_URL,
http=http,
log=MagicMock())
session_wrapper.apic_cookie = "cookie"
api = Api(common.ACI_URLS, http, common.USERNAME, **api_kwargs)
api.sessions = {common.ACI_URL: session_wrapper}
data = api.make_request("")
# Assert that we retrieved the value from `valid_response.json()`
assert data == {"foo": "bar"}
get_calls = http.get._mock_call_args_list
post_calls = http.post._mock_call_args_list
# Assert that the first call was to the ACI_URL
assert get_calls[0].args[0] == common.ACI_URL
if 'password' in api_kwargs:
# Assert that the second call was to the login endpoint
assert 'aaaLogin.xml' in post_calls[0].args[0]
# Assert that the last call was to the ACI_URL again
assert get_calls[1].args[0] == common.ACI_URL
# Assert session correctly renewed
assert len(api.sessions) == 1 # check the number of sessions doesn't grow
assert api.sessions[
common.ACI_URL] != session_wrapper # check session renewed
# Assert cookie to check the session changed
assert get_calls[0].kwargs['headers']['Cookie'] == 'cookie'
assert get_calls[1].kwargs['headers']['Cookie'] != 'cookie'
login_response = MagicMock()
# Third api answer, when the check retries the initial endpoint but is now authenticated
valid_response = MagicMock()
valid_response.json = MagicMock(return_value={"foo": "bar"})
http = MagicMock()
http.post = MagicMock(side_effect=[login_response])
http.get = MagicMock(side_effect=[unauthentified_response, valid_response])
session_wrapper = SessionWrapper(aci_url=common.ACI_URL,
http=http,
log=MagicMock())
session_wrapper.apic_cookie = "cookie"
api = Api(common.ACI_URLS, http, common.USERNAME, **api_kwargs)
api.sessions = {common.ACI_URL: session_wrapper}
data = api.make_request("")
# Assert that we retrieved the value from `valid_response.json()`
assert data == {"foo": "bar"}
get_calls = http.get._mock_call_args_list
post_calls = http.post._mock_call_args_list
# Assert that the first call was to the ACI_URL
assert get_calls[0].args[0] == common.ACI_URL
if 'password' in api_kwargs:
# Assert that the second call was to the login endpoint
assert 'aaaLogin.xml' in post_calls[0].args[0]
