diff options
Diffstat (limited to 'tests/fixtures/rdf.py')
-rw-r--r-- | tests/fixtures/rdf.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/tests/fixtures/rdf.py b/tests/fixtures/rdf.py new file mode 100644 index 0000000..a5fd711 --- /dev/null +++ b/tests/fixtures/rdf.py @@ -0,0 +1,41 @@ +"""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("") + sparql_conf.from_envvar("GN3_SECRETS") + sparql_conf.from_envvar("GN3_CONF") + 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"]) + # 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) |