updated tests

Author: Mihai10Mihai

ted_sws/data_manager/adapters/sparql_endpoint.py

@@ -104,6 +104,9 @@ def fetch_rdf(self) -> rdflib.Graph:
         :return:
         """
 
+    def add_data_to_repository(self, file_content, repository_name, mime_type):
+        pass
+
 
 class SPARQLTripleStoreEndpoint(TripleStoreEndpointABC):
 

ted_sws/data_manager/adapters/triple_store.py

@@ -116,16 +116,17 @@ def list_repositories(self):
         """
         return self.allegro.openCatalog(name=self.catalog_name).listRepositories()
 
-    def add_data_to_repository(self, file_content: str, mime_type: str, repository_name: str = None):
+    def add_data_to_repository(self, file_content: Union[str, bytes, bytearray], mime_type: str, repository_name: str = None):
         """
         Method to add triples from a string
         :param file_content:
         :param repository_name:
         :param mime_type:
         :return:
         """
+
         repository = self._get_repository(repository_name=repository_name)
-        repository.getConnection().addData(data=file_content, rdf_format=mime_type)
+        repository.getConnection().addData(data=file_content)
 
     def add_file_to_repository(self, file_path, repository_name: str = None):
         """

ted_sws/notice_publisher_triple_store/services/load_transformed_notice_into_triple_store.py

@@ -2,7 +2,7 @@
     This module implements functionality to load a given notice into a triple store.
 """
 from ted_sws.data_manager.adapters.repository_abc import NoticeRepositoryABC
-from ted_sws.data_manager.adapters.triple_store import TripleStoreEndpointABC
+from ted_sws.data_manager.adapters.triple_store import TripleStoreEndpointABC, RDF_MIME_TYPES
 
 DEFAULT_NOTICE_REPOSITORY_NAME = "notices"
 
@@ -16,6 +16,6 @@ def load_notice_into_triple_store(notice_id: str, notice_repository: NoticeRepos
     notice = notice_repository.get(reference=notice_id)
     if notice is None:
         raise ValueError('Notice, with "%s" notice_id, was not found' % notice_id)
-
+    mime_type = RDF_MIME_TYPES
     rdf_manifestation_string = notice.rdf_manifestation.object_data
-    triple_store_repository.add_data_to_repository(file_content=rdf_manifestation_string, repository_name=repository_name)
+    triple_store_repository.add_data_to_repository(file_content=rdf_manifestation_string, repository_name=repository_name, mime_type=mime_type)

tests/e2e/mapping_suite_processor/conftest.py

@@ -2,8 +2,10 @@
 import pymongo
 import pytest
 
+from ted_sws.data_manager.adapters.triple_store import RDF_MIME_TYPES
 from tests import TEST_DATA_PATH
 
+
 @pytest.fixture
 @mongomock.patch(servers=(('server.example.com', 27017),))
 def fake_mongodb_client():
@@ -14,6 +16,7 @@ def fake_mongodb_client():
 def file_system_repository_path():
     return TEST_DATA_PATH / "notice_transformer" / "mapping_suite_processor_repository"
 
+
 @pytest.fixture
 def yarrrml_file_content():
     return """prefixes:
@@ -29,6 +32,7 @@ def yarrrml_file_content():
       - [ex:name, $(firstname)]
 """
 
+
 @pytest.fixture
 def rml_file_result():
     return """@prefix rr: <http://www.w3.org/ns/r2rml#>.
@@ -84,3 +88,6 @@ def package_folder_path():
     return TEST_DATA_PATH / "notice_validator" / "test_repository" / "test_package"
 
 
+@pytest.fixture
+def mime_type():
+    return RDF_MIME_TYPES

tests/e2e/mapping_suite_processor/test_allegro_triple_store.py

@@ -1,10 +1,10 @@
 
 
-def test_allegro_allegro_triple_store(ttl_file, path_ttl_file, allegro_triple_store):
+def test_allegro_triple_store(ttl_file, path_ttl_file, allegro_triple_store, mime_type):
     allegro_triple_store.create_repository(repository_name="testing")
     assert isinstance(allegro_triple_store.list_repositories(), list)
     assert "testing" in allegro_triple_store.list_repositories()
-    allegro_triple_store.add_data_to_repository(file_content=ttl_file, repository_name="testing")
+    allegro_triple_store.add_data_to_repository(file_content=ttl_file,mime_type=mime_type,repository_name="testing")
     allegro_triple_store.add_file_to_repository(file_path=path_ttl_file, repository_name="testing")
     assert allegro_triple_store._get_repository(repository_name="testing").getConnection().size() == 484
     allegro_triple_store.delete_repository(repository_name="testing")