Merge pull request #94 from opendatadiscovery/feature/api_oddrn_gener…

…ator

Adds ApiGenerator

oddrn_generator/__init__.py

@@ -59,6 +59,7 @@
     BigQueryStorageGenerator,
     CKANGenerator,
     AzureDataFactoryGenerator,
+    ApiGenerator,
 )
 
 __all__ = [
@@ -122,4 +123,5 @@
     "BigQueryStorageGenerator",
     "CKANGenerator",
     "AzureDataFactoryGenerator",
+    "ApiGenerator",
 ]

oddrn_generator/generators.py

@@ -60,6 +60,7 @@
     BigQueryStoragePathsModel,
     CKANPathsModel,
     AzureDataFactoryPathsModel,
+    ApiPathsModel,
 )
 from oddrn_generator.server_models import (
     AbstractServerModel,
@@ -547,3 +548,9 @@ class AzureDataFactoryGenerator(Generator):
     source = "azure_data_factory"
     paths_model = AzureDataFactoryPathsModel
     server_model = AzureDomainCloudModel
+
+
+class ApiGenerator(Generator):
+    source = "api"
+    paths_model = ApiPathsModel
+    server_model = HostnameModel

oddrn_generator/path_models.py

@@ -911,3 +911,14 @@ class Config:
                 "activities_runs",
             ),
         }
+
+
+class ApiPathsModel(BasePathsModel):
+    resources: Optional[str]
+    fields: Optional[str]
+
+    class Config:
+        dependencies_map = {
+            "resources": ("resources",),
+            "fields": ("resources", "fields"),
+        }

tests/test_api_generator.py

@@ -0,0 +1,30 @@
+from oddrn_generator.generators import ApiGenerator
+
+
+def test_api_generator():
+    test_host_name = "www.test_host.com"
+    test_resource_name = "test_resource"
+    test_field_name = "test_field"
+
+    generator = ApiGenerator(
+        host_settings=test_host_name,
+        resources=test_resource_name,
+        fields=test_field_name
+    )
+    assert generator.get_data_source_oddrn() == f"//api/host/{test_host_name}"
+    assert (
+        generator.get_oddrn_by_path("resources")
+        == f"//api/host/{test_host_name}/resources/{test_resource_name}"
+    )
+    assert (
+        generator.get_oddrn_by_path("fields")
+        == f"//api/host/{test_host_name}/resources/{test_resource_name}/fields/{test_field_name}"
+    )
+
+    test_resource_name_2 = "test_resource_2"
+    test_field_name_2 = "test_field_2"
+    generator.set_oddrn_paths(resources=test_resource_name_2, fields=test_field_name_2)
+    assert (
+        generator.get_oddrn_by_path("fields")
+        == f"//api/host/{test_host_name}/resources/{test_resource_name_2}/fields/{test_field_name_2}"
+    )