aboutsummaryrefslogtreecommitdiff
path: root/tests/fixtures/rdf.py
blob: 776a5d4ce54f11741dc2b6b16f9dc57248626d9f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
"""Test fixtures to set up a test named graph for loading RDF data."""
from requests.auth import HTTPDigestAuth
from flask import config
import os
import requests
import pytest


def get_sparql_auth_conf() -> dict:
    """Fetch SPARQL auth configuration for the GN3_SECRETS file."""
    sparql_conf = config.Config("")
    # When loading from the environment, GN3_CONF precedes
    # GN3_SECRETS.  Don't change this order.
    sparql_conf.from_envvar("GN3_CONF")
    sparql_conf.from_envvar("GN3_SECRETS")
    return {
        "sparql_user": sparql_conf["SPARQL_USER"],
        "sparql_auth_uri": sparql_conf["SPARQL_AUTH_URI"],
        "sparql_crud_auth_uri": sparql_conf["SPARQL_CRUD_AUTH_URI"],
        "sparql_endpoint": sparql_conf["SPARQL_ENDPOINT"],
        "sparql_password": sparql_conf["SPARQL_PASSWORD"],
    }


@pytest.fixture(scope="module")
def rdf_setup():
    """Upload RDF to a Virtuoso named graph"""
    # Define the URL and file
    sparql_conf = get_sparql_auth_conf()
    url = sparql_conf["sparql_crud_auth_uri"]
    file_path = os.path.join(
        os.path.dirname(__file__).split("fixtures")[0],
        "test_data/ttl-files/test-data.ttl",
    )

    # Define the query parameters and authentication
    params = {"graph": "http://cd-test.genenetwork.org"}
    auth = HTTPDigestAuth(
        sparql_conf["sparql_user"], sparql_conf["sparql_password"])

    # Make sure this graph does not exist before running anything
    requests.delete(url, params=params, auth=auth)

    # Open the file in binary mode and send the request
    with open(file_path, "rb") as file:
        response = requests.put(url, params=params, auth=auth, data=file)
    yield response
    requests.delete(url, params=params, auth=auth)