Changeset View
Changeset View
Standalone View
Standalone View
swh/graphql/tests/conftest.py
| # Copyright (C) 2022 The Software Heritage developers | # Copyright (C) 2022 The Software Heritage developers | ||||
| # See the AUTHORS file at the top-level directory of this distribution | # See the AUTHORS file at the top-level directory of this distribution | ||||
| # License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| from ariadne import graphql_sync | from ariadne import graphql_sync | ||||
| from flask import Flask, jsonify, request | from flask import Flask, jsonify, request | ||||
| import pytest | import pytest | ||||
| from swh.graphql import server as app_server | from swh.graphql import server as app_server | ||||
| from swh.graphql.app import schema | from swh.graphql.app import schema, validation_rules | ||||
| from swh.graphql.errors import format_error | from swh.graphql.errors import format_error | ||||
| from swh.search import get_search as get_swh_search | from swh.search import get_search as get_swh_search | ||||
| from swh.storage import get_storage as get_swh_storage | from swh.storage import get_storage as get_swh_storage | ||||
| from .data import populate_dummy_data, populate_search_data | from .data import populate_dummy_data, populate_search_data | ||||
| @pytest.fixture(scope="session") | @pytest.fixture(scope="session") | ||||
| Show All 12 Lines | def search(): | ||||
| # set the global var to use the in-memory search | # set the global var to use the in-memory search | ||||
| app_server.search = search | app_server.search = search | ||||
| search.initialize() | search.initialize() | ||||
| # populate the in-memory search | # populate the in-memory search | ||||
| populate_search_data(search) | populate_search_data(search) | ||||
| return search | return search | ||||
| @pytest.fixture(autouse=True) | |||||
| def config(): | |||||
| app_server.graphql_cfg = {"max_query_cost": {"anonymous": 100}} | |||||
| @pytest.fixture(scope="session") | @pytest.fixture(scope="session") | ||||
| def test_app(storage, search): | def test_app(storage, search): | ||||
| app = Flask(__name__) | app = Flask(__name__) | ||||
| @app.route("/", methods=["POST"]) | @app.route("/", methods=["POST"]) | ||||
| def graphql_server(): | def graphql_server(): | ||||
| # GraphQL queries are always sent as POST | # GraphQL queries are always sent as POST | ||||
| data = request.get_json() | data = request.get_json() | ||||
| success, result = graphql_sync( | success, result = graphql_sync( | ||||
| schema, | schema, | ||||
| data, | data, | ||||
| context_value=request, | context_value=request, | ||||
| debug=app.debug, | debug=app.debug, | ||||
| validation_rules=validation_rules, | |||||
| error_formatter=format_error, | error_formatter=format_error, | ||||
| ) | ) | ||||
| status_code = 200 if success else 400 | status_code = 200 if success else 400 | ||||
| return jsonify(result), status_code | return jsonify(result), status_code | ||||
| yield app | yield app | ||||
| @pytest.fixture(scope="session") | @pytest.fixture(scope="session") | ||||
| def client(test_app): | def client(test_app): | ||||
| with test_app.test_client() as client: | with test_app.test_client() as client: | ||||
| yield client | yield client | ||||