for testing purposes

googleapis · Jul 5, 2023 · 65b465f · 65b465f
1 parent dd97722
commit 65b465f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit/v1/test_cross_language.py b/tests/unit/v1/test_cross_language.py
@@ -244,7 +244,7 @@ def callback(keys, applied_changes, read_time):
                 watch = Watch.for_query(query, callback, DocumentSnapshot)
 
                 wrapped_responses = [
-                    firestore.ListenResponse.wrap(proto) for proto in testcase.responses
+                    firestore.ListenResponse.wrap(proto._pb) for proto in testcase.responses
                 ]
                 if testcase.is_error:
                     try:
@@ -336,10 +336,10 @@ def convert_set_option(option):
 def convert_precondition(precond):
     from google.cloud.firestore_v1 import Client
 
-    if precond.HasField("exists"):
+    if precond._pb.HasField("exists"):
         return Client.write_option(exists=precond.exists)
 
-    assert precond.HasField("update_time")
+    assert precond._pb.HasField("update_time")
     return Client.write_option(last_update_time=precond.update_time)