def get_account_summary(context):
""" Test calling the RetirementServices.get_account_summary method """
with mock_benefits_requests(context):
try:
retirement = context.benefits_online.retirement_services
assert retirement.get_account_summary()
except PychexUnauthenticatedError:
context.exceptions.append(sys.exc_info()[1])
def _login(context, logging_into, password=None, http_status=200):
""" Call the login method of the specified type of object """
with mock_login_requests(context):
with mock_benefits_requests(context, http_status):
try:
if logging_into == 'paychex':
assert context.paychex.login(password)
elif logging_into == 'benefits_online':
context.benefits_online.password = password
assert context.benefits_online.login()
else:
assert context.benefits_online.retirement_services.login()
except (PychexInvalidPasswordError, PychexUnknownError,
PychexSecurityImageMissingError):
context.exceptions.append(sys.exc_info()[1])
def run_account_summary(context, json=False):
# Make sure the password check passes in the paychex_login mock
context.paychex = MagicMock()
context.paychex.password = context.password
# Reset the stdout capture
context.stdout_capture.truncate(0)
context.stdout_capture.seek(0)
with mock_login_requests(context):
mock_obj = ContextXmlMock(context)
with mock_request(context, mock_obj.paychex_account_data):
with mock_benefits_requests(context):
arguments = {
'authorize': False,
'account_summary': True,
'--config': context.config_file,
'--json': json
}
try:
PychexCli(arguments)
except Exception:
assert False, 'Exception: %s' % sys.exc_info()[1]
