renamed DocumentMatches

Author: daniel-sanche

packages/google-cloud-firestore/google/cloud/firestore_v1/pipeline_expressions.py

@@ -2676,20 +2676,24 @@ def __init__(self):
         super().__init__("rand", [], use_infix_repr=False)
 
 
-def document_matches(query: Expression | str) -> BooleanExpression:
+class DocumentMatches(BooleanExpression):
     """Creates a boolean expression for a document match query.
     
     Example:
         >>> # Find documents matching the query string
-        >>> document_matches("search query")
+        >>> DocumentMatches("search query")
 
     Args:
         query: The search query string or expression.
 
     Returns:
         A new `BooleanExpression` representing the document match.
     """
-    return BooleanExpression(
-        "document_matches", [Expression._cast_to_expr_or_convert_to_constant(query)]
-    )
+
+    def __init__(self, query: Expression | str):
+        super().__init__(
+            "document_matches",
+            [Expression._cast_to_expr_or_convert_to_constant(query)],
+            use_infix_repr=False,
+        )
 

packages/google-cloud-firestore/google/cloud/firestore_v1/pipeline_stages.py

@@ -152,8 +152,8 @@ def __init__(
             language_code (Optional[str]): The BCP-47 language code of text in the search query, such as "en-US" or "sr-Latn".
         """
         if isinstance(query, str):
-            from google.cloud.firestore_v1.pipeline_expressions import document_matches
-            self.query = document_matches(query)
+            from google.cloud.firestore_v1.pipeline_expressions import DocumentMatches
+            self.query = DocumentMatches(query)
         else:
             self.query = query
         self.limit = limit

packages/google-cloud-firestore/tests/unit/v1/test_pipeline_expressions.py

@@ -815,10 +815,10 @@ def test_geo_distance(self):
 
     def test_document_matches(self):
         arg1 = self._make_arg("Query")
-        instance = expr.document_matches(arg1)
+        instance = expr.DocumentMatches(arg1)
         assert instance.name == "document_matches"
         assert instance.params == [arg1]
-        assert repr(instance) == "document_matches(Query)"
+        assert repr(instance) == "DocumentMatches(Query)"
 
     def test_greater_than_or_equal(self):
         arg1 = self._make_arg("Left")