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})