Allow searching on entity name (#463)

Author: ggravlingen

pygleif/search.py

@@ -14,6 +14,7 @@ class SearchType(str, Enum):
     """Enum for search types."""
 
     FULL_TEXT = "fulltext"
+    LEGAL_NAME = "entity.legalName"
 
 
 class Search(PyGleifBase):
@@ -26,9 +27,7 @@ def __init__(
     ) -> None:
         """Init class."""
         encoded_search_string = parse.quote(search_string, safe="")
-        self.search_string = (
-            f"?filter[{search_type.value.lower()}]={encoded_search_string}"
-        )
+        self.search_string = f"?filter[{search_type.value}]={encoded_search_string}"
 
     @property
     def response(self) -> SearchResponse | None:

tests/test_search.py

@@ -3,6 +3,7 @@
 import pytest
 
 from pygleif import Search
+from pygleif.search import SearchType
 
 
 @pytest.fixture(scope="module", name="gleif_search_fixture")
@@ -11,6 +12,7 @@ def search_data() -> Search:
     return Search("917685991")
 
 
+
 def test_search_lei(gleif_search_fixture: Search):
     """Test LEI attribute."""
     assert gleif_search_fixture.response.data[0].attributes.lei, "549300MLUDYVRQOOXS22"
@@ -25,3 +27,8 @@ def test_search_no_result():
     """Test failing search."""
     search = Search("1bcdefghijklmnopqrstuvwxyz")
     assert search.response.data == []
+
+def test_search_no_result():
+    """Test matching search."""
+    search = Search("Sparbanken Rekarne", search_type=SearchType.LEGAL_NAME)
+    assert len(search.response.data) == 1