fetch_dcids_by_wikidata_id improvement

datacommons_client/endpoints/resolve.py

@@ -81,7 +81,7 @@ def fetch_dcids_by_name(self,
 
     return self.fetch(node_ids=names, expression=expression)
 
-  def fetch_dcid_by_wikidata_id(
+  def fetch_dcids_by_wikidata_id(
       self,
       wikidata_id: str | list[str],
       entity_type: Optional[str] = None) -> ResolveResponse:

datacommons_client/tests/endpoints/test_resolve_endpoint.py

@@ -52,8 +52,8 @@ def test_fetch_dcid_by_wikidata_id():
   api_mock = MagicMock(spec=API)
   endpoint = ResolveEndpoint(api=api_mock)
 
-  response = endpoint.fetch_dcid_by_wikidata_id(wikidata_id="Q12345",
-                                                entity_type="Country")
+  response = endpoint.fetch_dcids_by_wikidata_id(wikidata_id="Q12345",
+                                                 entity_type="Country")
 
   # Check the response
   assert isinstance(response, ResolveResponse)
@@ -68,6 +68,27 @@ def test_fetch_dcid_by_wikidata_id():
                                         next_token=None)
 
 
+def test_fetch_dcids_list_by_wikidata_id():
+  """Tests the fetch_dcid_by_wikidata_id method."""
+  api_mock = MagicMock(spec=API)
+  endpoint = ResolveEndpoint(api=api_mock)
+
+  response = endpoint.fetch_dcids_by_wikidata_id(
+      wikidata_id=["Q12345", "Q695660"])
+
+  # Check the response
+  assert isinstance(response, ResolveResponse)
+
+  # Check the post request
+  api_mock.post.assert_called_once_with(payload={
+      "nodes": ["Q12345", "Q695660"],
+      "property": "<-wikidataId->dcid",
+  },
+                                        endpoint="resolve",
+                                        all_pages=True,
+                                        next_token=None)
+
+
 def test_fetch_dcid_by_coordinates():
   """Tests the fetch_dcid_by_coordinates method."""
   api_mock = MagicMock(spec=API)