Ejemplos de find_suppliers en Python

Ejemplo n.º 1

Archivo: test_export_g8_suppliers.py Proyecto: alphagov/digitalmarketplace-scripts

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},
    ]

Ejemplo n.º 2

Archivo: test_export_g8_suppliers.py Proyecto: alphagov/digitalmarketplace-scripts

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