fixed issues after merging main

Author: daniel-sanche

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

@@ -3241,7 +3241,9 @@ class Score(FunctionExpression):
     """
 
     def __init__(self):
-        super().__init__("score", [], use_infix_repr=False)
+        super().__init__(
+            "score", [], repr_function=FunctionExpression._build_standalone_repr()
+        )
 
 
 class DocumentMatches(BooleanExpression):
@@ -3266,7 +3268,7 @@ def __init__(self, query: Expression | str):
         super().__init__(
             "document_matches",
             [Expression._cast_to_expr_or_convert_to_constant(query)],
-            use_infix_repr=False,
+            repr_function=FunctionExpression._build_standalone_repr(),
         )
 
 

packages/google-cloud-firestore/tests/system/pipeline_e2e/logical.yaml

@@ -786,6 +786,51 @@ tests:
     assert_results:
       - title: "Crime and Punishment"
       - title: "The Great Gatsby"
+    assert_proto:
+      pipeline:
+        stages:
+        - args:
+          - referenceValue: /books
+          name: collection
+        - args:
+          - functionValue:
+              args:
+              - functionValue:
+                  args:
+                  - fieldReferenceValue: genre
+                  - stringValue: Romance
+                  name: equal
+              - functionValue:
+                  args:
+                  - fieldReferenceValue: genre
+                  - stringValue: Dystopian
+                  name: equal
+              - functionValue:
+                  args:
+                  - fieldReferenceValue: genre
+                  - stringValue: Fantasy
+                  name: equal
+              - functionValue:
+                  args:
+                  - fieldReferenceValue: published
+                  - integerValue: '1949'
+                  name: greater_than
+              name: nor
+          name: where
+        - args:
+          - mapValue:
+              fields:
+                title:
+                  fieldReferenceValue: title
+          name: select
+        - args:
+          - mapValue:
+              fields:
+                direction:
+                  stringValue: ascending
+                expression:
+                  fieldReferenceValue: title
+          name: sort
   - description: expression_between
     pipeline:
     - Collection: restaurants