def test_add_framework_info(mock_data_client, on_framework):
mock_data_client.get_supplier_framework_info.return_value = {
'frameworkInterest': {
'declaration': {'status': 'complete'},
'onFramework': on_framework,
'countersignedPath': None,
'countersignedAt': None,
}
}
record = export_framework_applicant_details.add_framework_info(mock_data_client, 'g-cloud-8', {'supplier_id': 1})
assert mock_data_client.get_supplier_framework_info.call_args == ((1, 'g-cloud-8',),)
assert record == {
'supplier_id': 1,
'onFramework': on_framework,
'declaration': {
'status': 'complete',
},
'countersignedPath': '',
'countersignedAt': '',
}
def test_add_framework_info_fails_on_non_404_error(mock_data_client):
mock_data_client.get_supplier_framework_info.side_effect = HTTPError(Mock(status_code=400))
with pytest.raises(HTTPError):
export_framework_applicant_details.add_framework_info(mock_data_client, 'g-cloud-8', {'supplier_id': 1})
