def test_logoff(self):
"""Test Session.logoff() (and also Session.is_logon())."""
with requests_mock.Mocker() as m:
self.mock_server_1(m)
# Create a session in logged-off state
session = Session('fake-host', 'fake-userid', 'fake-pw')
session.logon()
logged_on = session.is_logon()
assert logged_on
# The code to be tested:
session.logoff()
assert session.session_id is None
assert session.session is None
assert 'X-API-Session' not in session.headers
assert len(session.headers) == 2
logged_on = session.is_logon()
assert not logged_on
def test_logon(
self, host, userid, password, use_get_password, exp_exc):
"""Test Session.logon() (and also Session.is_logon())."""
with requests_mock.Mocker() as m:
self.mock_server_1(m)
if use_get_password:
get_password = mock.MagicMock()
get_password.return_value = \
'fake-pw-{}-{}'.format(host, userid)
else:
get_password = None
# Create a session in logged-off state
session = Session(host, userid, password, None, get_password)
assert session.session_id is None
assert 'X-API-Session' not in session.headers
assert session.session is None
logged_on = session.is_logon()
assert not logged_on
if exp_exc:
try:
# The code to be tested:
session.logon()
except exp_exc:
pass
logged_on = session.is_logon()
assert not logged_on
else:
# The code to be tested:
session.logon()
assert session.session_id == 'fake-session-id'
assert 'X-API-Session' in session.headers
assert isinstance(session.session, requests.Session)
if get_password:
if password is None:
get_password.assert_called_with(host, userid)
assert session._password == get_password.return_value
else:
get_password.assert_not_called()
logged_on = session.is_logon()
assert logged_on