Adding the legal service endpoint support

Author: MartiONE

epo_ops/api.py

@@ -28,6 +28,7 @@ class Client(object):
 
     __family_path__ = "family"
     __images_path__ = "published-data/images"
+    __legal_path__ = "legal"
     __number_path__ = "number-service"
     __published_data_path__ = "published-data"
     __published_data_search_path__ = "published-data/search"
@@ -119,6 +120,29 @@ def image(
         """
         return self._image_request(path, range, document_format)
 
+    def legal(
+        self,
+        reference_type: str,
+        input: Union[Original, Docdb, Epodoc],
+    ) -> requests.Response:
+        """
+        Retrieval service for legal data.
+
+        Args:
+            reference_type (str): Any of "publication", "application", or "priority".
+            input (Epodoc or Docdb): The document number as a Epodoc or Docdb data object.
+        Returns:
+            requests.Response: a requests.Response object.
+
+        """
+
+        return self._service_request(
+            dict(
+                service=self.__legal_path__,
+                reference_type=reference_type,
+                input=input,
+            )
+        )
     def number(
         self,
         reference_type: str,

tests/helpers/api_helpers.py

@@ -48,6 +48,11 @@ def assert_image_success(client):
     assert_request_success(response)
     return response
 
+def assert_legal_success(client):
+    response = client.legal(*data)
+    assert_request_success(response)
+    assert "ops:legal" in response.text
+    return response
 
 def assert_published_data_success(client):
     response = client.published_data(*data)

tests/test_api.py

@@ -12,6 +12,7 @@
     assert_family_legal_success,
     assert_family_success,
     assert_image_success,
+    assert_legal_success,
     assert_number_service_success,
     assert_published_data_search_success,
     assert_published_data_search_with_range_success,
@@ -44,6 +45,8 @@ def test_family_legal(all_clients):
 def test_image(all_clients):
     assert_image_success(all_clients)
 
+def test_legal(all_clients):
+    assert_legal_success(all_clients)
 
 def test_published_data(all_clients):
     assert_published_data_success(all_clients)