about summary refs log tree commit diff
path: root/tests/unit/db/rdf/test_wiki.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/db/rdf/test_wiki.py')
-rw-r--r--tests/unit/db/rdf/test_wiki.py50
1 files changed, 50 insertions, 0 deletions
diff --git a/tests/unit/db/rdf/test_wiki.py b/tests/unit/db/rdf/test_wiki.py
index 3abf3ad..bab37ce 100644
--- a/tests/unit/db/rdf/test_wiki.py
+++ b/tests/unit/db/rdf/test_wiki.py
@@ -22,11 +22,15 @@ from tests.fixtures.rdf import (
     SPARQL_CONF,
 )
 
+from tests.unit.db.rdf.data import LPL_RIF_ENTRIES
+
 from gn3.db.rdf.wiki import (
     __sanitize_result,
     get_wiki_entries_by_symbol,
     get_comment_history,
     update_wiki_comment,
+    get_rif_entries_by_symbol,
+    delete_wiki_entries_by_id,
 )
 
 GRAPH = "<http://cd-test.genenetwork.org>"
@@ -396,3 +400,49 @@ def test_update_wiki_comment(rdf_setup):  # pylint: disable=W0613,W0621
         "version": 3,
         "web_url": "http://some-website.com",
     })
+
+
+@pytest.mark.rdf
+def test_get_rif_entries_by_symbol(rdf_setup):  # pylint: disable=W0613,W0621
+    """Test fetching NCBI Rif Metadata from RDF"""
+    sparql_conf = SPARQL_CONF
+    entries = get_rif_entries_by_symbol(
+        symbol="Lpl",
+        sparql_uri=sparql_conf["sparql_endpoint"],
+        graph=GRAPH,
+    )
+    assert len(LPL_RIF_ENTRIES["data"]) == len(entries["data"])
+    for result, expected in zip(LPL_RIF_ENTRIES["data"], entries["data"]):
+        TestCase().assertDictEqual(result, expected)
+
+
+@pytest.mark.rdf
+def test_delete_wiki_entries_by_id(rdf_setup):  # pylint: disable=W0613,W0621
+    """Test deleting a given RIF Wiki entry"""
+    sparql_conf = SPARQL_CONF
+    delete_wiki_entries_by_id(
+        230,
+        sparql_user=sparql_conf["sparql_user"],
+        sparql_password=sparql_conf["sparql_password"],
+        sparql_auth_uri=sparql_conf["sparql_auth_uri"],
+        graph=GRAPH)
+    entries = get_comment_history(
+        comment_id=230,
+        sparql_uri=sparql_conf["sparql_endpoint"],
+        graph=GRAPH,
+    )
+    assert len(entries["data"]) == 0
+
+    # Deleting a non-existent entry has no effect
+    delete_wiki_entries_by_id(
+        199999,
+        sparql_user=sparql_conf["sparql_user"],
+        sparql_password=sparql_conf["sparql_password"],
+        sparql_auth_uri=sparql_conf["sparql_auth_uri"],
+        graph=GRAPH)
+    entries = get_comment_history(
+        comment_id=230,
+        sparql_uri=sparql_conf["sparql_endpoint"],
+        graph=GRAPH,
+    )
+    assert len(entries["data"]) == 0