def test_get_report_jobs(mocker):
id_1 = 'RJW55950MF0S0HKN'
mock_response = {'results': (
ReportJob.from_dict({'id': id_1, 'positionSourceType': 'Portfolio', 'positionSourceId': 'MP1',
'parameters': {'transactionCostModel': 'FIXED'}}),
ReportJob.from_dict({'id': id_1, 'positionSourceType': 'Portfolio', 'positionSourceId': 'MP2',
'parameters': {'transactionCostModel': 'FIXED'}})
), 'totalResults': 2}
expected_response = (
ReportJob(id=id_1, positionSourceType='Portfolio', positionSourceId='MP1',
parameters=ReportParameters(transactionCostModel='FIXED')),
ReportJob(id=id_1, positionSourceType='Portfolio', positionSourceId='MP2',
parameters=ReportParameters(transactionCostModel='FIXED'))
)
# mock GsSession
mocker.patch.object(
GsSession.__class__,
'default_value',
return_value=GsSession.get(
Environment.QA,
'client_id',
'secret'))
mocker.patch.object(GsSession.current, '_get', return_value=mock_response)
# run test
response = GsReportApi.get_report_jobs(id_1)
GsSession.current._get.assert_called_with('/reports/{id}/jobs'.format(id=id_1))
assert response == expected_response
def test_report_job(mocker):
id_1 = 'RX1'
mock_response = ReportJob(id=id_1, positionSourceType='Portfolio', positionSourceId='MP1',
parameters=ReportParameters(transactionCostModel='FIXED'))
# mock GsSession
mocker.patch.object(
GsSession.__class__,
'default_value',
return_value=GsSession.get(
Environment.QA,
'client_id',
'secret'))
mocker.patch.object(GsSession.current, '_get', return_value=mock_response)
# run test
response = GsReportApi.get_report_job(id_1)
GsSession.current._get.assert_called_with('/reports/jobs/{id}'.format(id=id_1))
assert response == mock_response