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)) framework_info_adder = export_g8_suppliers.add_framework_info(mock_data_client, 'framework-slug') with pytest.raises(HTTPError): framework_info_adder({'supplier_id': 1})
def test_add_framework_info_fails_if_incomplete_declaration_is_not_failed(mock_data_client): mock_data_client.get_supplier_framework_info.return_value = { 'frameworkInterest': {'declaration': {'status': 'incomplete'}, 'onFramework': None} } framework_info_adder = export_g8_suppliers.add_framework_info(mock_data_client, 'framework-slug') with pytest.raises(AssertionError): framework_info_adder({'supplier_id': 1})
def test_add_framework_info(mock_data_client): mock_data_client.get_supplier_framework_info.side_effect = [ {'frameworkInterest': {'declaration': {'status': 'complete'}, 'onFramework': True}}, {'frameworkInterest': {'declaration': {'status': 'complete'}, 'onFramework': False}}, ] framework_info_adder = export_g8_suppliers.add_framework_info(mock_data_client, 'framework-slug') records = [ framework_info_adder({'supplier_id': 1}), framework_info_adder({'supplier_id': 2}), ] mock_data_client.get_supplier_framework_info.assert_has_calls([ call(1, 'framework-slug'), call(2, 'framework-slug') ]) assert records == [ {'supplier_id': 1, 'declaration': {'status': 'complete'}, 'onFramework': True}, {'supplier_id': 2, 'declaration': {'status': 'complete'}, 'onFramework': False}, ]