コード例 #1
0
def test_seminar_series_aggregation_and_filter(inspire_app, override_config):
    config = {
        "RECORDS_REST_FACETS": {
            "records-seminars": {
                "filters": {**current_app.config["SEMINARS_FILTERS"]},
                "aggs": {**seminar_series_aggregation(1)},
            }
        }
    }

    with override_config(**config):
        data = {"series": [{"name": "Series1"}]}
        expected_record = create_record("sem", data)
        data = {"series": [{"name": "ICHEP"}]}
        create_record("sem", data)
        with inspire_app.test_client() as client:
            response = client.get("/seminars/facets").json
        expected_aggregation = {
            "meta": {"title": "Series", "type": "checkbox", "order": 1},
            "doc_count_error_upper_bound": 0,
            "sum_other_doc_count": 0,
            "buckets": [
                {"key": "ICHEP", "doc_count": 1},
                {"key": "Series1", "doc_count": 1},
            ],
        }
        assert response["aggregations"]["series"] == expected_aggregation

        with inspire_app.test_client() as client:
            response = client.get("/seminars?series=Series1").json
        assert len(response["hits"]["hits"]) == 1
        assert (
            response["hits"]["hits"][0]["metadata"]["control_number"]
            == expected_record["control_number"]
        )
コード例 #2
0
ファイル: facets.py プロジェクト: MJedr/inspirehep
def records_seminars(order=None):
    if order is None:
        order = count(start=1)
    return {
        "filters": seminars_filters(),
        "aggs": {
            **seminar_series_aggregation(order=next(order)),
            **seminar_subject_aggregation(order=next(order)),
        },
    }
コード例 #3
0
ファイル: test_facets.py プロジェクト: MJedr/inspirehep
def test_records_seminars_facets(inspire_app):
    with current_app.test_request_context():
        expected_filters = {"subject", "series", "start_date"}
        expected_aggregations = {
            **seminar_series_aggregation(order=1),
            **seminar_subject_aggregation(order=2),
        }

        filters = current_app.config["RECORDS_REST_FACETS"][
            "records-seminars"]()["filters"].keys()
        aggregations = current_app.config["RECORDS_REST_FACETS"][
            "records-seminars"]()["aggs"]
        assert filters == expected_filters
        assert aggregations == expected_aggregations