# Copyright (C) 2022 The Software Heritage developers
# See the AUTHORS file at the top-level directory of this distribution
# License: GNU General Public License version 3, or any later version
# See top-level LICENSE file for more information


from ariadne import graphql_sync
from flask import Flask, jsonify, request
import pytest

from swh.graphql import server as app_server
from swh.graphql.app import schema
from swh.graphql.errors import format_error
from swh.search import get_search as get_swh_search
from swh.storage import get_storage as get_swh_storage

from .data import populate_dummy_data, populate_search_data


@pytest.fixture(scope="session")
def storage():
    storage = get_swh_storage(cls="memory")
    # set the global var to use the in-memory storage
    app_server.storage = storage
    # populate the in-memory storage
    populate_dummy_data(storage)
    return storage


@pytest.fixture(scope="session")
def search():
    search = get_swh_search("memory")
    # set the global var to use the in-memory search
    app_server.search = search
    search.initialize()
    # populate the in-memory search
    populate_search_data(search)
    return search


@pytest.fixture(scope="session")
def test_app(storage, search):
    app = Flask(__name__)

    @app.route("/", methods=["POST"])
    def graphql_server():
        # GraphQL queries are always sent as POST
        data = request.get_json()
        success, result = graphql_sync(
            schema,
            data,
            context_value=request,
            debug=app.debug,
            error_formatter=format_error,
        )
        status_code = 200 if success else 400
        return jsonify(result), status_code

    yield app


@pytest.fixture(scope="session")
def client(test_app):
    with test_app.test_client() as client:
        yield client