Python LiteratureRecord.serialize_for_es Examples

Example #1

def test_indexer_prepare_record(record_metadata_mock, current_app_mock,
                                receiver_mock):
    record = LiteratureRecord({})
    indexer = InspireRecordIndexer()

    # Assume that record methods was already tested
    expected = record.serialize_for_es()

    processed = indexer._prepare_record(record, "index_name", "document_type")
    assert receiver_mock.send.call_count == 1
    assert expected == processed

Example #2

File: test_base.py Project: tsgit/inspirehep

def test_indexer_prepare_record(
    mixins_current_app_mock,
    record_metadata_mock,
    current_app_mock,
    receiver_mock,
    query_mock,
    mock_latex_us_display,
    mock_latex_eu_display,
    mock_bibtex_display,
    mock_referenced_authors,
):
    query_mock.return_value.filter_by.return_value.count.return_value = 1
    query_mock.return_value.filter_by.return_value.filter.return_value.count.return_value = (
        1)
    record = LiteratureRecord({})
    indexer = InspireRecordIndexer()
    # Assume that record methods was already tested
    expected = record.serialize_for_es()

    processed = indexer._prepare_record(record, "index_name", "document_type")
    assert receiver_mock.send.call_count == 1
    assert expected == processed