def test_find_suppliers_filtered_by_supplier_ids(mock_data_client): mock_data_client.get_interested_suppliers.return_value = { 'interestedSuppliers': [4, 3, 2] } records = list(export_g8_suppliers.find_suppliers(mock_data_client, 'framework-slug', supplier_ids=[2, 4])) assert records == [ {'supplier_id': 4}, {'supplier_id': 2}, ]
def test_find_suppliers_produces_results_with_supplier_ids(mock_data_client): mock_data_client.get_interested_suppliers.return_value = { 'interestedSuppliers': [4, 3, 2] } records = list(export_g8_suppliers.find_suppliers(mock_data_client, 'framework-slug')) mock_data_client.get_interested_suppliers.assert_has_calls([ call('framework-slug') ]) assert records == [ {'supplier_id': 4}, {'supplier_id': 3}, {'supplier_id': 2} ]